File: censor.cpp

package info (click to toggle)
libircclient 1.10-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 1,580 kB
  • sloc: ansic: 3,135; objc: 790; cpp: 349; makefile: 223; sh: 46; python: 32
file content (213 lines) | stat: -rw-r--r-- 5,519 bytes parent folder | download | duplicates (3)
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
/*
 * Copyright (C) 2004-2009 Georgy Yunaev gyunaev@ulduzsoft.com
 *
 * This example is free, and not covered by LGPL license. There is no 
 * restriction applied to their modification, redistribution, using and so on.
 * You can study them, modify them, use them in your own program - either 
 * completely or partially. By using it you may give me some credits in your
 * program, but you don't have to.
 *
 *
 * This example 'guards' the channel against using abusive language. When 
 * someone says a bad word, it takes some action against him/her, taking 
 * in account the number of times the person uses it. The first time, it just 
 * warns the person through a private message, the second time it warns him 
 * publically in channel, and after the second time it will kicks the insolent
 * out of the channel.
 *
 * To keep it simple, this example reacts only on 'fuck' word, however 
 * is is easy to add more.
 *
 * Features used:
 * - nickname parsing;
 * - handling 'channel' event to track the messages;
 * - handling 'nick' event to track nickname changes;
 * - generating channel and private messages, and kicking.
 */

#include <map>
#include <string>

#include <stdio.h>
#include <errno.h>
#include <string.h>

#if !defined (_WIN32)
	#include <unistd.h>
#endif

#include "libircclient.h"


/*
 * We store data in IRC session context.
 */
typedef struct
{
	char 	* channel;
	char 	* nick;
	std::map <std::string, unsigned int> insolents;

} irc_ctx_t;




void event_connect (irc_session_t * session, const char * event, const char * origin, const char ** params, unsigned int count)
{
	irc_ctx_t * ctx = (irc_ctx_t *) irc_get_ctx (session);
	irc_cmd_join (session, ctx->channel, 0);
}


void event_nick (irc_session_t * session, const char * event, const char * origin, const char ** params, unsigned int count)
{
	char nickbuf[128];

	irc_ctx_t * ctx = (irc_ctx_t *) irc_get_ctx (session);

	if ( !origin || count != 1 )
		return;

	irc_target_get_nick (origin, nickbuf, sizeof(nickbuf));

	if ( ctx->insolents.find(nickbuf) != ctx->insolents.end() )
	{
		printf ("%s has changed its nick to %s to prevent penalties - no way!\n",
			nickbuf, params[0]);
		ctx->insolents[params[0]] = ctx->insolents[nickbuf];
		ctx->insolents.erase (nickbuf);
	}
}


void event_channel (irc_session_t * session, const char * event, const char * origin, const char ** params, unsigned int count)
{
	irc_ctx_t * ctx = (irc_ctx_t *) irc_get_ctx (session);

	if ( !origin || count != 2 )
		return;

	if ( strstr (params[1], "fuck") == 0 )
		return;

	char nickbuf[128], text[256];

	irc_target_get_nick (origin, nickbuf, sizeof(nickbuf));

	if ( ctx->insolents.find(nickbuf) == ctx->insolents.end() )
		ctx->insolents[nickbuf] = 0;

	ctx->insolents[nickbuf]++;

	printf ("'%s' swears in the channel '%s' %d times\n",
			nickbuf,
			params[1],
			ctx->insolents[nickbuf]);

	switch (ctx->insolents[nickbuf])
	{
	case 1:
		// Send a private message
		sprintf (text, "%s, please do not swear in this channel.", nickbuf);
		irc_cmd_msg (session, nickbuf, text);
		break;

	case 2:
		// Send a channel message
		sprintf (text, "%s, do not swear in this channel, or you'll leave it.", nickbuf);
		irc_cmd_msg (session, params[0], text);
		break;

	default:
		// Send a channel notice, and kick the insolent
		sprintf (text, "kicked %s from %s for swearing.", nickbuf, params[0]);
		irc_cmd_me (session, params[0], text);
		irc_cmd_kick (session, nickbuf, params[0], "swearing");
		break;
	}
}


void event_numeric (irc_session_t * session, unsigned int event, const char * origin, const char ** params, unsigned int count)
{
	if ( event > 400 )
	{
		std::string fulltext;
		for ( unsigned int i = 0; i < count; i++ )
		{
			if ( i > 0 )
				fulltext += " ";

			fulltext += params[i];
		}

		printf ("ERROR %d: %s: %s\n", event, origin ? origin : "?", fulltext.c_str());
	}
}


int main (int argc, char **argv)
{
	irc_callbacks_t	callbacks;
	irc_ctx_t ctx;
	unsigned short port = 6667;

	if ( argc != 4 )
	{
		printf ("Usage: %s <[#]server[:port]> <nick> <channel>\n", argv[0]);
		return 1;
	}

	// Initialize the callbacks
	memset (&callbacks, 0, sizeof(callbacks));

	// Set up the callbacks we will use
	callbacks.event_connect = event_connect;
	callbacks.event_channel = event_channel;
	callbacks.event_nick = event_nick;
	callbacks.event_numeric = event_numeric;

	// And create the IRC session; 0 means error
	irc_session_t * s = irc_create_session (&callbacks);

	if ( !s )
	{
		printf ("Could not create IRC session\n");
		return 1;
	}

	ctx.channel = argv[3];
    ctx.nick = argv[2];
	irc_set_ctx (s, &ctx);

	// If the port number is specified in the server string, use the port 0 so it gets parsed
	if ( strchr( argv[1], ':' ) != 0 )
		port = 0;

	// To handle the "SSL certificate verify failed" from command line we allow passing ## in front 
	// of the server name, and in this case tell libircclient not to verify the cert
	if ( argv[1][0] == '#' && argv[1][1] == '#' )
	{
		// Skip the first character as libircclient needs only one # for SSL support, i.e. #irc.freenode.net
		argv[1]++;
		
		irc_option_set( s, LIBIRC_OPTION_SSL_NO_VERIFY );
	}
	
	// Initiate the IRC server connection
	if ( irc_connect (s, argv[1], port, 0, argv[2], 0, 0) )
	{
		printf ("Could not connect: %s\n", irc_strerror (irc_errno(s)));
		return 1;
	}

	// and run into forever loop, generating events
	if ( irc_run (s) )
	{
		printf ("Could not connect or I/O error: %s\n", irc_strerror (irc_errno(s)));
		return 1;
	}

	return 0;
}