|
- /*
- * 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 emulates a simple file server. Only 'list' and 'get' commands
- * are supported.
- *
- * Features used:
- * - automatic nickname parsing using LIBIRC_OPTION_STRIPNICKS;
- * - handling privmsg events to parse commands;
- * - generating listings and DCC file transfer;
- *
- * $Id: ircftp.cpp 94 2012-01-18 08:04:49Z gyunaev $
- */
-
- #include <map>
- #include <string>
-
- #include <stdio.h>
- #include <errno.h>
- #include <string.h>
- #include <unistd.h>
- #include <dirent.h>
-
- #include "libircclient.h"
-
-
- #define FILES_DIR "."
-
- /*
- * We store data in IRC session context.
- */
- typedef struct
- {
- std::string channel;
- std::string nick;
-
- } irc_ctx_t;
-
-
-
- void dcc_callback (irc_session_t * session, irc_dcc_t id, int status, void * ctx, const char * data, unsigned int length)
- {
- if ( status == 0 && length == 0 )
- {
- printf ("File sent successfully\n");
- }
- else if ( status )
- {
- printf ("File sent error: %s (%d)\n", irc_strerror(status), status);
- }
- else
- {
- printf ("File sent progress: %d\n", length);
- }
- }
-
-
- 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.c_str(), 0);
- }
-
-
- void event_privmsg (irc_session_t * session, const char * event, const char * origin, const char ** params, unsigned int count)
- {
- if ( !origin || count != 2 )
- return;
-
- if ( !strcasecmp (params[1], "list") )
- {
- dirent *d;
- DIR *dir = opendir(FILES_DIR);
-
- if ( !dir )
- return;
-
- while ( (d = readdir (dir)) != 0 )
- {
- if ( !strcmp (d->d_name, ".") )
- continue;
-
- irc_cmd_msg (session, origin, d->d_name);
- }
-
- closedir (dir);
- }
- else if ( strstr (params[1], "get ") == params[1] )
- {
- irc_dcc_t dccid;
-
- if ( irc_dcc_sendfile (session, 0, origin, params[1] + 4, dcc_callback, &dccid) )
- irc_cmd_msg (session, origin, "Could not send this file");
- }
- else
- {
- irc_cmd_msg (session, origin, "Commands: send | list");
- }
- }
-
-
- 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> <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_privmsg = event_privmsg;
- 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);
- irc_option_set (s, LIBIRC_OPTION_STRIPNICKS);
-
- // 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 1;
- }
|