| 1 | using System;
 | 
|---|
| 2 | using System.Net;
 | 
|---|
| 3 | using System.Net.Sockets;
 | 
|---|
| 4 | using System.Text;
 | 
|---|
| 5 | 
 | 
|---|
| 6 | namespace AllocsFixes.NetConnections.Servers.Telnet
 | 
|---|
| 7 | {
 | 
|---|
| 8 |         public class TelnetConnection : IConnection
 | 
|---|
| 9 |         {
 | 
|---|
| 10 |                 private bool authenticated = false;
 | 
|---|
| 11 |                 private bool authEnabled;
 | 
|---|
| 12 |                 private TcpClient client;
 | 
|---|
| 13 |                 private NetworkStream stream;
 | 
|---|
| 14 |                 private int lineBufferLength = 0;
 | 
|---|
| 15 |                 private byte[] lineBuffer = new byte[200];
 | 
|---|
| 16 |                 private EndPoint endpoint;
 | 
|---|
| 17 | 
 | 
|---|
| 18 |                 public TelnetConnection (TcpClient client, bool authEnabled)
 | 
|---|
| 19 |                 {
 | 
|---|
| 20 |                         this.authEnabled = authEnabled;
 | 
|---|
| 21 |                         this.client = client;
 | 
|---|
| 22 |                         this.stream = client.GetStream ();
 | 
|---|
| 23 |                         this.endpoint = client.Client.RemoteEndPoint;
 | 
|---|
| 24 |                 }
 | 
|---|
| 25 | 
 | 
|---|
| 26 |                 public EndPoint GetEndPoint ()
 | 
|---|
| 27 |                 {
 | 
|---|
| 28 |                         return endpoint;
 | 
|---|
| 29 |                 }
 | 
|---|
| 30 | 
 | 
|---|
| 31 |                 private void WriteByte (byte v)
 | 
|---|
| 32 |                 {
 | 
|---|
| 33 |                         if (!IsClosed () && stream.CanWrite) {
 | 
|---|
| 34 |                                 stream.WriteByte (v);
 | 
|---|
| 35 |                         }
 | 
|---|
| 36 |                 }
 | 
|---|
| 37 | 
 | 
|---|
| 38 |                 public void WriteLine (string s)
 | 
|---|
| 39 |                 {
 | 
|---|
| 40 |                         try {
 | 
|---|
| 41 |                                 if (!IsClosed () && stream.CanWrite) {
 | 
|---|
| 42 |                                         byte[] utfData = Encoding.UTF8.GetBytes (s);
 | 
|---|
| 43 |                                         stream.Write(utfData, 0, utfData.Length);
 | 
|---|
| 44 |                                         WriteByte (13);
 | 
|---|
| 45 |                                         WriteByte (10);
 | 
|---|
| 46 |                                 }
 | 
|---|
| 47 |                         } catch (Exception e) {
 | 
|---|
| 48 |                                 Log.Out ("Error writing to client: " + e);
 | 
|---|
| 49 |                         }
 | 
|---|
| 50 |                 }
 | 
|---|
| 51 | 
 | 
|---|
| 52 |                 public void Close ()
 | 
|---|
| 53 |                 {
 | 
|---|
| 54 |                         if (client != null)
 | 
|---|
| 55 |                                 client.Close ();
 | 
|---|
| 56 |                         client = null;
 | 
|---|
| 57 |                 }
 | 
|---|
| 58 | 
 | 
|---|
| 59 |                 public bool IsClosed ()
 | 
|---|
| 60 |                 {
 | 
|---|
| 61 |                         if (client != null && !client.Connected) {
 | 
|---|
| 62 |                                 Log.Out ("Telnet connection interrupted: " + endpoint);
 | 
|---|
| 63 |                         }
 | 
|---|
| 64 |                         return (client == null) || (!client.Connected);
 | 
|---|
| 65 |                 }
 | 
|---|
| 66 | 
 | 
|---|
| 67 |                 public bool IsAuthenticated ()
 | 
|---|
| 68 |                 {
 | 
|---|
| 69 |                         return !authEnabled || authenticated;
 | 
|---|
| 70 |                 }
 | 
|---|
| 71 | 
 | 
|---|
| 72 |                 public void SetAuthenticated ()
 | 
|---|
| 73 |                 {
 | 
|---|
| 74 |                         authenticated = true;
 | 
|---|
| 75 |                 }
 | 
|---|
| 76 | 
 | 
|---|
| 77 |                 public bool Read ()
 | 
|---|
| 78 |                 {
 | 
|---|
| 79 |                         while (!IsClosed() && stream.CanRead && stream.DataAvailable) {
 | 
|---|
| 80 |                                 int b = stream.ReadByte ();
 | 
|---|
| 81 |                                 if (b == '\r' || b == '\n') {
 | 
|---|
| 82 |                                         if (lineBufferLength > 0)
 | 
|---|
| 83 |                                                 return true;
 | 
|---|
| 84 |                                 } else if (b >= 0) {
 | 
|---|
| 85 |                                         lineBuffer[lineBufferLength] = (byte)b;
 | 
|---|
| 86 |                                         lineBufferLength++;
 | 
|---|
| 87 |                                 }
 | 
|---|
| 88 |                         }
 | 
|---|
| 89 |                         return false;
 | 
|---|
| 90 |                 }
 | 
|---|
| 91 | 
 | 
|---|
| 92 |                 public string GetLine ()
 | 
|---|
| 93 |                 {
 | 
|---|
| 94 |                         string res = Encoding.UTF8.GetString(lineBuffer, 0, lineBufferLength);;
 | 
|---|
| 95 |                         lineBufferLength = 0;
 | 
|---|
| 96 |                         return res;
 | 
|---|
| 97 |                 }
 | 
|---|
| 98 | 
 | 
|---|
| 99 |         }
 | 
|---|
| 100 | }
 | 
|---|