Asterisk.org: Developer Documentation (28 Oct 2009)

Wed Oct 28 15:47:54 2009

## Asterisk developer's documentation

- Main Page
- Related Pages
- Modules
- Data Structures
- Files
- Directories

# 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: 59608 $")
    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    const int reuseFlag = 1;
    00132
    00133    /* Make a UDP socket */
    00134    netsocket = socket(AF_INET, SOCK_DGRAM, IPPROTO_IP);
    00135
    00136    if (netsocket < 0) {
    00137       ast_log(LOG_ERROR, "Unable to create network socket: %s\n", strerror(errno));
    00138       return NULL;
    00139    }
    00140    if (setsockopt(netsocket, SOL_SOCKET, SO_REUSEADDR, (char *)&reuseFlag, sizeof reuseFlag) < 0)
    00141          ast_log(LOG_WARNING, "Error setting SO_REUSEADDR on sockfd '%d'\n", netsocket);
    00142    if (bind(netsocket,(struct sockaddr *)bindaddr, sizeof(struct sockaddr_in))) {
    00143       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));
    00144       close(netsocket);
    00145       return NULL;
    00146    }
    00147    if (option_verbose > 1)
    00148       ast_verbose(VERBOSE_PREFIX_2 "Using TOS bits %d\n", tos);
    00149
    00150    if (setsockopt(netsocket, IPPROTO_IP, IP_TOS, &tos, sizeof(tos)))
    00151       ast_log(LOG_WARNING, "Unable to set TOS to %d\n", tos);
    00152
    00153    ast_enable_packet_fragmentation(netsocket);
    00154
    00155    ns = malloc(sizeof(struct ast_netsock));
    00156    if (ns) {
    00157       /* Establish I/O callback for socket read */
    00158       ioref = ast_io_add(ioc, netsocket, callback, AST_IO_IN, ns);
    00159       if (!ioref) {
    00160          ast_log(LOG_WARNING, "Out of memory!\n");
    00161          close(netsocket);
    00162          free(ns);
    00163          return NULL;
    00164       }
    00165       ASTOBJ_INIT(ns);
    00166       ns->ioref = ioref;
    00167       ns->ioc = ioc;
    00168       ns->sockfd = netsocket;
    00169       ns->data = data;
    00170       memcpy(&ns->bindaddr, bindaddr, sizeof(ns->bindaddr));
    00171       ASTOBJ_CONTAINER_LINK(list, ns);
    00172    } else {
    00173       ast_log(LOG_WARNING, "Out of memory!\n");
    00174       close(netsocket);
    00175    }
    00176
    00177    return ns;
    00178 }
    00179
    00180 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)
    00181 {
    00182    struct sockaddr_in sin;
    00183    char *tmp;
    00184    char *host;
    00185    char *port;
    00186    int portno;
    00187
    00188    memset(&sin, 0, sizeof(sin));
    00189    sin.sin_family = AF_INET;
    00190    sin.sin_port = htons(defaultport);
    00191    tmp = ast_strdupa(bindinfo);
    00192    if (!tmp) {
    00193       ast_log(LOG_WARNING, "Out of memory!\n");
    00194       return NULL;
    00195    }
    00196
    00197    host = strsep(&tmp, ":");
    00198    port = tmp;
    00199
    00200    if (port && ((portno = atoi(port)) > 0))
    00201       sin.sin_port = htons(portno);
    00202
    00203    inet_aton(host, &sin.sin_addr);
    00204
    00205    return ast_netsock_bindaddr(list, ioc, &sin, tos, callback, data);
    00206 }
    00207
    00208 int ast_netsock_sockfd(const struct ast_netsock *ns)
    00209 {
    00210    return ns ? ns-> sockfd : -1;
    00211 }
    00212
    00213 const struct sockaddr_in *ast_netsock_boundaddr(const struct ast_netsock *ns)
    00214 {
    00215    return &(ns->bindaddr);
    00216 }
    00217
    00218 void *ast_netsock_data(const struct ast_netsock *ns)
    00219 {
    00220    return ns->data;
    00221 }
    00222
    00223 void ast_netsock_unref(struct ast_netsock *ns)
    00224 {
    00225    ASTOBJ_UNREF(ns, ast_netsock_destroy);
    00226 }