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.Runtime.InteropServices;
using System.ServiceProcess;
using System.Text;
using System.Threading;

public class CCSFinger : ServiceBase
{
        public enum CurrentState : uint
        {
                SERVICE_STOPPED = 0x1,
                SERVICE_START_PENDING = 0x2,
                SERVICE_STOP_PENDING = 0x3,
                SERVICE_RUNNING = 0x4,
                SERVICE_CONTINUE_PENDING = 0x5,
                SERVICE_PAUSE_PENDING = 0x6,
                SERVICE_PAUSED = 0x7
        }

        [StructLayout(LayoutKind.Sequential)]
        public struct SERVICE_STATUS
        {
                public uint dwServiceType;
                public CurrentState dwCurrentState;
                public uint dwControlsAccepted;
                public uint dwWin32ExitCode;
                public uint dwServiceSpecificExitCode;
                public uint dwCheckPoint;
                public uint dwWaitHint;
        }

        private Thread thread;

        public CCSFinger()
        {
                ServiceName = "CCSFinger";
        }

        [DllImport("Advapi32.dll", EntryPoint = "SetServiceStatus")]
        public static extern bool SetServiceStatus(IntPtr hServiceStatus, SERVICE_STATUS lpServiceStatus);

        [DllImport("Advapi32.dll", EntryPoint = "QueryServiceStatus")]
        public static extern bool QueryServiceStatus(IntPtr hService, out SERVICE_STATUS lpServiceStatus);

        protected override void OnStart(string[] args)
        {
                SERVICE_STATUS status;

                QueryServiceStatus(ServiceHandle, out status);

                status.dwCurrentState = CurrentState.SERVICE_START_PENDING;

                SetServiceStatus(ServiceHandle, status);

                thread = new Thread(new ThreadStart(Do));

                thread.Start();

                ExitCode = 0;

                QueryServiceStatus(ServiceHandle, out status);

                status.dwCurrentState = CurrentState.SERVICE_RUNNING;

                SetServiceStatus(ServiceHandle, status);
        }

        protected override void OnStop()
        {
                SERVICE_STATUS status;

                QueryServiceStatus(ServiceHandle, out status);

                status.dwCurrentState = CurrentState.SERVICE_STOP_PENDING;

                thread.Abort();
                thread.Join();

                ExitCode = 0;

                QueryServiceStatus(ServiceHandle, out status);

                status.dwCurrentState = CurrentState.SERVICE_STOPPED;
        }

        private void Do()
        {
                TcpListener listener = null;

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

                        listener.Start();

                        while (true)
                        {
                                Socket socket = listener.AcceptSocket();

                                socket.ReceiveTimeout = 60000;

                                Thread thread = new Thread(new ParameterizedThreadStart(Do));

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

        private void Do(object socket_)
        {
                Socket socket = (Socket)socket_;
                NetworkStream stream = new NetworkStream(socket);

                try
                {
                        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.WriteLine("{0} [{1}]", socket.RemoteEndPoint, line);

                        List<string> names = new List<string>();
                        bool full = false;

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

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

                        Finger finger = names.Count > 0 ? new Finger(writer, names) : new Finger(writer, full);

                        writer.Flush();
                }
                catch (IOException) {}

                socket.Close(60);
        }

        public static int Main(string[] args)
        {
                ServiceBase.Run(new ServiceBase[] { new CCSFinger() });

                return 0;
        }
}
Revision:	725
Committed:	2006-03-31T04:13:36-08:00 (19 years, 3 months ago) by douglas
File size:	3841 byte(s)
Log Message:	Service Worky!
#	User	Rev	Content
1	douglas	719	// Douglas Thrift
2	douglas	710	//
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	douglas	725	using System.Runtime.InteropServices;
13	douglas	710	using System.ServiceProcess;
14			using System.Text;
15			using System.Threading;
16
17			public class CCSFinger : ServiceBase
18			{
19	douglas	725	public enum CurrentState : uint
20	douglas	710	{
21	douglas	725	SERVICE_STOPPED = 0x1,
22			SERVICE_START_PENDING = 0x2,
23			SERVICE_STOP_PENDING = 0x3,
24			SERVICE_RUNNING = 0x4,
25			SERVICE_CONTINUE_PENDING = 0x5,
26			SERVICE_PAUSE_PENDING = 0x6,
27			SERVICE_PAUSED = 0x7
28			}
29	douglas	710
30	douglas	725	[StructLayout(LayoutKind.Sequential)]
31			public struct SERVICE_STATUS
32			{
33			public uint dwServiceType;
34			public CurrentState dwCurrentState;
35			public uint dwControlsAccepted;
36			public uint dwWin32ExitCode;
37			public uint dwServiceSpecificExitCode;
38			public uint dwCheckPoint;
39			public uint dwWaitHint;
40			}
41	douglas	710
42	douglas	725	private Thread thread;
43	douglas	718
44	douglas	725	public CCSFinger()
45			{
46			ServiceName = "CCSFinger";
47			}
48	douglas	718
49	douglas	725	[DllImport("Advapi32.dll", EntryPoint = "SetServiceStatus")]
50			public static extern bool SetServiceStatus(IntPtr hServiceStatus, SERVICE_STATUS lpServiceStatus);
51	douglas	710
52	douglas	725	[DllImport("Advapi32.dll", EntryPoint = "QueryServiceStatus")]
53			public static extern bool QueryServiceStatus(IntPtr hService, out SERVICE_STATUS lpServiceStatus);
54	douglas	710
55	douglas	725	protected override void OnStart(string[] args)
56			{
57			SERVICE_STATUS status;
58	douglas	710
59	douglas	725	QueryServiceStatus(ServiceHandle, out status);
60	douglas	710
61	douglas	725	status.dwCurrentState = CurrentState.SERVICE_START_PENDING;
62	douglas	718
63	douglas	725	SetServiceStatus(ServiceHandle, status);
64	douglas	718
65	douglas	725	thread = new Thread(new ThreadStart(Do));
66	douglas	710
67	douglas	725	thread.Start();
68	douglas	718
69	douglas	725	ExitCode = 0;
70	douglas	710
71	douglas	725	QueryServiceStatus(ServiceHandle, out status);
72	douglas	710
73	douglas	725	status.dwCurrentState = CurrentState.SERVICE_RUNNING;
74	douglas	721
75	douglas	725	SetServiceStatus(ServiceHandle, status);
76			}
77	douglas	718
78	douglas	725	protected override void OnStop()
79			{
80			SERVICE_STATUS status;
81	douglas	710
82	douglas	725	QueryServiceStatus(ServiceHandle, out status);
83	douglas	710
84	douglas	725	status.dwCurrentState = CurrentState.SERVICE_STOP_PENDING;
85	douglas	710
86	douglas	725	thread.Abort();
87			thread.Join();
88	douglas	710
89	douglas	725	ExitCode = 0;
90	douglas	718
91	douglas	725	QueryServiceStatus(ServiceHandle, out status);
92	douglas	720
93	douglas	725	status.dwCurrentState = CurrentState.SERVICE_STOPPED;
94	douglas	710	}
95
96	douglas	725	private void Do()
97	douglas	710	{
98			TcpListener listener = null;
99
100			try
101			{
102			listener = new TcpListener(IPAddress.Any, 79);
103
104			listener.Start();
105
106			while (true)
107			{
108	douglas	722	Socket socket = listener.AcceptSocket();
109	douglas	721
110	douglas	722	socket.ReceiveTimeout = 60000;
111	douglas	721
112	douglas	710	Thread thread = new Thread(new ParameterizedThreadStart(Do));
113
114	douglas	722	thread.Start(socket);
115	douglas	710	}
116			}
117			catch (SocketException exception)
118			{
119			Console.Error.WriteLine(exception);
120			}
121	douglas	725	catch (ThreadAbortException) {}
122	douglas	710	finally
123			{
124			listener.Stop();
125			}
126	douglas	720	}
127	douglas	710
128	douglas	725	private void Do(object socket_)
129	douglas	710	{
130	douglas	722	Socket socket = (Socket)socket_;
131			NetworkStream stream = new NetworkStream(socket);
132	douglas	710
133	douglas	722	try
134			{
135			byte[] buffer = new byte[1024];
136			int size = 0, byte_;
137	douglas	710
138	douglas	722	while (size != buffer.Length && (byte_ = stream.ReadByte()) != '\r' && byte_ != '\n' && byte_ != -1)
139			buffer[size++] = (byte)byte_;
140	douglas	710
141	douglas	722	string line = Encoding.ASCII.GetString(buffer, 0, size);
142
143			Console.WriteLine("{0} [{1}]", socket.RemoteEndPoint, line);
144
145			List<string> names = new List<string>();
146			bool full = false;
147
148			foreach (string name in line.Split(new char[] { ' ' }, StringSplitOptions.RemoveEmptyEntries))
149			{
150			if (name.ToUpper() == "/W")
151			full = true;
152			else
153			names.Add(name);
154			}
155
156			StreamWriter writer = new StreamWriter(stream, new UTF8Encoding(false));
157
158			Finger finger = names.Count > 0 ? new Finger(writer, names) : new Finger(writer, full);
159
160			writer.Flush();
161	douglas	710	}
162	douglas	722	catch (IOException) {}
163	douglas	710
164	douglas	722	socket.Close(60);
165	douglas	710	}
166
167			public static int Main(string[] args)
168			{
169	douglas	725	ServiceBase.Run(new ServiceBase[] { new CCSFinger() });
170	douglas	710
171			return 0;
172			}
173			}