Fri Sep 29 11:12:30 2006

Asterisk developer's documentation


netsock.c

Go to the documentation of this file.
00001 /*
00002  * Asterisk -- An open source telephony toolkit.
00003  *
00004  * Copyright (C) 1999 - 2005, Digium, Inc.
00005  *
00006  * Kevin P. Fleming <kpfleming@digium.com>
00007  * Mark Spencer <markster@digium.com>
00008  *
00009  * See http://www.asterisk.org for more information about
00010  * the Asterisk project. Please do not directly contact
00011  * any of the maintainers of this project for assistance;
00012  * the project provides a web site, mailing lists and IRC
00013  * channels for your use.
00014  *
00015  * This program is free software, distributed under the terms of
00016  * the GNU General Public License Version 2. See the LICENSE file
00017  * at the top of the source tree.
00018  */
00019 
00020 /*! \file
00021  *
00022  * \brief Network socket handling
00023  * 
00024  */
00025 
00026 #include <stdio.h>
00027 #include <stdlib.h>
00028 #include <string.h>
00029 #include <sys/time.h>
00030 #include <signal.h>
00031 #include <errno.h>
00032 #include <unistd.h>
00033 #include <netinet/in.h>
00034 #include <arpa/inet.h>
00035 #include <sys/socket.h>
00036 #include <netdb.h>
00037 #include <net/if.h>
00038 #include <netinet/in_systm.h>
00039 #include <netinet/ip.h>
00040 #include <sys/ioctl.h>
00041 
00042 #if defined(__OpenBSD__) || defined(__NetBSD__) || defined(__FreeBSD__)
00043 #include <fcntl.h>
00044 #include <net/route.h>
00045 #endif
00046 
00047 #if defined (SOLARIS)
00048 #include <sys/sockio.h>
00049 #endif
00050 
00051 #include "asterisk.h"
00052 
00053 ASTERISK_FILE_VERSION(__FILE__, "$Revision: 7221 $")
00054 
00055 #include "asterisk/netsock.h"
00056 #include "asterisk/logger.h"
00057 #include "asterisk/channel.h"
00058 #include "asterisk/options.h"
00059 #include "asterisk/utils.h"
00060 #include "asterisk/lock.h"
00061 #include "asterisk/srv.h"
00062 
00063 struct ast_netsock {
00064    ASTOBJ_COMPONENTS(struct ast_netsock);
00065    struct sockaddr_in bindaddr;
00066    int sockfd;
00067    int *ioref;
00068    struct io_context *ioc;
00069    void *data;
00070 };
00071 
00072 struct ast_netsock_list {
00073    ASTOBJ_CONTAINER_COMPONENTS(struct ast_netsock);
00074    struct io_context *ioc;
00075 };
00076 
00077 static void ast_netsock_destroy(struct ast_netsock *netsock)
00078 {
00079    ast_io_remove(netsock->ioc, netsock->ioref);
00080    close(netsock->sockfd);
00081    free(netsock);
00082 }
00083 
00084 struct ast_netsock_list *ast_netsock_list_alloc(void)
00085 {
00086    struct ast_netsock_list *res;
00087 
00088    res = calloc(1, sizeof(*res));
00089 
00090    return res;
00091 }
00092 
00093 int ast_netsock_init(struct ast_netsock_list *list)
00094 {
00095    memset(list, 0, sizeof(*list));
00096    ASTOBJ_CONTAINER_INIT(list);
00097 
00098    return 0;
00099 }
00100 
00101 int ast_netsock_release(struct ast_netsock_list *list)
00102 {
00103    ASTOBJ_CONTAINER_DESTROYALL(list, ast_netsock_destroy);
00104    ASTOBJ_CONTAINER_DESTROY(list);
00105 
00106    return 0;
00107 }
00108 
00109 struct ast_netsock *ast_netsock_find(struct ast_netsock_list *list,
00110                  struct sockaddr_in *sa)
00111 {
00112    struct ast_netsock *sock = NULL;
00113 
00114    ASTOBJ_CONTAINER_TRAVERSE(list, !sock, {
00115       ASTOBJ_RDLOCK(iterator);
00116       if (!inaddrcmp(&iterator->bindaddr, sa))
00117          sock = iterator;
00118       ASTOBJ_UNLOCK(iterator);
00119    });
00120 
00121    return sock;
00122 }
00123 
00124 struct ast_netsock *ast_netsock_bindaddr(struct ast_netsock_list *list, struct io_context *ioc, struct sockaddr_in *bindaddr, int tos, ast_io_cb callback, void *data)
00125 {
00126    int netsocket = -1;
00127    int *ioref;
00128    char iabuf[INET_ADDRSTRLEN];
00129    
00130    struct ast_netsock *ns;
00131    
00132    /* Make a UDP socket */
00133    netsocket = socket(AF_INET, SOCK_DGRAM, IPPROTO_IP);
00134    
00135    if (netsocket < 0) {
00136       ast_log(LOG_ERROR, "Unable to create network socket: %s\n", strerror(errno));
00137       return NULL;
00138    }
00139    if (bind(netsocket,(struct sockaddr *)bindaddr, sizeof(struct sockaddr_in))) {
00140       ast_log(LOG_ERROR, "Unable to bind to %s port %d: %s\n", ast_inet_ntoa(iabuf, sizeof(iabuf), bindaddr->sin_addr), ntohs(bindaddr->sin_port), strerror(errno));
00141       close(netsocket);
00142       return NULL;
00143    }
00144    if (option_verbose > 1)
00145       ast_verbose(VERBOSE_PREFIX_2 "Using TOS bits %d\n", tos);
00146 
00147    if (setsockopt(netsocket, IPPROTO_IP, IP_TOS, &tos, sizeof(tos))) 
00148       ast_log(LOG_WARNING, "Unable to set TOS to %d\n", tos);
00149 
00150    ns = malloc(sizeof(struct ast_netsock));
00151    if (ns) {
00152       /* Establish I/O callback for socket read */
00153       ioref = ast_io_add(ioc, netsocket, callback, AST_IO_IN, ns);
00154       if (!ioref) {
00155          ast_log(LOG_WARNING, "Out of memory!\n");
00156          close(netsocket);
00157          free(ns);
00158          return NULL;
00159       }  
00160       ASTOBJ_INIT(ns);
00161       ns->ioref = ioref;
00162       ns->ioc = ioc;
00163       ns->sockfd = netsocket;
00164       ns->data = data;
00165       memcpy(&ns->bindaddr, bindaddr, sizeof(ns->bindaddr));
00166       ASTOBJ_CONTAINER_LINK(list, ns);
00167    } else {
00168       ast_log(LOG_WARNING, "Out of memory!\n");
00169       close(netsocket);
00170    }
00171 
00172    return ns;
00173 }
00174 
00175 struct ast_netsock *ast_netsock_bind(struct ast_netsock_list *list, struct io_context *ioc, const char *bindinfo, int defaultport, int tos, ast_io_cb callback, void *data)
00176 {
00177    struct sockaddr_in sin;
00178    char *tmp;
00179    char *host;
00180    char *port;
00181    int portno;
00182 
00183    memset(&sin, 0, sizeof(sin));
00184    sin.sin_family = AF_INET;
00185    sin.sin_port = htons(defaultport);
00186    tmp = ast_strdupa(bindinfo);
00187    if (!tmp) {
00188       ast_log(LOG_WARNING, "Out of memory!\n");
00189       return NULL;
00190    }
00191 
00192    host = strsep(&tmp, ":");
00193    port = tmp;
00194 
00195    if (port && ((portno = atoi(port)) > 0))
00196       sin.sin_port = htons(portno);
00197 
00198    inet_aton(host, &sin.sin_addr);
00199 
00200    return ast_netsock_bindaddr(list, ioc, &sin, tos, callback, data);
00201 }
00202 
00203 int ast_netsock_sockfd(const struct ast_netsock *ns)
00204 {
00205    return ns ? ns-> sockfd : -1;
00206 }
00207 
00208 const struct sockaddr_in *ast_netsock_boundaddr(const struct ast_netsock *ns)
00209 {
00210    return &(ns->bindaddr);
00211 }
00212 
00213 void *ast_netsock_data(const struct ast_netsock *ns)
00214 {
00215    return ns->data;
00216 }
00217 
00218 void ast_netsock_unref(struct ast_netsock *ns)
00219 {
00220    ASTOBJ_UNREF(ns, ast_netsock_destroy);
00221 }

Generated on Fri Sep 29 11:12:30 2006 for Asterisk - the Open Source PBX by  doxygen 1.4.7