| 1 | #include <Constants.au3> |
|---|
| 2 | |
|---|
| 3 | ;; SETTINGS HERE, CHANGE THESE |
|---|
| 4 | Global Const $sAdminPass = "manadar" ;; Admin password. Change this please. |
|---|
| 5 | ;; END OF SETTINGS, STOP CHANGING!! |
|---|
| 6 | |
|---|
| 7 | ;; TCP Variables |
|---|
| 8 | Dim $sMaxConnections = 10 |
|---|
| 9 | Dim $hSocket[$sMaxConnections], $sBuffer[$sMaxConnections], $iAuth[$sMaxConnections], $hCommand[$sMaxConnections], $bIgnoreCommand[$sMaxConnections] |
|---|
| 10 | |
|---|
| 11 | ;; TCP Options |
|---|
| 12 | Dim $sIPAddress = @IPAddress1, $nPort = 5000 |
|---|
| 13 | |
|---|
| 14 | TCPStartup() |
|---|
| 15 | |
|---|
| 16 | $sMainSocket = TCPListen($sIPAddress, $nPort, 5) |
|---|
| 17 | If @error Then |
|---|
| 18 | Switch @error |
|---|
| 19 | Case 1 |
|---|
| 20 | _FatalError("The listening address was incorrect (Possibly another server was already running): " & $sIPAddress) |
|---|
| 21 | Case 2 |
|---|
| 22 | _FatalError("The listening port was incorrect (Possibly another server was already running): " & $nPort) |
|---|
| 23 | Case Else |
|---|
| 24 | _FatalError("Unable to set up a listening server on " & $sIPAddress & ":" & $nPort) |
|---|
| 25 | EndSwitch |
|---|
| 26 | EndIf |
|---|
| 27 | |
|---|
| 28 | While 1 |
|---|
| 29 | ;; Accept new incoming clients, and ask them to authorise. |
|---|
| 30 | $sNewSocket = TCPAccept($sMainSocket) |
|---|
| 31 | If $sNewSocket > -1 Then |
|---|
| 32 | For $x = 0 To UBound($hSocket) - 1 |
|---|
| 33 | If Not $hSocket[$x] Then |
|---|
| 34 | $hSocket[$x] = $sNewSocket |
|---|
| 35 | $iAuth[$x] = 0 |
|---|
| 36 | ;TCPSend($hSocket[$x], $sWelcomeMessage & @CRLF & @CRLF) |
|---|
| 37 | TCPSend($hSocket[$x], "Please enter the administrator password" & @CRLF & ">") |
|---|
| 38 | ExitLoop |
|---|
| 39 | EndIf |
|---|
| 40 | Next |
|---|
| 41 | EndIf |
|---|
| 42 | |
|---|
| 43 | ;; Loop through existing connections, check if they sent us any data |
|---|
| 44 | For $x = 0 To UBound($hSocket) - 1 |
|---|
| 45 | If $hSocket[$x] Then |
|---|
| 46 | |
|---|
| 47 | ;; Handle incoming data |
|---|
| 48 | $sData = TCPRecv($hSocket[$x], 100) |
|---|
| 49 | $sBuffer[$x] &= $sData |
|---|
| 50 | If @error Then |
|---|
| 51 | MsgBox(0,"test",@error) |
|---|
| 52 | TCPCloseSocket($hSocket[$x]) |
|---|
| 53 | $hSocket[$x] = "" |
|---|
| 54 | $sBuffer[$x] = "" |
|---|
| 55 | $iAuth[$x] = 0 |
|---|
| 56 | ProcessClose($hCommand[$x]) |
|---|
| 57 | ElseIf Asc($sData) = 0x8 Then ;backspace received |
|---|
| 58 | $len = StringLen($sBuffer[$x]) |
|---|
| 59 | $sBuffer[$x] = StringTrimRight($sBuffer[$x], 2) ; trim the buffer |
|---|
| 60 | If $len = 1 Then |
|---|
| 61 | TCPSend($hSocket[$x], ">") |
|---|
| 62 | Else |
|---|
| 63 | TCPSend($hSocket[$x], " " & Chr(0x8)) |
|---|
| 64 | EndIf |
|---|
| 65 | EndIf |
|---|
| 66 | |
|---|
| 67 | ;; Handle data, in case data is complete: ended with newline |
|---|
| 68 | If StringInStr($sBuffer[$x], @CRLF) Then |
|---|
| 69 | $sBuffer[$x] = StringTrimRight($sBuffer[$x], 2) |
|---|
| 70 | If StringLen($sBuffer[$x]) = 0 Then |
|---|
| 71 | $sBuffer[$x] = " " ; You cannot send "nothing" over TCP (no packet is sent), thus I replace it with something non obstrusive |
|---|
| 72 | EndIf |
|---|
| 73 | |
|---|
| 74 | ;; Check if user is authorised |
|---|
| 75 | If $iAuth[$x] == 0 Then |
|---|
| 76 | ;; Not authorised, user is typing password |
|---|
| 77 | If ($sBuffer[$x] == $sAdminPass) Then |
|---|
| 78 | $iAuth[$x] = 1 |
|---|
| 79 | $hCommand[$x] = Run("cmd", @ScriptDir, @SW_HIDE, $STDIN_CHILD + $STDERR_CHILD + $STDOUT_CHILD) |
|---|
| 80 | TCPSend($hSocket[$x], "Administrator authorisation granted." & @CRLF & @CRLF) |
|---|
| 81 | ConsoleWrite($x & " logged in with admin password." & @CRLF) |
|---|
| 82 | Else |
|---|
| 83 | TCPSend($hSocket[$x], "Access denied." & @CRLF & ">") |
|---|
| 84 | EndIf |
|---|
| 85 | Else |
|---|
| 86 | If $sBuffer[$x] <> "" Then ; stop making output if user starts typing, then punch him in the face or something for fucking it up |
|---|
| 87 | ConsoleWrite($sBuffer[$x] & @CRLF) |
|---|
| 88 | StdinWrite($hCommand[$x], $sBuffer[$x] & @CRLF) |
|---|
| 89 | $bIgnoreCommand[$x] = True |
|---|
| 90 | EndIf |
|---|
| 91 | EndIf |
|---|
| 92 | $sBuffer[$x] = "" |
|---|
| 93 | EndIf |
|---|
| 94 | |
|---|
| 95 | If $iAuth[$x] Then ; no @CRLF in buffer, but user is authed for output |
|---|
| 96 | $line = StdoutRead($hCommand[$x]) |
|---|
| 97 | If $line <> "" Then |
|---|
| 98 | If $bIgnoreCommand[$x] Then |
|---|
| 99 | $line = StringTrimLeft($line, StringInStr($line, @LF)+1) |
|---|
| 100 | $bIgnoreCommand[$x] = False |
|---|
| 101 | EndIf |
|---|
| 102 | ConsoleWrite($line & @CRLF) |
|---|
| 103 | TCPSend($hSocket[$x], $line) |
|---|
| 104 | EndIf |
|---|
| 105 | EndIf |
|---|
| 106 | EndIf |
|---|
| 107 | Next |
|---|
| 108 | WEnd |
|---|
| 109 | |
|---|
| 110 | Func MessageBroadcast($sMsg) |
|---|
| 111 | For $n = 0 To UBound($hSocket) - 1 |
|---|
| 112 | If $hSocket[$n] AND $iAuth[$n] == 1 Then |
|---|
| 113 | TCPSend($hSocket[$n], $sMsg) |
|---|
| 114 | EndIf |
|---|
| 115 | Next |
|---|
| 116 | EndFunc ;==>MessageBroadcast |
|---|
| 117 | |
|---|
| 118 | Func _FatalError($msg) |
|---|
| 119 | ConsoleWrite(@CRLF & "! " & $msg & @CRLF) |
|---|
| 120 | ;MsgBox(0, $sAppName & $nVersion, "A fatal error has occured and " & $sAppName & " has to be closed with the error message: " & @CRLF & $msg) |
|---|
| 121 | Exit |
|---|
| 122 | EndFunc |
|---|
