use auto-build for cdr modules
[asterisk/asterisk.git] / io.c
diff --git a/io.c b/io.c
old mode 100755 (executable)
new mode 100644 (file)
index 91aec69..5935c9e
--- a/io.c
+++ b/io.c
@@ -1,23 +1,41 @@
 /*
- * Asterisk
- * 
- * Mark Spencer <markster@marko.net>
+ * Asterisk -- An open source telephony toolkit.
  *
- * Copyright(C) Mark Spencer
- * 
- * Distributed under the terms of the GNU General Public License (GPL) Version 2
+ * Copyright (C) 1999 - 2005, Digium, Inc.
  *
- * I/O Managment (Derived from Cheops-NG)
+ * Mark Spencer <markster@digium.com>
  *
+ * See http://www.asterisk.org for more information about
+ * the Asterisk project. Please do not directly contact
+ * any of the maintainers of this project for assistance;
+ * the project provides a web site, mailing lists and IRC
+ * channels for your use.
+ *
+ * This program is free software, distributed under the terms of
+ * the GNU General Public License Version 2. See the LICENSE file
+ * at the top of the source tree.
+ */
+
+/*! \file
+ *
+ * \brief I/O Managment (Derived from Cheops-NG)
+ *
+ * \author Mark Spencer <markster@digium.com>
  */
 
 #include <stdio.h>
-#include <sys/poll.h>
 #include <unistd.h>
 #include <stdlib.h>
 #include <termios.h>
-#include <asterisk/io.h>
-#include <asterisk/logger.h>
+#include <string.h> /* for memset */
+#include <sys/ioctl.h>
+
+#include "asterisk.h"
+
+ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
+
+#include "asterisk/io.h"
+#include "asterisk/logger.h"
 
 #ifdef DEBUG_IO
 #define DEBUG DEBUG_M
@@ -56,20 +74,35 @@ struct io_context {
        unsigned int maxfdcnt;
        /* Currently used io callback */
        int current_ioc;
+       /* Whether something has been deleted */
+       int needshrink;
 };
 
-
 struct io_context *io_context_create(void)
 {
        /* Create an I/O context */
        struct io_context *tmp;
        tmp = malloc(sizeof(struct io_context));
        if (tmp) {
-               tmp->fds = NULL;
-               tmp->ior =  NULL;
+               tmp->needshrink = 0;
                tmp->fdcnt = 0;
-               tmp->maxfdcnt = -1;
+               tmp->maxfdcnt = GROW_SHRINK_SIZE/2;
                tmp->current_ioc = -1;
+               tmp->fds = malloc((GROW_SHRINK_SIZE/2) * sizeof(struct pollfd));
+               if (!tmp->fds) {
+                       free(tmp);
+                       tmp = NULL;
+               } else {
+                       memset(tmp->fds, 0, (GROW_SHRINK_SIZE / 2) * sizeof(struct pollfd));
+                       tmp->ior =  malloc((GROW_SHRINK_SIZE / 2) * sizeof(struct io_rec));
+                       if (!tmp->ior) {
+                               free(tmp->fds);
+                               free(tmp);
+                               tmp = NULL;
+                       } else {
+                               memset(tmp->ior, 0, (GROW_SHRINK_SIZE / 2) * sizeof(struct io_rec));
+                       }
+               }
        }
        return tmp;
 }
@@ -109,7 +142,6 @@ static int io_grow(struct io_context *ioc)
                        ioc->maxfdcnt -= GROW_SHRINK_SIZE;
                        return -1;
                }
-               
        } else {
                /*
                 * Out of memory.  We return to the old size, and return a failure
@@ -127,8 +159,9 @@ int *ast_io_add(struct io_context *ioc, int fd, ast_io_cb callback, short events
         * with the given event mask, to call callback with
         * data as an argument.  Returns NULL on failure.
         */
+       int *ret;
        DEBUG(ast_log(LOG_DEBUG, "ast_io_add()\n"));
-       if (ioc->fdcnt < ioc->maxfdcnt) {
+       if (ioc->fdcnt >= ioc->maxfdcnt) {
                /* 
                 * We don't have enough space for this entry.  We need to
                 * reallocate maxfdcnt poll fd's and io_rec's, or back out now.
@@ -144,14 +177,17 @@ int *ast_io_add(struct io_context *ioc, int fd, ast_io_cb callback, short events
         */
        ioc->fds[ioc->fdcnt].fd = fd;
        ioc->fds[ioc->fdcnt].events = events;
+       ioc->fds[ioc->fdcnt].revents = 0;
        ioc->ior[ioc->fdcnt].callback = callback;
        ioc->ior[ioc->fdcnt].data = data;
        ioc->ior[ioc->fdcnt].id = (int *)malloc(sizeof(int));
        /* Bonk if we couldn't allocate an int */
        if (!ioc->ior[ioc->fdcnt].id)
                return NULL;
-       *ioc->ior[ioc->fdcnt].id = ioc->fdcnt;
-       return ioc->ior[ioc->fdcnt++].id;
+       *(ioc->ior[ioc->fdcnt].id) = ioc->fdcnt;
+       ret = ioc->ior[ioc->fdcnt].id;
+       ioc->fdcnt++;
+       return ret;
 }
 
 int *ast_io_change(struct io_context *ioc, int *id, int fd, ast_io_cb callback, short events, void *data)
@@ -166,44 +202,59 @@ int *ast_io_change(struct io_context *ioc, int *id, int fd, ast_io_cb callback,
                if (data)
                        ioc->ior[*id].data = data;
                return id;
-       } else return NULL;
+       }
+       return NULL;
 }
 
-static int io_shrink(struct io_context *ioc, int which)
+static int io_shrink(struct io_context *ioc)
 {
+       int getfrom;
+       int putto = 0;
        /* 
         * Bring the fields from the very last entry to cover over
         * the entry we are removing, then decrease the size of the 
         * arrays by one.
         */
-       ioc->fdcnt--;
-
-       /* Free the int */
-       free(ioc->ior[which].id);
-       
-       /* If we're not deleting the last one, move the last one to
-          the current position */
-       if (which != ioc->fdcnt) {
-               ioc->fds[which] = ioc->fds[ioc->fdcnt];
-               ioc->ior[which] = ioc->ior[ioc->fdcnt];
-               *ioc->ior[which].id = which;
+       for (getfrom = 0; getfrom < ioc->fdcnt; getfrom++) {
+               if (ioc->ior[getfrom].id) {
+                       /* In use, save it */
+                       if (getfrom != putto) {
+                               ioc->fds[putto] = ioc->fds[getfrom];
+                               ioc->ior[putto] = ioc->ior[getfrom];
+                               *(ioc->ior[putto].id) = putto;
+                       }
+                       putto++;
+               }
        }
+       ioc->fdcnt = putto;
+       ioc->needshrink = 0;
        /* FIXME: We should free some memory if we have lots of unused
           io structs */
        return 0;
 }
 
-int ast_io_remove(struct io_context *ioc, int *id)
+int ast_io_remove(struct io_context *ioc, int *_id)
 {
-       if (ioc->current_ioc == *id) {
-               ast_log(LOG_NOTICE, "Callback for %d tried to remove itself\n", *id);
-       } else
+       int x;
+       if (!_id) {
+               ast_log(LOG_WARNING, "Asked to remove NULL?\n");
+               return -1;
+       }
+       for (x = 0; x < ioc->fdcnt; x++) {
+               if (ioc->ior[x].id == _id) {
+                       /* Free the int immediately and set to NULL so we know it's unused now */
+                       free(ioc->ior[x].id);
+                       ioc->ior[x].id = NULL;
+                       ioc->fds[x].events = 0;
+                       ioc->fds[x].revents = 0;
+                       ioc->needshrink = 1;
+                       if (!ioc->current_ioc)
+                               io_shrink(ioc);
+                       return 0;
+               }
+       }
        
-       if (*id < ioc->fdcnt) {
-               return io_shrink(ioc, *id);
-       } else 
-               ast_log(LOG_NOTICE, "Unable to remove unknown id %d\n", *id);
-
+       ast_log(LOG_NOTICE, "Unable to remove unknown id %p\n", _id);
        return -1;
 }
 
@@ -216,23 +267,31 @@ int ast_io_wait(struct io_context *ioc, int howlong)
         */
        int res;
        int x;
+       int origcnt;
        DEBUG(ast_log(LOG_DEBUG, "ast_io_wait()\n"));
        res = poll(ioc->fds, ioc->fdcnt, howlong);
        if (res > 0) {
                /*
                 * At least one event
                 */
-               for(x=0;x<ioc->fdcnt;x++) {
-                       if (ioc->fds[x].revents) {
+               origcnt = ioc->fdcnt;
+               for(x = 0; x < origcnt; x++) {
+                       /* Yes, it is possible for an entry to be deleted and still have an
+                          event waiting if it occurs after the original calling id */
+                       if (ioc->fds[x].revents && ioc->ior[x].id) {
                                /* There's an event waiting */
                                ioc->current_ioc = *ioc->ior[x].id;
-                               if (!ioc->ior[x].callback(ioc->ior[x].id, ioc->fds[x].fd, ioc->fds[x].revents, ioc->ior[x].data)) {
-                                       /* Time to delete them since they returned a 0 */
-                                       io_shrink(ioc, x);
+                               if (ioc->ior[x].callback) {
+                                       if (!ioc->ior[x].callback(ioc->ior[x].id, ioc->fds[x].fd, ioc->fds[x].revents, ioc->ior[x].data)) {
+                                               /* Time to delete them since they returned a 0 */
+                                               ast_io_remove(ioc, ioc->ior[x].id);
+                                       }
                                }
                                ioc->current_ioc = -1;
                        }
                }
+               if (ioc->needshrink)
+                       io_shrink(ioc);
        }
        return res;
 }
@@ -248,7 +307,7 @@ void ast_io_dump(struct io_context *ioc)
        ast_log(LOG_DEBUG, "================================================\n");
        ast_log(LOG_DEBUG, "| ID    FD     Callback    Data        Events  |\n");
        ast_log(LOG_DEBUG, "+------+------+-----------+-----------+--------+\n");
-       for (x=0;x<ioc->fdcnt;x++) {
+       for (x = 0; x < ioc->fdcnt; x++) {
                ast_log(LOG_DEBUG, "| %.4d | %.4d | %p | %p | %.6x |\n", 
                                *ioc->ior[x].id,
                                ioc->fds[x].fd,
@@ -297,3 +356,22 @@ int ast_restore_tty(int fd, int oldstate)
        return 0;
 }
 
+int ast_get_termcols(int fd)
+{
+       struct winsize win;
+       int cols = 0;
+
+       if (!isatty(fd))
+               return -1;
+
+       if ( ioctl(fd, TIOCGWINSZ, &win) != -1 ) {
+               if ( !cols && win.ws_col > 0 )
+                       cols = (int) win.ws_col;
+       } else {
+               /* assume 80 characters if the ioctl fails for some reason */
+               cols = 80;
+       }
+
+       return cols;
+}
+