D Paste by Chris Miller
Description: ircdsplat.d - nonstandard IRC server example code using splat (Phobos and Tango)
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  
352  
353  
354  
355  
356  
357  
358  
359  
360  
361  
362  
363  
364  
365  
366  
367  
368  
369  
370  
371  
372  
373  
374  
375  
376  
377  
378  
379  
380  
381  
382  
383  
384  
385  
386  
387  
388  
389  
390  
391  
392  
393  
394  
395  
396  
397  
398  
399  
400  
401  
402  
403  
404  
405  
406  
407  
408  
409  
410  
411  
412  
413  
414  
415  
416  
417  
418  
419  
420  
421  
422  
423  
424  
425  
426  
427  
428  
429  
430  
431  
432  
433  
434  
435  
436  
437  
438  
439  
440  
441  
442  
443  
444  
445  
446  
447  
448  
449  
450  
451  
452  
453  
454  
455  
456  
457  
458  
459  
460  
461  
462  
/*
    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.
*/


// To compile:
//     dmd ircdsplat splat.lib ws2_32.lib -Id:\d\import


version(Tango)
{
    import tango.net.Socket;
    
    alias tango.net.Socket.IPv4Address NetAddr;
    
    
    import tango.text.Ascii;
    
    alias tango.text.Ascii.toUpper strtoupper;
    
    alias tango.text.Ascii.icompare stricompare;
    
    
    import tango.text.Util;
    
    int strfindchar(char[] src, char ch)
    {
        auto x = tango.text.Util.locate(src, ch);
        if(x == src.length)
            return -1;
        return x;
    }
    
    
    import tango.io.Console;
    
    void writestrline(char[] s)
    {
        tango.io.Console.Cout.append(s).newline;
    }
    
    
    bool charisdigit(dchar ch)
    {
        return ch >= '0' && ch <= '9';
    }
    
    bool charisalnum(dchar ch)
    {
        return (ch >= '0' && ch <= '9')
            || (ch >= 'a' && ch <= 'z')
            || (ch >= 'A' && ch <= 'Z');
    }
}
else
{
    import std.socket;
    
    alias std.socket.InternetAddress NetAddr;
    
    
    import std.string;
    
    alias std.string.toupper strtoupper;
    
    alias std.string.find strfindchar;
    
    alias std.string.icmp stricompare;
    
    
    import std.ctype;
    
    alias std.ctype.isalnum charisalnum;
    
    alias std.ctype.isdigit charisdigit;
    
    
    import std.stdio;
    
    void writestrline(char[] s)
    {
        writefln("%s", s);
    }
}

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 = strfindchar(line, ' ');
        if(-1 == i)
        {
            cmd = line;
            line = null;
        }
        else
        {
            cmd = line[0 .. i];
            line = line[i + 1 .. line.length];
        }
        cmd = strtoupper(cmd);
        
        // Handle commands...
        if(!allowed)
        {
            switch(cmd)
            {
                case "NICK":
                    for(i = 0; i != line.length; i++)
                    {
                        if(!charisalnum(line[i]) && '_' != line[i]
                            && '[' != line[i] && ']' != line[i] && '-' != line[i])
                            break;
                    }
                    line = line[0 .. i];
                    if(!line.length || charisdigit(line[0]))
                    {
                        sendLine(":foo.bar 432  :Bad nickname!");
                    }
                    else
                    {
                        foreach(client; clients)
                        {
                            if(!stricompare(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);
                    closeSocket();
                    break;
                
                default: ;
            }
        }
        else // allowed
        {
            switch(cmd)
            {
                case "QUIT":
                    clients.remove(this);
                    closeSocket();
                    //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 = strfindchar(line, ' ');
                    if(-1 != i)
                    {
                        if(!stricompare(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":
                    if(!stricompare(line, "#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);
            closeSocket();
            if(allowed)
                broadcast(":" ~ fulladdress ~ " QUIT :Connection error");
            return;
        }
        
        switch(type)
        {
            case EventType.CLOSE:
                clients.remove(this);
                closeSocket();
                if(allowed)
                    broadcast(":" ~ fulladdress ~ " QUIT :Connection closed");
                break;
            
            case EventType.READ:
                queue.readEvent();
                if(queue.receiveBytes > 1024 * 4)
                {
                    clients.remove(this);
                    closeSocket();
                    queue.reset();
                    if(allowed)
                        broadcast(":" ~ fulladdress ~ " QUIT :Excess flood");
                }
                else
                {
                    gotReadEvent();
                }
                break;
            
            case EventType.WRITE:
                if(queue.sendBytes > 1024 * 8)
                {
                    clients.remove(this);
                    closeSocket();
                    queue.reset();
                    if(allowed)
                        broadcast(":" ~ fulladdress ~ " QUIT :Excess send-queue");
                }
                else
                {
                    queue.writeEvent();
                }
                break;
            
            default: ;
        }
    }
    
    
    version(Tango)
        private const bool _IS_TANGO = true;
    else
        private const bool _IS_TANGO = false;
    
    static if(_IS_TANGO && is(typeof(&this.detach)))
    {
        alias detach closeSocket;
    }
    else
    {
        alias close closeSocket;
    }
}


class IrcListenSocket: AsyncTcpSocket
{
    version(Tango)
    {
        override IrcClientSocket accept()
        {
            return accept(new IrcClientSocket());
        }
        
        // Overload.
        alias AsyncTcpSocket.accept accept;
    }
    else
    {
        override IrcClientSocket accepting()
        {
            return new IrcClientSocket();
        }
    }
    
    
    void netEvent(Socket sock, EventType type, int err)
    {
        if(!err)
        {
            IrcClientSocket nsock;
            nsock = cast(IrcClientSocket)sock.accept();
            version(Tango)
                writestrline("Connection accepted from " ~ nsock.remoteAddress.toUtf8());
            else
                writestrline("Connection accepted from " ~ nsock.remoteAddress.toString());
            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 NetAddr(NetAddr.ADDR_ANY, 1667); // Not standard IRC port.. not standard IRC server.
    lsock.bind(lsockaddr);
    lsock.listen(10);
    lsock.event(EventType.ACCEPT, &lsock.netEvent);
    writestrline("Server ready on port 1667");
    do
    {
        try
        {
            splat.run();
            done = true; // run returned upon request.
        }
        catch(Object o)
        {
            version(Tango)
                writestrline("Error: " ~ o.toUtf8());
            else
                writestrline("Error: " ~ o.toString());
        }
    }
    while(!done);
}

Replies:
Reply by Chris Miller
http://www.dprogramming.com/splat.php

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).

    (some replies deleted)