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 | }
|
---|