changing to use a separate client structure for each udp client
[radsecproxy.git] / util.c
1 /*
2  * Copyright (C) 2006-2008 Stig Venaas <venaas@uninett.no>
3  *
4  * Permission to use, copy, modify, and distribute this software for any
5  * purpose with or without fee is hereby granted, provided that the above
6  * copyright notice and this permission notice appear in all copies.
7  */
8
9 #include <sys/socket.h>
10 #include <netinet/in.h>
11 #include <netdb.h>
12 #include <stdio.h>
13 #include <stdlib.h>
14 #include <string.h>
15 #include <unistd.h>
16 #include <stdarg.h>
17 #include "debug.h"
18
19 #if 0
20 #include <errno.h>
21 void errx(char *format, ...) {
22     extern int errno;
23
24     va_list ap;
25     va_start(ap, format);
26     vfprintf(stderr, format, ap);
27     va_end(ap);
28     if (errno) {
29         fprintf(stderr, ": ");
30         perror(NULL);
31         fprintf(stderr, "errno=%d\n", errno);
32     } else
33         fprintf(stderr, "\n");
34     exit(1);
35 }
36
37 void err(char *format, ...) {
38     extern int errno;
39
40     va_list ap;
41     va_start(ap, format);
42     vfprintf(stderr, format, ap);
43     va_end(ap);
44     if (errno) {
45         fprintf(stderr, ": ");
46         perror(NULL);
47         fprintf(stderr, "errno=%d\n", errno);
48     } else
49         fprintf(stderr, "\n");
50 }
51 #endif
52
53 char *stringcopy(const char *s, int len) {
54     char *r;
55     if (!len)
56         len = strlen(s);
57     r = malloc(len + 1);
58     if (!r)
59         debug(DBG_ERR, "stringcopy: malloc failed");
60     memcpy(r, s, len);
61     r[len] = '\0';
62     return r;
63 }
64
65 void printfchars(char *prefixfmt, char *prefix, char *charfmt, char *chars, int len) {
66     int i;
67     unsigned char *s = (unsigned char *)chars;
68     if (prefix)
69         printf(prefixfmt ? prefixfmt : "%s: ", prefix);
70     for (i = 0; i < len; i++)
71         printf(charfmt ? charfmt : "%c", s[i]);
72     printf("\n");
73 }
74
75 int addr_equal(struct sockaddr *a, struct sockaddr *b) {
76     switch (a->sa_family) {
77     case AF_INET:
78         return !memcmp(&((struct sockaddr_in*)a)->sin_addr,
79                        &((struct sockaddr_in*)b)->sin_addr,
80                        sizeof(struct in_addr));
81     case AF_INET6:
82         return IN6_ARE_ADDR_EQUAL(&((struct sockaddr_in6*)a)->sin6_addr,
83                                   &((struct sockaddr_in6*)b)->sin6_addr);
84     default:
85         /* Must not reach */
86         return 0;
87     }
88 }
89
90 struct sockaddr *addr_copy(struct sockaddr *in) {
91     struct sockaddr *out = NULL;
92     
93     switch (in->sa_family) {
94     case AF_INET:
95         out = malloc(sizeof(struct sockaddr_in));
96         if (out) {
97             memset(out, 0, sizeof(struct sockaddr_in));
98             ((struct sockaddr_in *)out)->sin_addr = ((struct sockaddr_in *)in)->sin_addr;
99         }
100         break;
101     case AF_INET6:
102         out = malloc(sizeof(struct sockaddr_in6));
103         if (out) {
104             memset(out, 0, sizeof(struct sockaddr_in6));
105             ((struct sockaddr_in6 *)out)->sin6_addr = ((struct sockaddr_in6 *)in)->sin6_addr;
106         }
107         break;
108     }
109     return out;
110 }
111
112 char *addr2string(struct sockaddr *addr, socklen_t len) {
113     struct sockaddr_in6 *sa6;
114     struct sockaddr_in sa4;
115     static char addr_buf[2][INET6_ADDRSTRLEN];
116     static int i = 0;
117     i = !i;
118     if (addr->sa_family == AF_INET6) {
119         sa6 = (struct sockaddr_in6 *)addr;
120         if (IN6_IS_ADDR_V4MAPPED(&sa6->sin6_addr)) {
121             memset(&sa4, 0, sizeof(sa4));
122             sa4.sin_family = AF_INET;
123             sa4.sin_port = sa6->sin6_port;
124             memcpy(&sa4.sin_addr, &sa6->sin6_addr.s6_addr[12], 4);
125             addr = (struct sockaddr *)&sa4;
126         }
127     }
128     len = addr->sa_family == AF_INET ? sizeof(struct sockaddr_in) : sizeof(struct sockaddr_in6);
129     
130     if (getnameinfo(addr, len, addr_buf[i], sizeof(addr_buf[i]),
131                     NULL, 0, NI_NUMERICHOST)) {
132         debug(DBG_WARN, "getnameinfo failed");
133         return "getnameinfo_failed";
134     }
135     return addr_buf[i];
136 }
137
138 int connectport(int type, char *host, char *port) {
139     struct addrinfo hints, *res0, *res;
140     int s = -1;
141     
142     memset(&hints, 0, sizeof(hints));
143     hints.ai_socktype = type;
144     hints.ai_family = AF_UNSPEC;
145
146     if (getaddrinfo(host, port, &hints, &res0) != 0) {
147         debug(DBG_ERR, "connectport: can't resolve host %s port %s", host, port);
148         return -1;
149     }
150
151     for (res = res0; res; res = res->ai_next) {
152         s = socket(res->ai_family, res->ai_socktype, res->ai_protocol);
153         if (s < 0) {
154             debug(DBG_WARN, "connectport: socket failed");
155             continue;
156         }
157         if (connect(s, res->ai_addr, res->ai_addrlen) == 0)
158             break;
159         debug(DBG_WARN, "connectport: connect failed");
160         close(s);
161         s = -1;
162     }
163     freeaddrinfo(res0);
164     return s;
165 }