root/trunk/wifidog/src/util.c @ 247

Revision 247, 3.4 KB (checked in by alexcv, 9 years ago)

Thread safe version of gethostbyname()

  • Property svn:eol-style set to native
  • Property svn:keywords set to Author Date Id Revision
Line 
1/********************************************************************\
2 * This program is free software; you can redistribute it and/or    *
3 * modify it under the terms of the GNU General Public License as   *
4 * published by the Free Software Foundation; either version 2 of   *
5 * the License, or (at your option) any later version.              *
6 *                                                                  *
7 * This program is distributed in the hope that it will be useful,  *
8 * but WITHOUT ANY WARRANTY; without even the implied warranty of   *
9 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the    *
10 * GNU General Public License for more details.                     *
11 *                                                                  *
12 * You should have received a copy of the GNU General Public License*
13 * along with this program; if not, contact:                        *
14 *                                                                  *
15 * Free Software Foundation           Voice:  +1-617-542-5942       *
16 * 59 Temple Place - Suite 330        Fax:    +1-617-542-2652       *
17 * Boston, MA  02111-1307,  USA       gnu@gnu.org                   *
18 *                                                                  *
19 \********************************************************************/
20
21/*
22 * $Header$
23 */
24/**
25  @file util.c
26  @brief Misc utility functions
27  @author Copyright (C) 2004 Philippe April <papril777@yahoo.com>
28 */
29
30#define _GNU_SOURCE
31
32#include <stdio.h>
33#include <stdlib.h>
34#include <syslog.h>
35#include <errno.h>
36#include <pthread.h>
37#include <sys/wait.h>
38#include <sys/types.h>
39#include <sys/unistd.h>
40
41#include <string.h>
42#include <pthread.h>
43#include <netdb.h>
44
45#include "util.h"
46#include "conf.h"
47#include "debug.h"
48
49static pthread_mutex_t ghbn_mutex = PTHREAD_MUTEX_INITIALIZER;
50
51/** Fork a child and execute a shell command, the parent
52 * process waits for the child to return and returns the child's exit()
53 * value.
54 * @return Return code of the command
55 */
56int
57execute(char *cmd_line, int quiet)
58{
59    int pid,
60        status,
61        rc;
62
63    const char *new_argv[4];
64    new_argv[0] = "/bin/sh";
65    new_argv[1] = "-c";
66    new_argv[2] = cmd_line;
67    new_argv[3] = NULL;
68
69    if ((pid = fork()) < 0) {    /* fork a child process           */
70        debug(LOG_ERR, "fork(): %s", strerror(errno));
71        exit(1);
72    } else if (pid == 0) {    /* for the child process:         */
73        /* We don't want to see any errors if quiet flag is on */
74        if (quiet) close(2);
75        if (execvp("/bin/sh", (char *const *)new_argv) < 0) {    /* execute the command  */
76            debug(LOG_ERR, "execvp(): %s", strerror(errno));
77            exit(1);
78        }
79    } else {        /* for the parent:      */
80        do {
81            rc = wait(&status);
82        } while (rc != pid && rc != -1);    /* wait for completion  */
83    }
84
85    return (WEXITSTATUS(status));
86}
87
88struct in_addr *
89wd_gethostbyname(const char *name)
90{
91        struct hostent *he;
92        struct in_addr *h_addr, *in_addr_temp;
93
94        /* XXX Calling function is reponsible for free() */
95
96        h_addr = (struct in_addr *)malloc(sizeof(struct in_addr));
97       
98        if (h_addr == NULL)
99                return NULL;
100       
101        pthread_mutex_lock(&ghbn_mutex);
102
103        he = gethostbyname(name);
104
105        if (he == NULL) {
106                free(h_addr);
107                pthread_mutex_unlock(&ghbn_mutex);
108                return NULL;
109        }
110
111        in_addr_temp = (struct in_addr *)he->h_addr_list[0];
112        h_addr->s_addr = in_addr_temp->s_addr;
113       
114        pthread_mutex_unlock(&ghbn_mutex);
115
116        return h_addr;
117}
Note: See TracBrowser for help on using the browser.