Changeset 225

Timestamp:

03/18/08 08:52:00 (16 years ago)

Author:

niltonneto

Message:

 

Location:

trunk/instant_messenger/webjabber

Files:

• webjabber.c (modified) (17 diffs)
• webjabber.cfg (added)

trunk/instant_messenger/webjabber/webjabber.c

@@ -27 +27 @@
 // Global variables
 char g_program_name[100];       // Initialised from argv[0] in ParseArgs()
+struct allowed_ip * allowed_ips = (struct allowed_ip *)NULL;
 int quantity_of_fd = 0;         // quantity of file descriptor
 
+static struct connections * connections = (struct connections *)NULL;
 pthread_mutex_t mutexsum;
 
 // structs
+struct allowed_ip
+{
+        char * ip;
+        struct allowed_ip * next;
+};
+
 struct accept
 {
@@ -47 +55 @@
         int * jabber_fd,
                 * read;
+        time_t last_access;
 };
 
@@ -65 +74 @@
 
 void recursive (int, struct in_addr, unsigned short);
+int getPermission(char *);
 void ParseArgs (int, char **, struct in_addr *, unsigned short *, struct in_addr *, unsigned short *);
 void Initialise (void);
@@ -71 +81 @@
 void Daemonise (void);
 void MainLoop (int, struct in_addr, unsigned short);
+void ClearList(void);
 struct accept * AcceptClientConnection (int);
 void VerifyClient(void *);
 struct client * Handshake (struct accept *, struct read_write *);
-struct client * LookFor (char *, char *);
+void * withConnections(void * (*func)(), ...);
+//struct client * LookFor (char *, char *);
+//void * LookFor (struct connections *, char *, char *);
+void * LookFor (char *, char *);
 void Reading (void *);
 void Writing (void *);
@@ -99 +113 @@
         int listen_fd;
 
+        //pthread_t clear_list_thread;
+
         ParseArgs (argc, argv, &remote_addr, &remote_port, &local_addr, &local_port);
 
@@ -111 +127 @@
         //Daemonise ();
 
+        //pthread_mutex_init(&mutexsum, NULL);
+        //pthread_create(&clear_list_thread, NULL, (void *) &ClearList, (void *)NULL );
+
         //MainLoop (listen_fd, remote_addr, remote_port); // never returns
         recursive(listen_fd, remote_addr, remote_port);
+
+        //pthread_mutex_destroy(&mutexsum);
 
         exit (EXIT_SUCCESS);
@@ -189 +210 @@
 }
 
+int getPermission(char * ip)
+{
+        struct allowed_ip * allow_ip = allowed_ips;
+        int allow = 0;
+
+        while ( !allow && allow_ip->next != (struct allowed_ip *)NULL )
+        {
+                //printf("strcmp(%s, %s)\n\n", ip, allow_ip->ip);
+                if ( strcmp(ip, allow_ip->ip) == 0 )
+                        allow = 1;
+                else
+                        allow_ip = allow_ip->next;
+        }
+
+        return allow;
+}
+
 // Initialise()
 // Setup syslog, signal handlers, and other intialisation.
 void Initialise (void)
 {
+        FILE * allowed_ips_fd;
+        struct allowed_ip * allow_ip;
+
+        if ( !(allowed_ips_fd = fopen("webjabber.cfg", "r")) )
+                syslog (LOG_ERR, "Impossivel abrir o arquivo 'webjabber.cfg'"), exit (EXIT_FAILURE);
+
+        while ( !feof(allowed_ips_fd) )
+        {
+                char * line = (char *) malloc(16 * sizeof(char));
+
+                bzero(line, 16);
+                fgets(line, 16, allowed_ips_fd);
+
+                if ( strlen(line) )
+                {
+                        if ( line[strlen(line) - 1] == '\n' )
+                                line[strlen(line) - 1] = '\0';
+
+                        if ( allowed_ips == (struct allowed_ip *)NULL )
+                                allow_ip = allowed_ips = (struct allowed_ip *) malloc(sizeof(struct allowed_ip));
+                        else
+                        {
+                                allow_ip = allowed_ips;
+                                while ( allow_ip->next != (struct allowed_ip *)NULL )
+                                        allow_ip = allow_ip->next;
+                                allow_ip->next = (struct allowed_ip *) malloc(sizeof(struct allowed_ip));
+                                allow_ip = allow_ip->next;
+                        }
+                        allow_ip->next = (struct allowed_ip *)NULL;
+                        allow_ip->ip = line;
+                }
+        }
+
+        char * initial_path = (char *) malloc(128 * sizeof(char));
+        getcwd(initial_path, 128);
+
         openlog (g_program_name, LOG_PID, LOG_USER);
-        syslog (LOG_INFO, "%s started", g_program_name);
+        syslog (LOG_INFO, "%s started in path [#%d] %s", g_program_name, strlen(initial_path), initial_path);
 
         chdir ("/"); // Change working directory to the root.
@@ -329 +403 @@
 void MainLoop (int listen_fd, struct in_addr rem_addr, unsigned short rem_port)
 {
-        pthread_mutex_init(&mutexsum, NULL);
         //pid_t helper_pid;
+        pthread_t clear_list_thread;
 
         syslog (LOG_INFO, "MainLoop :: listen_fd(fd%d)", listen_fd);
         //signal (SIGCHLD, SIG_IGN);
 
+        pthread_mutex_init(&mutexsum, NULL);
+        pthread_create(&clear_list_thread, NULL, (void *) &ClearList, (void *)NULL );
         while ( quantity_of_fd < 50 )
         {
@@ -347 +423 @@
                 pthread_create(&client->accept_thread, NULL, (void *) &VerifyClient, (void *) client);
         }
-
         pthread_mutex_destroy(&mutexsum);
+}
+
+void ClearList(void)
+{
+        for ( ; ; )
+        {
+                pthread_mutex_lock(&mutexsum);
+                struct connections * connection = connections;
+                while ( connection != (struct connections *)NULL )
+                {
+                        if ( *(connection->client->jabber_fd) != (int)NULL && time(NULL) - connection->client->last_access > 70 )
+                        {
+                                shutdown(*(connection->client->jabber_fd), SHUT_RDWR);
+                                close(*(connection->client->jabber_fd));
+                                *(connection->client->jabber_fd) = (int)NULL;
+                        }
+                        connection = connection->next;
+                }
+                pthread_mutex_unlock(&mutexsum);
+                sleep(5);
+        }
 }
 
@@ -425 +521 @@
         //printf("\nNew connection [#%d]: %s\n", *(accept->fd), ip);
 
-        if ( strcmp(ip, "10.15.20.42") == 0 )
+        if ( getPermission(ip) )
                 if ( (client = Handshake(accept, read_write)) == (struct client *)NULL )
                 {
@@ -531 +627 @@
         //printf("  fd[%2d]: client\n", *(accept->fd));
 
-        if ( (client = LookFor (user, pass)) != (struct client *)NULL )
+        //if ( (client = LookFor (user, pass)) != (struct client *)NULL )
+        if ( (client = (struct client *) withConnections(LookFor, user, pass)) != (struct client *)NULL )
         {
                 //puts("\n\n:::::::::::::::: INFO #0 :::::::::::::::::::::\n");
@@ -573 +670 @@
 }
 
-struct client * LookFor (char * user, char * pass)
+void * withConnections(void * (*func)(), ...)
+{
+        //static struct connections * connections = (struct connections *)NULL;
+        if ( (*func) == LookFor )
+        {
+                va_list ap;
+                char * user,
+                         * pass;
+                void * _return;
+
+                va_start(ap, (*func));
+                user = va_arg(ap, char *);
+                pass = va_arg(ap, char *);
+                va_end(ap);
+
+                pthread_mutex_lock(&mutexsum);
+                _return = (*func)(user, pass);
+                pthread_mutex_unlock(&mutexsum);
+                return _return;
+        }
+        else
+        {
+        }
+}
+
+//struct client * LookFor (char * user, char * pass)
+//void * LookFor (struct connections * connections, char * user, char * pass)
+void * LookFor (char * user, char * pass)
 {
         //puts("\n\n:::::::::::::::: LOOKFOR :::::::::::::::::::::\n");
-        static struct connections * connections = (struct connections *)NULL;
+        //static struct connections * connections = (struct connections *)NULL;
 
         //pthread_mutex_lock(&mutexsum);
@@ -597 +721 @@
 
                 *(connections->client->jabber_fd) = (int)NULL;
+                connections->client->last_access = time(NULL);
+
                 connections->next = (struct connections *)NULL;
 
@@ -615 +741 @@
                                         //pthread_mutex_unlock(&mutexsum);
                                         //puts(":::::::::::::::: ACHOU #1 :::::::::::::::::::::");
+                                        connection->client->last_access = time(NULL);
                                         return connection->client;
                                 }
@@ -645 +772 @@
 
                 *(connection->client->jabber_fd) = (int)NULL;
+                connection->client->last_access = time(NULL);
+
                 connection->next = (struct connections *)NULL;
                 //pthread_mutex_unlock(&mutexsum);
@@ -687 +816 @@
                 {
                         // descomentar para ver oq esta passado de informacao
-                        //printf("write(#%d/client)[%d]:\n{\n%s.......\n}\n", client_fd, strlen(buf_2_client), buf_2_client);
+                        //printf("(jabber/client[#%d])[%d]:\n{\n%s.......\n}\n", client_fd, strlen(buf_2_client), buf_2_client);
                         for ( i = 0; i < bytes_rcvd_jabber; i += bytes_sent_client )
                         {
@@ -701 +830 @@
                 {
                         // descomentar para ver oq esta passado de informacao
-                        //printf("write(#%d/jabber)[%d]:\n{\n%s.......\n}\n", jabber_fd, strlen(buf_2_jabber), buf_2_jabber);
+                        //printf("(client/jabber[#%d])[%d]:\n{\n%s.......\n}\n", jabber_fd, strlen(buf_2_jabber), buf_2_jabber);
                         for ( i = 0; i < bytes_rcvd_client; i += bytes_sent_jabber )
                         {