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