D Paste by Chris Miller
Description: ircdsplat.d - nonstandard IRC server example code using splat
Hide line numbers

Create new paste
Post a reply
View replies

Paste:
1  
2  
3  
4  
5  
6  
7  
8  
9  
10  
11  
12  
13  
14  
15  
16  
17  
18  
19  
20  
21  
22  
23  
24  
25  
26  
27  
28  
29  
30  
31  
32  
33  
34  
35  
36  
37  
38  
39  
40  
41  
42  
43  
44  
45  
46  
47  
48  
49  
50  
51  
52  
53  
54  
55  
56  
57  
58  
59  
60  
61  
62  
63  
64  
65  
66  
67  
68  
69  
70  
71  
72  
73  
74  
75  
76  
77  
78  
79  
80  
81  
82  
83  
84  
85  
86  
87  
88  
89  
90  
91  
92  
93  
94  
95  
96  
97  
98  
99  
100  
101  
102  
103  
104  
105  
106  
107  
108  
109  
110  
111  
112  
113  
114  
115  
116  
117  
118  
119  
120  
121  
122  
123  
124  
125  
126  
127  
128  
129  
130  
131  
132  
133  
134  
135  
136  
137  
138  
139  
140  
141  
142  
143  
144  
145  
146  
147  
148  
149  
150  
151  
152  
153  
154  
155  
156  
157  
158  
159  
160  
161  
162  
163  
164  
165  
166  
167  
168  
169  
170  
171  
172  
173  
174  
175  
176  
177  
178  
179  
180  
181  
182  
183  
184  
185  
186  
187  
188  
189  
190  
191  
192  
193  
194  
195  
196  
197  
198  
199  
200  
201  
202  
203  
204  
205  
206  
207  
208  
209  
210  
211  
212  
213  
214  
215  
216  
217  
218  
219  
220  
221  
222  
223  
224  
225  
226  
227  
228  
229  
230  
231  
232  
233  
234  
235  
236  
237  
238  
239  
240  
241  
242  
243  
244  
245  
246  
247  
248  
249  
250  
251  
252  
253  
254  
255  
256  
257  
258  
259  
260  
261  
262  
263  
264  
265  
266  
267  
268  
269  
270  
271  
272  
273  
274  
275  
276  
277  
278  
279  
280  
281  
282  
283  
284  
285  
286  
287  
288  
289  
290  
291  
292  
293  
294  
295  
296  
297  
298  
299  
300  
301  
302  
303  
304  
305  
306  
307  
308  
309  
310  
311  
312  
313  
314  
315  
316  
317  
318  
319  
320  
321  
322  
323  
324  
325  
326  
327  
328  
329  
330  
331  
332  
333  
334  
335  
336  
337  
338  
339  
340  
341  
342  
343  
344  
345  
346  
347  
348  
349  
350  
351  
/*
    Copyright (C) 2007 Christopher E. Miller
    
    This software is provided 'as-is', without any express or implied
    warranty.  In no event will the authors be held liable for any damages
    arising from the use of this software.
    
    Permission is granted to anyone to use this software for any purpose,
    including commercial applications, and to alter it and redistribute it
    freely, subject to the following restrictions:
    
    1. The origin of this software must not be misrepresented; you must not
       claim that you wrote the original software. If you use this software
       in a product, an acknowledgment in the product documentation would be
       appreciated but is not required.
    2. Altered source versions must be plainly marked as such, and must not be
       misrepresented as being the original software.
    3. This notice may not be removed or altered from any source distribution.
*/

import std.socket, std.stdio, std.string, std.ctype;
import splat;


class IrcClientSocket: AsyncTcpSocket
{
    char[] nick;
    SocketQueue queue;
    
    
    char[] fulladdress() // getter
    {
        return nick ~ "!user@foo.bar";
    }
    
    
    bool allowed() // getter
    {
        return 0 != nick.length;
    }
    
    
    void sendLine(char[] s)
    {
        queue.send(s ~ "\r\n");
    }
    
    
    void onLine(char[] line)
    {
        int i;
        
        char[] cmd;
        i = std.string.find(line, ' ');
        if(-1 == i)
        {
            cmd = line;
            line = null;
        }
        else
        {
            cmd = line[0 .. i];
            line = line[i + 1 .. line.length];
        }
        cmd = std.string.toupper(cmd);
        
        // Handle commands...
        if(!allowed)
        {
            switch(cmd)
            {
                case "NICK":
                    for(i = 0; i != line.length; i++)
                    {
                        if(!std.ctype.isalnum(line[i]) && '_' != line[i]
                            && '[' != line[i] && ']' != line[i] && '-' != line[i])
                            break;
                    }
                    line = line[0 .. i];
                    if(!line.length || std.ctype.isdigit(line[0]))
                    {
                        sendLine(":foo.bar 432  :Bad nickname!");
                    }
                    else
                    {
                        foreach(client; clients)
                        {
                            if(!std.string.icmp(client.nick, line))
                            {
                                sendLine(":foo.bar 433  :That nickname is in use!");
                                goto nick_done;
                            }
                        }
                        
                        nick = line.dup;
                        
                        sendLine(":foo.bar 001 " ~ nick ~ " :Welcome to SPLAT!");
                        sendLine(":foo.bar 002 " ~ nick ~ " :Your host is foo.bar");
                        sendLine(":foo.bar 003 " ~ nick ~ " :This server was created Fri Dec 8 2006 at 14:00:00 GMT");
                        sendLine(":foo.bar 422 " ~ nick ~ " :No MOTD");
                        
                        clients[this] = this;
                        
                        broadcast(":" ~ fulladdress ~ " JOIN #splat");
                        sendNamesList();
                    }
                    nick_done:
                    break;
                
                case "QUIT":
                    clients.remove(this);
                    close();
                    break;
                
                default: ;
            }
        }
        else // allowed
        {
            switch(cmd)
            {
                case "QUIT":
                    clients.remove(this);
                    close();
                    //broadcast(":" ~ fulladdress ~ " QUIT " ~ line);
                    if(!line.length || ":" == line)
                    {
                        broadcast(":" ~ fulladdress ~ " QUIT");
                    }
                    else
                    {
                        if(':' == line[0])
                            line = line[1 .. line.length];
                        broadcast(":" ~ fulladdress ~ " QUIT :Quit: " ~ line);
                    }
                    break;
                
                case "PRIVMSG":
                    i = std.string.find(line, ' ');
                    if(-1 != i)
                    {
                        if(!std.string.icmp(line[0 .. i], "#splat"))
                        {
                            line = line[i + 1 .. line.length];
                            if(line.length)
                            {
                                broadcastButOne(":" ~ fulladdress ~ " PRIVMSG #splat " ~ line, this);
                            }
                        }
                        else
                        {
                            cantdothat();
                        }
                    }
                    break;
                
                case "NOTICE":
                    cantdothat();
                    break;
                
                case "NAMES":
                    i = std.string.find(line, ' ');
                    if(-1 != i)
                    {
                        if(!std.string.icmp(line[0 .. i], "#splat"))
                        {
                            sendNamesList();
                        }
                    }
                    break;
                
                case "PING":
                    sendLine(":foo.bar PONG foo.bar " ~ line);
                    break;
                
                default: ;
            }
        }
    }
    
    
    void sendNamesList()
    {
        char[] s;
        s = ":foo.bar 353 " ~ nick ~ " = #splat :";
        foreach(client; clients)
        {
            if(client.allowed)
            {
                s ~= client.nick;
                s ~= " ";
            }
        }
        sendLine(s);
        sendLine(":foo.bar 366 " ~ nick ~ " #splat :End of /NAMES list.");
    }
    
    
    void cantdothat()
    {
        //sendLine(":the.server PRIVMSG " ~ nick ~ " :You can't do that!");
    }
    
    
    void gotReadEvent()
    {
        byte[] peek;
        find_line:
        peek = cast(byte[])queue.peek();
        foreach(idx, b; peek)
        {
            if('\r' == b || '\n' == b)
            {
                if(!idx)
                {
                    queue.receive(1); // Remove from queue.
                    goto find_line;
                }
                queue.receive(idx + 1); // Remove from queue.
                onLine(cast(char[])peek[0 .. idx]);
                goto find_line;
            }
        }
    }
    
    
    void netEvent(Socket sock, EventType type, int err)
    {
        if(err)
        {
            clients.remove(this);
            close();
            if(allowed)
                broadcast(":" ~ fulladdress ~ " QUIT :Connection error");
            return;
        }
        
        switch(type)
        {
            case EventType.CLOSE:
                clients.remove(this);
                close();
                if(allowed)
                    broadcast(":" ~ fulladdress ~ " QUIT :Connection closed");
                break;
            
            case EventType.READ:
                queue.readEvent();
                if(queue.receiveBytes > 1024 * 4)
                {
                    clients.remove(this);
                    close();
                    queue.reset();
                    if(allowed)
                        broadcast(":" ~ fulladdress ~ " QUIT :Excess flood");
                }
                else
                {
                    gotReadEvent();
                }
                break;
            
            case EventType.WRITE:
                if(queue.sendBytes > 1024 * 8)
                {
                    clients.remove(this);
                    close();
                    queue.reset();
                    if(allowed)
                        broadcast(":" ~ fulladdress ~ " QUIT :Excess send-queue");
                }
                else
                {
                    queue.writeEvent();
                }
                break;
            
            default: ;
        }
    }
}


class IrcListenSocket: AsyncTcpSocket
{
    override IrcClientSocket accepting()
    {
        return new IrcClientSocket();
    }
    
    
    void netEvent(Socket sock, EventType type, int err)
    {
        if(!err)
        {
            IrcClientSocket nsock;
            nsock = cast(IrcClientSocket)sock.accept();
            writefln("Connection accepted from %s", nsock.remoteAddress);
            nsock.queue = new SocketQueue(nsock);
            nsock.event(EventType.READ | EventType.WRITE | EventType.CLOSE, &nsock.netEvent);
        }
    }
}


IrcClientSocket[IrcClientSocket] clients;


void broadcast(char[] s)
{
    foreach(client; clients)
    {
        client.sendLine(s);
    }
}


void broadcastButOne(char[] s, IrcClientSocket except)
{
    foreach(client; clients)
    {
        if(client !is except)
            client.sendLine(s);
    }
}


void main()
{
    bool done = false;
    scope lsock = new IrcListenSocket;
    scope lsockaddr = new InternetAddress(InternetAddress.ADDR_ANY, 1667); // Not standard IRC port.. not standard IRC server.
    lsock.bind(lsockaddr);
    lsock.listen(10);
    lsock.event(EventType.ACCEPT, &lsock.netEvent);
    writefln("Server ready");
    do
    {
        try
        {
            splat.run();
            done = true; // run returned upon request.
        }
        catch(Object o)
        {
            writefln("Error: %s", o);
        }
    }
    while(!done);
}

Replies:
Reply by Chris Miller

Reply by Chris Miller
Note that it's just example code. It doesn't even check for max users or expect many users; flooding it with clients will probably crash it (it's related to the SocketSet constructor limit).

Reply by Chris Miller
Another note: the above code is Phobos-only.

Reply by Chris Miller
Here is a version of the above code that works with both Phobos and Tango: http://paste.dprogramming.com/dpudsook
The code is a bit ugly because it is supporting both Phobos and Tango in one, so might not be a good example. If someone wants to clean it up, please do so.

    (some replies deleted)