wip

OpenVpnManagement/Manager.cs

@@ -1,136 +1,171 @@
-using System;
-using System.Collections.Generic;
-using System.Linq;
-using System.Text;
-using System.Threading.Tasks;
-using System.Net.Sockets;
-using System.Net;
-
-namespace OpenVpnManagement {
- public class Manager {
- enum Signal {
- Hup,
- Term,
- Usr1,
- Usr2
- }
-
- private String ovpnFileName; //todo
- private String host;
- private int port;
- private int timeout;
- private Socket socket; 
-
- public void Manager(String host = "localhost", int port = 1194, int timeout = 10) {
- socket = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);
- try {
- socket.Connect(host, port);
- } catch (Exception ex) {
- //todo
- }
- }
-
- public string GetStatus() {
- return this.SendCommand("status");
- }
-
- /// <summary>
- /// State
- /// </summary>
- /// <returns></returns>
- public string GetState() {
- return this.SendCommand("state");
- }
-
- public string GetState(int n = 1) {
- return this.SendCommand(string.Format("state {}", n.ToString());
- }
-
- public string GetStateAll() {
- return this.SendCommand("state all");
- }
-
- public string SetStateOn() {
- return this.SendCommand("state on");
- }
-
- public string SetStateOnAll() {
- return this.SendCommand("state on all");
- }
-
- public string GetStateOff() {
- return this.SendCommand("state off");
- }
-
- public string GetVersion() {
- return this.SendCommand("version");
- }
-
- public string GetGetPid() {
- return this.SendCommand("pid");
- }
-
- public string SendSignal(Signal sgn) {
- return this.SendCommand(string.Format("SIG{}", sgn.ToString().ToUpper()); 
- }
-
- public string Mute() {
- return this.SendCommand("pid");
- }
-
- public string GetEcho() {
- return this.SendCommand("echo");
- }
-
- public string GetHelp() {
- return this.SendCommand("help");
- }
-
- public string Kill(string name) {
- return this.SendCommand(string.Format("kill {}", name));
- }
-
- public string Kill(string host, int port) {
- return this.SendCommand(string.Format("kill {}:{}", host, port));
- }
-
- public string Net() {
- return this.SendCommand("net");
- }
-
-
- /// <summary>
- /// Logs
- /// </summary>
- /// <returns></returns>
- public string GetLogAll() {
- return this.SendCommand("state off");
- }
-
- public string SetLogOn() {
- return this.SendCommand("log on");
- }
-
- public string SetLogOnAll() {
- return this.SendCommand("log on all");
- }
-
- public string SetLogOff() {
- return this.SendCommand("log off");
- }
-
- public string GetLog(int n = 1) {
- return this.SendCommand(string.Format("log {}", n));
- }
-
-
- private string SendCommand(String cmd) {
- byte[] buffer = Encoding.Default.GetBytes(cmd);
- socket.Send(buffer, 0, buffer.Length, 0);
- buffer = new byte[255];
- int rec = socket.Receive(buffer, 0, buffer.Length, 0);
-
- return rec.ToString();
- }
- }
-}
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+using System.Net.Sockets;
+using System.Net;
+
+namespace OpenVpnManagement {
+ public class Manager {
+
+ public enum Signal {
+ Hup,
+ Term,
+ Usr1,
+ Usr2
+ }
+
+ private String host;
+ private int port;
+ private int timeout;
+ private Socket socket;
+ private const int bufferSize = 1024;
+
+ public Manager(String host, int port) {
+ socket = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);
+ socket.Connect(host, port);
+ GreetServer();
+ }
+
+ public string GetStatus() {
+ return this.SendCommand("status");
+ }
+
+ /// <summary>
+ /// State
+ /// </summary>
+ /// <returns></returns>
+ public string GetState() {
+ return this.SendCommand("state");
+ }
+
+ public string GetState(int n = 1) {
+ return this.SendCommand(string.Format("state {}", n.ToString()));
+ }
+
+ public string GetStateAll() {
+ return this.SendCommand("state all");
+ }
+
+ public string SetStateOn() {
+ return this.SendCommand("state on");
+ }
+
+ public string SetStateOnAll() {
+ return this.SendCommand("state on all");
+ }
+
+ public string GetStateOff() {
+ return this.SendCommand("state off");
+ }
+
+ public string GetVersion() {
+ return this.SendCommand("version");
+ }
+
+ public string GetPid() {
+ return this.SendCommand("pid");
+ }
+
+ public string SendSignal(Signal sgn) {
+ return this.SendCommand(string.Format("SIG{}", sgn.ToString().ToUpper()));
+ }
+
+ public string Mute() {
+ return this.SendCommand("pid");
+ }
+
+ public string GetEcho() {
+ return this.SendCommand("echo");
+ }
+
+ public string GetHelp() {
+ return this.SendCommand("help");
+ }
+
+ public string Kill(string name) {
+ return this.SendCommand(string.Format("kill {}", name));
+ }
+
+ public string Kill(string host, int port) {
+ return this.SendCommand(string.Format("kill {}:{}", host, port));
+ }
+
+ public string GetNet() {
+ return this.SendCommand("net");
+ }
+
+
+ /// <summary>
+ /// Logs
+ /// </summary>
+ /// <returns></returns>
+ public string GetLogAll() {
+ return this.SendCommand("state off");
+ }
+
+ public string SetLogOn() {
+ return this.SendCommand("log on");
+ }
+
+ public string SetLogOnAll() {
+ return this.SendCommand("log on all");
+ }
+
+ public string SetLogOff() {
+ return this.SendCommand("log off");
+ }
+
+ public string GetLog(int n = 1) {
+ return this.SendCommand(string.Format("log {}", n));
+ }
+
+ public string SendMalCommand() {
+ return this.SendCommand("fdsfds");
+ }
+
+ private static string TreamRetrievedString(string s) {
+ return s.Replace("\0", "");
+ }
+
+ private void GreetServer() {
+ var bf = new byte[bufferSize];
+ int rb = socket.Receive(bf, 0, bf.Length, SocketFlags.None);
+ if (rb < 1) {
+ throw new SocketException();
+ }
+ }
+
+ private string SendCommand(String cmd) {
+ socket.Send(Encoding.Default.GetBytes(cmd + "\r\n"));
+
+ var bf = new byte[bufferSize];
+ var sb = new System.Text.StringBuilder();
+ int rb;
+ string s = "";
+ while (true) {
+ rb = socket.Receive(bf, 0, bf.Length, 0);
+ s = Encoding.UTF8.GetString(bf).Replace("\0", "");
+ if (rb < bf.Length) {
+ if (s.Contains("\r\nEND")) {
+ var a = s.Substring(0, s.IndexOf("\r\nEND"));
+ sb.Append(a);
+ } else if (s.Contains("SUCCESS: ")) {
+ var a = s.Replace("SUCCESS: ", "").Replace("\r\n", "");
+ sb.Append(a);
+ } else if (s.Contains("ERROR: ")) {
+ var msg = s.Replace("ERROR: ", "").Replace("\r\n", "");
+ throw new ArgumentException(msg);
+ }
+
+ break;
+ } else {
+ sb.Append(s);
+ }
+ }
+
+ return sb.ToString();
+ }
+ }
+}