repos/CCSFinger/CCSFinger.cs

// Douglas Thrift
//
// CCS Computer Science
//
// Windows Finger Daemon

using System;
using System.Collections.Generic;
using System.IO;
using System.Net;
using System.Net.Sockets;
using System.ServiceProcess;
using System.Text;
using System.Threading;

public class CCSFinger : ServiceBase
{
        private class Finger
        {
                private class Login
                {
                        private Net.USER_INFO_11 info;
                        private uint id;
                        private string session;
                        private string client;
                        private Wts.WTS_PROTOCOL_TYPE protocol;
                        private Wts.WTS_CONNECTSTATE_CLASS status;
                        private DateTime logon = DateTime.MinValue;

                        public Login(string login)
                        {
                                Net.UserGetInfo(null, login, 11, out info);

                                this.id = uint.MaxValue;
                        }

                        public Login(string login, uint id, string session, string client, Wts.WTS_PROTOCOL_TYPE protocol, Wts.WTS_CONNECTSTATE_CLASS status)
                        {
                                Net.UserGetInfo(null, login, 11, out info);

                                this.id = id;
                                this.session = session;
                                this.client = client;
                                this.protocol = protocol;
                                this.status = status;
                        }

                        public string Client
                        {
                                get { return protocol == Wts.WTS_PROTOCOL_TYPE.WTS_PROTOCOL_TYPE_CONSOLE ? "the Console" : client; }
                        }

                        public string Directory
                        {
                                get { return @"\\Zweihander\" + info.usri11_name; }
                        }

                        public uint Id
                        {
                                get { return id; }
                        }

                        public DateTime LastLogon
                        {
                                get
                                {
                                        if (logon == DateTime.MinValue)
                                                logon = new DateTime(1970, 1, 1) + new TimeSpan((long)info.usri11_last_logon * 10000000);

                                        return logon;
                                }
                        }

                        public string Name
                        {
                                get { return info.usri11_full_name; }
                        }

                        public string Session
                        {
                                get { return session; }
                        }

                        public string Status
                        {
                                get { return status.ToString().Substring(3); }
                        }
                }

                private StringWriter writer = new StringWriter();
                private SortedDictionary<string, List<Login>> logins = new SortedDictionary<string, List<Login>>();
                private List<string> nots = new List<string>();

                private void Full()
                {
                        foreach (string not in nots)
                                writer.Write("finger: {0}: no such user\r\n", not);

                        bool first = true;

                        foreach (KeyValuePair<string, List<Login>> login in logins)
                        {
                                if (first)
                                        first = false;
                                else
                                        writer.Write("\r\n");

                                writer.Write("Login: {0,-32} Name: {1}\r\nDirectory: {2}\r\n", login.Key, login.Value[0].Name, login.Value[0].Directory);

                                foreach (Login login_ in login.Value)
                                        if (login_.Id != uint.MaxValue)
                                                writer.Write("{0} on {1}, from {2}\r\n", login_.Status, login_.Id, login_.Client);

                                if (login.Value[0].LastLogon != new DateTime(1970, 1, 1))
                                        writer.Write("Last login {0:ddd MMM dd HH:mm} ({1})\r\n", login.Value[0].LastLogon.ToLocalTime(), login.Value[0].LastLogon.ToLocalTime().IsDaylightSavingTime() ? "PDT" : "PST");
                                else
                                        writer.Write("Never logged in.\r\n");

                                writer.Write("No Mail.\r\n");

                                string[] files = { ".project", ".plan" };

                                foreach (string file in files)
                                        try
                                        {
                                                StreamReader reader = new StreamReader(new FileStream(Directory.GetFiles(login.Value[0].Directory, file)[0], FileMode.Open, FileAccess.Read));

                                                switch (file)
                                                {
                                                        case ".project":
                                                                writer.Write("Project:\r\n");
                                                                break;
                                                        case ".plan":
                                                                writer.Write("Plan:\r\n");
                                                                break;
                                                }

                                                writer.Write("{0}", reader.ReadToEnd());
                                        }
                                        catch (Exception)
                                        {
                                                if (file == ".plan")
                                                        writer.Write("No Plan.\r\n");
                                        }
                        }
                }

                private void Sessions(IDictionary<string, List<Login>> logins)
                {
                        Wts.WTS_SESSION_INFO[] sessions;
                        uint count;

                        Wts.EnumerateSessions(IntPtr.Zero, 0, 1, out sessions, out count);

                        for (uint index = 0; index != count; ++index)
                        {
                                string name;
                                uint size;

                                Wts.QuerySessionInformation(IntPtr.Zero, sessions[index].SessionId, Wts.WTS_INFO_CLASS.WTSUserName, out name, out size);

                                if (name.Length > 0)
                                {
                                        string session, client;
                                        Wts.WTS_PROTOCOL_TYPE protocol;
                                        Wts.WTS_CONNECTSTATE_CLASS status;

                                        Wts.QuerySessionInformation(IntPtr.Zero, sessions[index].SessionId, Wts.WTS_INFO_CLASS.WTSWinStationName, out session, out size);
                                        Wts.QuerySessionInformation(IntPtr.Zero, sessions[index].SessionId, Wts.WTS_INFO_CLASS.WTSClientName, out client, out size);
                                        Wts.QuerySessionInformation(IntPtr.Zero, sessions[index].SessionId, Wts.WTS_INFO_CLASS.WTSClientProtocolType, out protocol, out size);
                                        Wts.QuerySessionInformation(IntPtr.Zero, sessions[index].SessionId, Wts.WTS_INFO_CLASS.WTSConnectState, out status, out size);

                                        if (!logins.ContainsKey(name))
                                                logins.Add(string.Copy(name), new List<Login>());

                                        logins[name].Add(new Login(name, sessions[index].SessionId, session, client, protocol, status));
                                }
                        }
                }

                public Finger(bool full)
                {
                        Sessions(logins);

                        if (logins.Count < 1)
                                writer.Write("No one logged on.\r\n");
                        else if (full)
                                Full();
                        else
                        {
                                writer.Write("Login           Name                 Id     Session       Status\r\n");

                                foreach (KeyValuePair<string, List<Login>> login_ in logins)
                                        foreach (Login login in login_.Value)
                                                writer.Write("{0,-15} {1,-20} {2,-6} {3,-13} {4}\r\n", login_.Key, login.Name.Length > 20 ? login.Name.Substring(0, 20) : login.Name, login.Id, login.Session, login.Status);
                        }
                }

                public Finger(ICollection<string> names)
                {
                        uint count;
                        Net.NET_DISPLAY_USER[] users;

                        Net.QueryDisplayInformation(null, 1, 0, 100, uint.MaxValue, out count, out users);

                        Dictionary<string, List<Login>> logins_ = new Dictionary<string, List<Login>>();

                        Sessions(logins_);

                        foreach (string name in names)
                        {
                                bool not = true;

                                for (uint index = 0; index != count; ++index)
                                        if (users[index].usri1_name.IndexOf(name, StringComparison.OrdinalIgnoreCase) != -1 || users[index].usri1_full_name.IndexOf(name, StringComparison.OrdinalIgnoreCase) != -1 && !logins.ContainsKey(users[index].usri1_name))
                                        {
                                                logins.Add(users[index].usri1_name, new List<Login>());

                                                logins[users[index].usri1_name].Add(new Login(users[index].usri1_name));

                                                if (logins_.ContainsKey(users[index].usri1_name))
                                                        logins[users[index].usri1_name].AddRange(logins_[users[index].usri1_name]);

                                                not = false;
                                        }

                                if (not)
                                        nots.Add(name);
                        }

                        Full();
                }

                public override string ToString()
                {
                        return writer.ToString();
                }
        }

        private static void Do()
        {
                TcpListener listener = null;

                try
                {
                        listener = new TcpListener(IPAddress.Any, 79);

                        listener.Start();

                        while (true)
                        {
                                TcpClient client = listener.AcceptTcpClient();
                                Thread thread = new Thread(new ParameterizedThreadStart(Do));

                                thread.Start(client);
                        }
                }
                catch (SocketException exception)
                {
                        Console.Error.WriteLine(exception);
                }
                finally
                {
                        listener.Stop();
                }
        }

        private static void Do(object client)
        {
                NetworkStream stream = ((TcpClient)client).GetStream();
                byte[] buffer = new byte[1024];
                int size = 0, byte_;

                while (size != buffer.Length && (byte_ = stream.ReadByte()) != '\r' && byte_ != '\n' && byte_ != -1)
                        buffer[size++] = (byte)byte_;

                string line = Encoding.ASCII.GetString(buffer, 0, size)/*Console.ReadLine()*/;
                List<string> names = new List<string>();
                bool full = false;

                foreach (string name in line.Split(new char[] { ' ' }, StringSplitOptions.RemoveEmptyEntries))
                {
                        if (name == "/W")
                                full = true;
                        else
                                names.Add(name);
                }

                StreamWriter writer = new StreamWriter(stream, new UTF8Encoding(false));

                writer/*Console*/.Write(names.Count > 0 ? new Finger(names) : new Finger(full));
                writer.Close();
                //stream.Close();
                //((TcpClient)client).Close();
        }

        public static int Main(string[] args)
        {
                Do(/*null*/);

                return 0;
        }
}
Revision:	720
Committed:	2006-03-30T04:25:10-08:00 (19 years, 2 months ago) by douglas
File size:	8065 byte(s)
Log Message:	This ought to work better on a socket, but apparently it doesn't...
#	Content
1	// Douglas Thrift
2	//
3	// CCS Computer Science
4	//
5	// Windows Finger Daemon
6
7	using System;
8	using System.Collections.Generic;
9	using System.IO;
10	using System.Net;
11	using System.Net.Sockets;
12	using System.ServiceProcess;
13	using System.Text;
14	using System.Threading;
15
16	public class CCSFinger : ServiceBase
17	{
18	private class Finger
19	{
20	private class Login
21	{
22	private Net.USER_INFO_11 info;
23	private uint id;
24	private string session;
25	private string client;
26	private Wts.WTS_PROTOCOL_TYPE protocol;
27	private Wts.WTS_CONNECTSTATE_CLASS status;
28	private DateTime logon = DateTime.MinValue;
29
30	public Login(string login)
31	{
32	Net.UserGetInfo(null, login, 11, out info);
33
34	this.id = uint.MaxValue;
35	}
36
37	public Login(string login, uint id, string session, string client, Wts.WTS_PROTOCOL_TYPE protocol, Wts.WTS_CONNECTSTATE_CLASS status)
38	{
39	Net.UserGetInfo(null, login, 11, out info);
40
41	this.id = id;
42	this.session = session;
43	this.client = client;
44	this.protocol = protocol;
45	this.status = status;
46	}
47
48	public string Client
49	{
50	get { return protocol == Wts.WTS_PROTOCOL_TYPE.WTS_PROTOCOL_TYPE_CONSOLE ? "the Console" : client; }
51	}
52
53	public string Directory
54	{
55	get { return @"\\Zweihander\" + info.usri11_name; }
56	}
57
58	public uint Id
59	{
60	get { return id; }
61	}
62
63	public DateTime LastLogon
64	{
65	get
66	{
67	if (logon == DateTime.MinValue)
68	logon = new DateTime(1970, 1, 1) + new TimeSpan((long)info.usri11_last_logon * 10000000);
69
70	return logon;
71	}
72	}
73
74	public string Name
75	{
76	get { return info.usri11_full_name; }
77	}
78
79	public string Session
80	{
81	get { return session; }
82	}
83
84	public string Status
85	{
86	get { return status.ToString().Substring(3); }
87	}
88	}
89
90	private StringWriter writer = new StringWriter();
91	private SortedDictionary<string, List<Login>> logins = new SortedDictionary<string, List<Login>>();
92	private List<string> nots = new List<string>();
93
94	private void Full()
95	{
96	foreach (string not in nots)
97	writer.Write("finger: {0}: no such user\r\n", not);
98
99	bool first = true;
100
101	foreach (KeyValuePair<string, List<Login>> login in logins)
102	{
103	if (first)
104	first = false;
105	else
106	writer.Write("\r\n");
107
108	writer.Write("Login: {0,-32} Name: {1}\r\nDirectory: {2}\r\n", login.Key, login.Value[0].Name, login.Value[0].Directory);
109
110	foreach (Login login_ in login.Value)
111	if (login_.Id != uint.MaxValue)
112	writer.Write("{0} on {1}, from {2}\r\n", login_.Status, login_.Id, login_.Client);
113
114	if (login.Value[0].LastLogon != new DateTime(1970, 1, 1))
115	writer.Write("Last login {0:ddd MMM dd HH:mm} ({1})\r\n", login.Value[0].LastLogon.ToLocalTime(), login.Value[0].LastLogon.ToLocalTime().IsDaylightSavingTime() ? "PDT" : "PST");
116	else
117	writer.Write("Never logged in.\r\n");
118
119	writer.Write("No Mail.\r\n");
120
121	string[] files = { ".project", ".plan" };
122
123	foreach (string file in files)
124	try
125	{
126	StreamReader reader = new StreamReader(new FileStream(Directory.GetFiles(login.Value[0].Directory, file)[0], FileMode.Open, FileAccess.Read));
127
128	switch (file)
129	{
130	case ".project":
131	writer.Write("Project:\r\n");
132	break;
133	case ".plan":
134	writer.Write("Plan:\r\n");
135	break;
136	}
137
138	writer.Write("{0}", reader.ReadToEnd());
139	}
140	catch (Exception)
141	{
142	if (file == ".plan")
143	writer.Write("No Plan.\r\n");
144	}
145	}
146	}
147
148	private void Sessions(IDictionary<string, List<Login>> logins)
149	{
150	Wts.WTS_SESSION_INFO[] sessions;
151	uint count;
152
153	Wts.EnumerateSessions(IntPtr.Zero, 0, 1, out sessions, out count);
154
155	for (uint index = 0; index != count; ++index)
156	{
157	string name;
158	uint size;
159
160	Wts.QuerySessionInformation(IntPtr.Zero, sessions[index].SessionId, Wts.WTS_INFO_CLASS.WTSUserName, out name, out size);
161
162	if (name.Length > 0)
163	{
164	string session, client;
165	Wts.WTS_PROTOCOL_TYPE protocol;
166	Wts.WTS_CONNECTSTATE_CLASS status;
167
168	Wts.QuerySessionInformation(IntPtr.Zero, sessions[index].SessionId, Wts.WTS_INFO_CLASS.WTSWinStationName, out session, out size);
169	Wts.QuerySessionInformation(IntPtr.Zero, sessions[index].SessionId, Wts.WTS_INFO_CLASS.WTSClientName, out client, out size);
170	Wts.QuerySessionInformation(IntPtr.Zero, sessions[index].SessionId, Wts.WTS_INFO_CLASS.WTSClientProtocolType, out protocol, out size);
171	Wts.QuerySessionInformation(IntPtr.Zero, sessions[index].SessionId, Wts.WTS_INFO_CLASS.WTSConnectState, out status, out size);
172
173	if (!logins.ContainsKey(name))
174	logins.Add(string.Copy(name), new List<Login>());
175
176	logins[name].Add(new Login(name, sessions[index].SessionId, session, client, protocol, status));
177	}
178	}
179	}
180
181	public Finger(bool full)
182	{
183	Sessions(logins);
184
185	if (logins.Count < 1)
186	writer.Write("No one logged on.\r\n");
187	else if (full)
188	Full();
189	else
190	{
191	writer.Write("Login Name Id Session Status\r\n");
192
193	foreach (KeyValuePair<string, List<Login>> login_ in logins)
194	foreach (Login login in login_.Value)
195	writer.Write("{0,-15} {1,-20} {2,-6} {3,-13} {4}\r\n", login_.Key, login.Name.Length > 20 ? login.Name.Substring(0, 20) : login.Name, login.Id, login.Session, login.Status);
196	}
197	}
198
199	public Finger(ICollection<string> names)
200	{
201	uint count;
202	Net.NET_DISPLAY_USER[] users;
203
204	Net.QueryDisplayInformation(null, 1, 0, 100, uint.MaxValue, out count, out users);
205
206	Dictionary<string, List<Login>> logins_ = new Dictionary<string, List<Login>>();
207
208	Sessions(logins_);
209
210	foreach (string name in names)
211	{
212	bool not = true;
213
214	for (uint index = 0; index != count; ++index)
215	if (users[index].usri1_name.IndexOf(name, StringComparison.OrdinalIgnoreCase) != -1 \|\| users[index].usri1_full_name.IndexOf(name, StringComparison.OrdinalIgnoreCase) != -1 && !logins.ContainsKey(users[index].usri1_name))
216	{
217	logins.Add(users[index].usri1_name, new List<Login>());
218
219	logins[users[index].usri1_name].Add(new Login(users[index].usri1_name));
220
221	if (logins_.ContainsKey(users[index].usri1_name))
222	logins[users[index].usri1_name].AddRange(logins_[users[index].usri1_name]);
223
224	not = false;
225	}
226
227	if (not)
228	nots.Add(name);
229	}
230
231	Full();
232	}
233
234	public override string ToString()
235	{
236	return writer.ToString();
237	}
238	}
239
240	private static void Do()
241	{
242	TcpListener listener = null;
243
244	try
245	{
246	listener = new TcpListener(IPAddress.Any, 79);
247
248	listener.Start();
249
250	while (true)
251	{
252	TcpClient client = listener.AcceptTcpClient();
253	Thread thread = new Thread(new ParameterizedThreadStart(Do));
254
255	thread.Start(client);
256	}
257	}
258	catch (SocketException exception)
259	{
260	Console.Error.WriteLine(exception);
261	}
262	finally
263	{
264	listener.Stop();
265	}
266	}
267
268	private static void Do(object client)
269	{
270	NetworkStream stream = ((TcpClient)client).GetStream();
271	byte[] buffer = new byte[1024];
272	int size = 0, byte_;
273
274	while (size != buffer.Length && (byte_ = stream.ReadByte()) != '\r' && byte_ != '\n' && byte_ != -1)
275	buffer[size++] = (byte)byte_;
276
277	string line = Encoding.ASCII.GetString(buffer, 0, size)/Console.ReadLine()/;
278	List<string> names = new List<string>();
279	bool full = false;
280
281	foreach (string name in line.Split(new char[] { ' ' }, StringSplitOptions.RemoveEmptyEntries))
282	{
283	if (name == "/W")
284	full = true;
285	else
286	names.Add(name);
287	}
288
289	StreamWriter writer = new StreamWriter(stream, new UTF8Encoding(false));
290
291	writer/Console/.Write(names.Count > 0 ? new Finger(names) : new Finger(full));
292	writer.Close();
293	//stream.Close();
294	//((TcpClient)client).Close();
295	}
296
297	public static int Main(string[] args)
298	{
299	Do(/null/);
300
301	return 0;
302	}
303	}