Sun, 03 Jul 2011 13:13:36 -0700
context: Support ECDH cipher suites, where applicable
0 | 1 | /*=========================================================================*\ |
2 | * LuaSocket 2.0.2 | |
3 | * Copyright (C) 2004-2007 Diego Nehab | |
4 | * | |
5 | * Socket compatibilization module for Unix | |
6 | * | |
7 | * The code is now interrupt-safe. | |
8 | * The penalty of calling select to avoid busy-wait is only paid when | |
9 | * the I/O call fail in the first place. | |
10 | * | |
11 | * RCS ID: $Id: usocket.c,v 1.38 2007/10/13 23:55:20 diego Exp $ | |
12 | \*=========================================================================*/ | |
13 | #include <errno.h> | |
14 | #include <sys/types.h> | |
15 | #include <sys/time.h> | |
16 | #include <sys/socket.h> | |
17 | #include <string.h> | |
18 | #include <signal.h> | |
19 | #include <unistd.h> | |
20 | #include <fcntl.h> | |
21 | ||
22 | #include "socket.h" | |
23 | #include "usocket.h" | |
24 | ||
25 | /*-------------------------------------------------------------------------*\ | |
26 | * Wait for readable/writable/connected socket with timeout | |
27 | \*-------------------------------------------------------------------------*/ | |
28 | #ifdef SOCKET_POLL | |
29 | int socket_waitfd(p_socket ps, int sw, p_timeout tm) { | |
30 | int ret; | |
31 | struct pollfd pfd; | |
32 | pfd.fd = *ps; | |
33 | pfd.events = sw; | |
34 | pfd.revents = 0; | |
35 | if (timeout_iszero(tm)) return IO_TIMEOUT; /* optimize timeout == 0 case */ | |
36 | do { | |
37 | int t = (int)(timeout_getretry(tm)*1e3); | |
38 | ret = poll(&pfd, 1, t >= 0? t: -1); | |
39 | } while (ret == -1 && errno == EINTR); | |
40 | if (ret == -1) return errno; | |
41 | if (ret == 0) return IO_TIMEOUT; | |
42 | if (sw == WAITFD_C && (pfd.revents & (POLLIN|POLLERR))) return IO_CLOSED; | |
43 | return IO_DONE; | |
44 | } | |
45 | #else | |
46 | int socket_waitfd(p_socket ps, int sw, p_timeout tm) { | |
47 | int ret; | |
48 | fd_set rfds, wfds, *rp, *wp; | |
49 | struct timeval tv, *tp; | |
50 | double t; | |
51 | if (timeout_iszero(tm)) return IO_TIMEOUT; /* optimize timeout == 0 case */ | |
52 | do { | |
53 | /* must set bits within loop, because select may have modifed them */ | |
54 | rp = wp = NULL; | |
55 | if (sw & WAITFD_R) { FD_ZERO(&rfds); FD_SET(*ps, &rfds); rp = &rfds; } | |
56 | if (sw & WAITFD_W) { FD_ZERO(&wfds); FD_SET(*ps, &wfds); wp = &wfds; } | |
57 | t = timeout_getretry(tm); | |
58 | tp = NULL; | |
59 | if (t >= 0.0) { | |
60 | tv.tv_sec = (int)t; | |
61 | tv.tv_usec = (int)((t-tv.tv_sec)*1.0e6); | |
62 | tp = &tv; | |
63 | } | |
64 | ret = select(*ps+1, rp, wp, NULL, tp); | |
65 | } while (ret == -1 && errno == EINTR); | |
66 | if (ret == -1) return errno; | |
67 | if (ret == 0) return IO_TIMEOUT; | |
68 | if (sw == WAITFD_C && FD_ISSET(*ps, &rfds)) return IO_CLOSED; | |
69 | return IO_DONE; | |
70 | } | |
71 | #endif | |
72 | ||
73 | ||
74 | /*-------------------------------------------------------------------------*\ | |
75 | * Initializes module | |
76 | \*-------------------------------------------------------------------------*/ | |
77 | int socket_open(void) { | |
78 | /* instals a handler to ignore sigpipe or it will crash us */ | |
79 | signal(SIGPIPE, SIG_IGN); | |
80 | return 1; | |
81 | } | |
82 | ||
83 | /*-------------------------------------------------------------------------*\ | |
84 | * Close module | |
85 | \*-------------------------------------------------------------------------*/ | |
86 | int socket_close(void) { | |
87 | return 1; | |
88 | } | |
89 | ||
90 | /*-------------------------------------------------------------------------*\ | |
91 | * Close and inutilize socket | |
92 | \*-------------------------------------------------------------------------*/ | |
93 | void socket_destroy(p_socket ps) { | |
94 | if (*ps != SOCKET_INVALID) { | |
95 | socket_setblocking(ps); | |
96 | close(*ps); | |
97 | *ps = SOCKET_INVALID; | |
98 | } | |
99 | } | |
100 | ||
101 | /*-------------------------------------------------------------------------*\ | |
102 | * Put socket into blocking mode | |
103 | \*-------------------------------------------------------------------------*/ | |
104 | void socket_setblocking(p_socket ps) { | |
105 | int flags = fcntl(*ps, F_GETFL, 0); | |
106 | flags &= (~(O_NONBLOCK)); | |
107 | fcntl(*ps, F_SETFL, flags); | |
108 | } | |
109 | ||
110 | /*-------------------------------------------------------------------------*\ | |
111 | * Put socket into non-blocking mode | |
112 | \*-------------------------------------------------------------------------*/ | |
113 | void socket_setnonblocking(p_socket ps) { | |
114 | int flags = fcntl(*ps, F_GETFL, 0); | |
115 | flags |= O_NONBLOCK; | |
116 | fcntl(*ps, F_SETFL, flags); | |
117 | } | |
118 | ||
119 | /*-------------------------------------------------------------------------*\ | |
120 | * Error translation functions | |
121 | * Make sure important error messages are standard | |
122 | \*-------------------------------------------------------------------------*/ | |
123 | const char *socket_strerror(int err) { | |
124 | if (err <= 0) return io_strerror(err); | |
125 | switch (err) { | |
126 | case EADDRINUSE: return "address already in use"; | |
127 | case EISCONN: return "already connected"; | |
128 | case EACCES: return "permission denied"; | |
129 | case ECONNREFUSED: return "connection refused"; | |
130 | case ECONNABORTED: return "closed"; | |
131 | case ECONNRESET: return "closed"; | |
132 | case EPIPE: return "closed"; | |
133 | case ETIMEDOUT: return "timeout"; | |
134 | default: return strerror(errno); | |
135 | } | |
136 | } | |
137 | ||
138 | /*-------------------------------------------------------------------------*\ | |
139 | * Underline error code. | |
140 | \*-------------------------------------------------------------------------*/ | |
141 | int socket_error() | |
142 | { | |
143 | return errno; | |
144 | } | |
145 |