db_sockets.c 42 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480
  1. /* -*- mode:c; c-file-style:"k&r"; c-basic-offset: 4; tab-width:4; indent-tabs-mode:nil; mode:auto-fill; fill-column:78; -*- */
  2. /* vim: set ts=4 sw=4 et tw=78 fo=cqt wm=0: */
  3. /* Copyright (C) 2014 Stony Brook University
  4. This file is part of Graphene Library OS.
  5. Graphene Library OS is free software: you can redistribute it and/or
  6. modify it under the terms of the GNU Lesser General Public License
  7. as published by the Free Software Foundation, either version 3 of the
  8. License, or (at your option) any later version.
  9. Graphene Library OS is distributed in the hope that it will be useful,
  10. but WITHOUT ANY WARRANTY; without even the implied warranty of
  11. MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  12. GNU Lesser General Public License for more details.
  13. You should have received a copy of the GNU Lesser General Public License
  14. along with this program. If not, see <http://www.gnu.org/licenses/>. */
  15. /*
  16. * db_socket.c
  17. *
  18. * This file contains operands for streams with URIs that start with
  19. * "tcp:", "tcp.srv:", "udp:", "udp.srv:".
  20. */
  21. #include "pal_defs.h"
  22. #include "pal_freebsd_defs.h"
  23. #include "pal.h"
  24. #include "pal_internal.h"
  25. #include "pal_freebsd.h"
  26. #include "pal_debug.h"
  27. #include "pal_security.h"
  28. #include "pal_error.h"
  29. #include "api.h"
  30. #include <sys/types.h>
  31. #include <poll.h>
  32. typedef __kernel_pid_t pid_t;
  33. #include <fcntl.h>
  34. #include <sys/socket.h>
  35. #include <netinet/in.h>
  36. #include <arpa/inet.h>
  37. #include <netinet/tcp.h>
  38. #include <errno.h>
  39. #include <sys/filio.h>
  40. /* FreeBSD doesn't have addr_check_any. */
  41. #undef ALLOW_BIND_ANY
  42. #define ALLOW_BIND_ANY 1
  43. /* 96 bytes is the minimal size of buffer to store a IPv4/IPv6
  44. address */
  45. #define PAL_SOCKADDR_SIZE 96
  46. typedef uint16_t __be16;
  47. #define SOL_TCP 6
  48. #define TCP_CORK TCP_NOPUSH
  49. static inline int addr_size (struct sockaddr * addr)
  50. {
  51. switch (addr->sa_family) {
  52. case AF_INET:
  53. return sizeof(struct sockaddr_in);
  54. case AF_INET6:
  55. return sizeof(struct sockaddr_in6);
  56. default:
  57. return 0;
  58. }
  59. }
  60. /* parsing the string of uri, and fill in the socket address structure.
  61. the latest pointer of uri, length of socket address are returned. */
  62. static int inet_parse_uri (char ** uri, struct sockaddr * addr, int * addrlen)
  63. {
  64. char * tmp = *uri, * end;
  65. char * addr_str = NULL, * port_str;
  66. int af;
  67. void * addr_buf;
  68. int addr_len;
  69. __be16 * port_buf;
  70. int slen;
  71. if (tmp[0] == '[') {
  72. /* for IPv6, the address will be in the form of
  73. "[xx:xx:xx:xx:xx:xx:xx:xx]:port". */
  74. struct sockaddr_in6 * addr_in6 = (struct sockaddr_in6 *) addr;
  75. slen = sizeof(struct sockaddr_in6);
  76. memset(addr, 0, slen);
  77. end = strchr(tmp + 1, ']');
  78. if (!end || *(end + 1) != ':')
  79. goto inval;
  80. addr_str = tmp + 1;
  81. addr_len = end - tmp + 1;
  82. port_str = end + 2;
  83. for (end = port_str ; *end >= '0' && *end <= '9' ; end++);
  84. addr_in6->sin6_family = af = AF_INET6;
  85. addr_buf = &addr_in6->sin6_addr.s6_addr;
  86. port_buf = &addr_in6->sin6_port;
  87. } else {
  88. /* for IP, the address will be in the form of "x.x.x.x:port". */
  89. struct sockaddr_in * addr_in = (struct sockaddr_in *) addr;
  90. slen = sizeof(struct sockaddr_in);
  91. memset(addr, 0, slen);
  92. end = strchr(tmp, ':');
  93. if (!end)
  94. goto inval;
  95. addr_str = tmp;
  96. addr_len = end - tmp;
  97. port_str = end + 1;
  98. for (end = port_str ; *end >= '0' && *end <= '9' ; end++);
  99. addr_in->sin_family = af = AF_INET;
  100. addr_buf = &addr_in->sin_addr.s_addr;
  101. port_buf = &addr_in->sin_port;
  102. }
  103. if (af == AF_INET) {
  104. if (inet_pton4(addr_str, addr_len, addr_buf) < 0)
  105. goto inval;
  106. } else {
  107. if (inet_pton6(addr_str, addr_len, addr_buf) < 0)
  108. goto inval;
  109. }
  110. *port_buf = __htons(atoi(port_str));
  111. *uri = *end ? end + 1 : NULL;
  112. if (addrlen)
  113. *addrlen = slen;
  114. return 0;
  115. inval:
  116. return -PAL_ERROR_INVAL;
  117. }
  118. /* create the string of uri from the given socket address */
  119. static int inet_create_uri (char * uri, int count, struct sockaddr * addr,
  120. int addrlen)
  121. {
  122. int len = 0;
  123. if (addr->sa_family == AF_INET) {
  124. if (addrlen != sizeof(struct sockaddr_in))
  125. return PAL_ERROR_INVAL;
  126. struct sockaddr_in * addr_in = (struct sockaddr_in *) addr;
  127. char * addr = (char *) &addr_in->sin_addr.s_addr;
  128. /* for IP, the address will be in the form of "x.x.x.x:port". */
  129. len = snprintf(uri, count, "%u.%u.%u.%u:%u",
  130. (unsigned char) addr[0],
  131. (unsigned char) addr[1],
  132. (unsigned char) addr[2],
  133. (unsigned char) addr[3],
  134. __ntohs(addr_in->sin_port));
  135. } else if (addr->sa_family == AF_INET6) {
  136. if (addrlen != sizeof(struct sockaddr_in6))
  137. return PAL_ERROR_INVAL;
  138. struct sockaddr_in6 * addr_in6 = (struct sockaddr_in6 *) addr;
  139. short * addr = (short *) &addr_in6->sin6_addr.s6_addr;
  140. /* for IPv6, the address will be in the form of
  141. "[xx:xx:xx:xx:xx:xx:xx:xx]:port". */
  142. len = snprintf(uri, count, "[%x:%x:%x:%x:%x:%x:%x:%x]:%u",
  143. addr[0], addr[1], addr[2], addr[3],
  144. addr[4], addr[5], addr[6], addr[7],
  145. __ntohs(addr_in6->sin6_port));
  146. } else {
  147. return -PAL_ERROR_INVAL;
  148. }
  149. return len;
  150. }
  151. /* parse the uri for a socket stream. The uri might have both binding
  152. address and connecting address, or connecting address only. The form
  153. of uri will be either "bind-addr:bind-port:connect-addr:connect-port"
  154. or "addr:port". */
  155. static int socket_parse_uri (char * uri,
  156. struct sockaddr ** bind_addr, int * bind_addrlen,
  157. struct sockaddr ** dest_addr, int * dest_addrlen)
  158. {
  159. int ret;
  160. if (!bind_addr && !dest_addr)
  161. return 0;
  162. if (!uri || !(*uri)) {
  163. if (bind_addr)
  164. *bind_addr = NULL;
  165. if (dest_addr)
  166. *dest_addr = NULL;
  167. return 0;
  168. }
  169. /* at least parse uri once */
  170. if ((ret = inet_parse_uri(&uri, bind_addr ? *bind_addr : *dest_addr,
  171. bind_addr ? bind_addrlen : dest_addrlen)) < 0)
  172. return ret;
  173. if (!(bind_addr && dest_addr))
  174. return 0;
  175. /* if you reach here, it can only be connection address */
  176. if (!uri || (ret = inet_parse_uri(&uri, *dest_addr, dest_addrlen)) < 0) {
  177. *dest_addr = *bind_addr;
  178. *dest_addrlen = *bind_addrlen;
  179. *bind_addr = NULL;
  180. *bind_addrlen = 0;
  181. }
  182. return 0;
  183. }
  184. /* fill in the PAL handle based on the file descriptors and address given. */
  185. static inline
  186. PAL_HANDLE socket_create_handle (int type, int fd, int options,
  187. struct sockaddr * bind_addr, int bind_addrlen,
  188. struct sockaddr * dest_addr, int dest_addrlen)
  189. {
  190. PAL_HANDLE hdl = malloc(HANDLE_SIZE(sock) + (bind_addr ? bind_addrlen : 0) +
  191. (dest_addr ? dest_addrlen : 0));
  192. if (!hdl)
  193. return NULL;
  194. options = HOST_SOCKET_OPTIONS(options);
  195. memset(hdl, 0, sizeof(union pal_handle));
  196. PAL_GET_TYPE(hdl) = type;
  197. hdl->hdr.flags |= RFD(0)|(type != pal_type_tcpsrv ? WFD(0) : 0);
  198. hdl->sock.fd = fd;
  199. void * addr = (void *) hdl + HANDLE_SIZE(sock);
  200. if (bind_addr) {
  201. hdl->sock.bind = addr;
  202. memcpy(addr, bind_addr, bind_addrlen);
  203. addr += bind_addrlen;
  204. } else {
  205. hdl->sock.bind = NULL;
  206. }
  207. if (dest_addr) {
  208. hdl->sock.conn = addr;
  209. memcpy(addr, dest_addr, dest_addrlen);
  210. addr += dest_addrlen;
  211. } else {
  212. hdl->sock.conn = NULL;
  213. }
  214. hdl->sock.nonblocking = (options & SOCK_NONBLOCK) ?
  215. PAL_TRUE : PAL_FALSE;
  216. hdl->sock.linger = 0;
  217. if (type == pal_type_tcpsrv) {
  218. hdl->sock.receivebuf = 0;
  219. hdl->sock.sendbuf = 0;
  220. } else {
  221. int ret, val, len = sizeof(int);
  222. ret = INLINE_SYSCALL(getsockopt, 5, fd, SOL_SOCKET, SO_RCVBUF,
  223. &val, &len);
  224. hdl->sock.receivebuf = IS_ERR(ret) ? 0 : val;
  225. ret = INLINE_SYSCALL(getsockopt, 5, fd, SOL_SOCKET, SO_SNDBUF,
  226. &val, &len);
  227. hdl->sock.sendbuf = IS_ERR(ret) ? 0 : val;
  228. }
  229. hdl->sock.receivetimeout = 0;
  230. hdl->sock.sendtimeout = 0;
  231. hdl->sock.tcp_cork = PAL_FALSE;
  232. hdl->sock.tcp_keepalive = PAL_FALSE;
  233. hdl->sock.tcp_nodelay = PAL_FALSE;
  234. return hdl;
  235. }
  236. #if ALLOW_BIND_ANY == 0
  237. static bool check_zero (void * mem, size_t size)
  238. {
  239. void * p = mem, * q = mem + size;
  240. while (p < q) {
  241. if (p <= q - sizeof(long)) {
  242. if (*(long *) p)
  243. return false;
  244. p += sizeof(long);
  245. } else if (p <= q - sizeof(int)) {
  246. if (*(int *) p)
  247. return false;
  248. p += sizeof(int);
  249. } else if (p <= q - sizeof(short)) {
  250. if (*(short *) p)
  251. return false;
  252. p += sizeof(short);
  253. } else {
  254. if (*(char *) p)
  255. return false;
  256. p++;
  257. }
  258. }
  259. return true;
  260. }
  261. /* check if an address is "Any" */
  262. static bool check_any_addr (struct sockaddr * addr)
  263. {
  264. if (addr->sa_family == AF_INET) {
  265. struct sockaddr_in * addr_in =
  266. (struct sockaddr_in *) addr;
  267. return addr_in->sin_port == 0 &&
  268. check_zero(&addr_in->sin_addr,
  269. sizeof(addr_in->sin_addr));
  270. } else if (addr->sa_family == AF_INET6) {
  271. struct sockaddr_in6 * addr_in6 =
  272. (struct sockaddr_in6 *) addr;
  273. return addr_in6->sin6_port == 0 &&
  274. check_zero(&addr_in6->sin6_addr,
  275. sizeof(addr_in6->sin6_addr));
  276. }
  277. return false;
  278. }
  279. #endif
  280. /* listen on a tcp socket */
  281. static int tcp_listen (PAL_HANDLE * handle, char * uri, int options)
  282. {
  283. struct sockaddr buffer, * bind_addr = &buffer;
  284. int bind_addrlen;
  285. int ret, fd = -1;
  286. if ((ret = socket_parse_uri(uri, &bind_addr, &bind_addrlen,
  287. NULL, NULL)) < 0)
  288. return ret;
  289. #if ALLOW_BIND_ANY == 0
  290. /* the socket need to have a binding address, a null address or an
  291. any address is not allowed */
  292. if (addr_check_any(bind_addr))
  293. return -PAL_ERROR_INVAL;
  294. #endif
  295. options = HOST_SOCKET_OPTIONS(options);
  296. /* the socket need to have a binding address, a null address or an
  297. any address is not allowed */
  298. if (!bind_addr == 0)
  299. return -PAL_ERROR_INVAL;
  300. fd = INLINE_SYSCALL(socket, 3, bind_addr->sa_family,
  301. SOCK_STREAM|SOCK_CLOEXEC|options, 0);
  302. if (IS_ERR(fd))
  303. return -PAL_ERROR_DENIED;
  304. /* must set the socket to be reuseable */
  305. int reuseaddr = 1;
  306. INLINE_SYSCALL(setsockopt, 5, fd, SOL_SOCKET, SO_REUSEADDR, &reuseaddr,
  307. sizeof(int)); // maybe SO_REUSEPORT
  308. ret = INLINE_SYSCALL(bind, 3, fd, bind_addr, bind_addrlen);
  309. if (IS_ERR(ret)) {
  310. switch(ERRNO(ret)) {
  311. case EINVAL:
  312. ret = -PAL_ERROR_INVAL;
  313. goto failed;
  314. case EADDRINUSE:
  315. ret = -PAL_ERROR_STREAMEXIST;
  316. goto failed;
  317. default:
  318. ret = -PAL_ERROR_DENIED;
  319. goto failed;
  320. }
  321. }
  322. ret = INLINE_SYSCALL(listen, 2, fd, DEFAULT_BACKLOG);
  323. if (IS_ERR(ret))
  324. return -PAL_ERROR_DENIED;
  325. *handle = socket_create_handle(pal_type_tcpsrv, fd, options,
  326. bind_addr, bind_addrlen, NULL, 0);
  327. if (!(*handle)) {
  328. ret = -PAL_ERROR_NOMEM;
  329. goto failed;
  330. }
  331. return 0;
  332. failed:
  333. INLINE_SYSCALL(close, 1, fd);
  334. return ret;
  335. }
  336. /* accept a tcp connection */
  337. static int tcp_accept (PAL_HANDLE handle, PAL_HANDLE * client)
  338. {
  339. if (!IS_HANDLE_TYPE(handle, tcpsrv) ||
  340. !handle->sock.bind || handle->sock.conn)
  341. return -PAL_ERROR_NOTSERVER;
  342. if (handle->sock.fd == PAL_IDX_POISON)
  343. return -PAL_ERROR_BADHANDLE;
  344. struct sockaddr * bind_addr = (struct sockaddr *) handle->sock.bind;
  345. int bind_addrlen = addr_size(bind_addr);
  346. struct sockaddr buffer;
  347. socklen_t addrlen = sizeof(struct sockaddr);
  348. int ret = 0;
  349. int newfd = INLINE_SYSCALL(accept4, 4, handle->sock.fd, &buffer,
  350. &addrlen, SOCK_CLOEXEC);
  351. if (IS_ERR(newfd))
  352. switch(ERRNO(newfd)) {
  353. case EWOULDBLOCK:
  354. return -PAL_ERROR_TRYAGAIN;
  355. case ECONNABORTED:
  356. return -PAL_ERROR_STREAMNOTEXIST;
  357. default:
  358. return unix_to_pal_error(ERRNO(newfd));
  359. }
  360. struct sockaddr * dest_addr = &buffer;
  361. int dest_addrlen = addrlen;
  362. *client = socket_create_handle(pal_type_tcp, newfd, 0,
  363. bind_addr, bind_addrlen,
  364. dest_addr, dest_addrlen);
  365. if (!(*client)) {
  366. ret = -PAL_ERROR_NOMEM;
  367. goto failed;
  368. }
  369. return 0;
  370. failed:
  371. INLINE_SYSCALL(close, 1, newfd);
  372. return ret;
  373. }
  374. /* connect on a tcp socket */
  375. static int tcp_connect (PAL_HANDLE * handle, char * uri, int options)
  376. {
  377. struct sockaddr buffer[3];
  378. struct sockaddr * bind_addr = buffer, * dest_addr = buffer + 1;
  379. int bind_addrlen, dest_addrlen;
  380. int ret, fd = -1;
  381. options = HOST_SOCKET_OPTIONS(options);
  382. /* accepting two kind of different uri:
  383. dest-ip:dest-port or bind-ip:bind-port:dest-ip:dest-port */
  384. if ((ret = socket_parse_uri(uri, &bind_addr, &bind_addrlen,
  385. &dest_addr, &dest_addrlen)) < 0)
  386. return ret;
  387. if (!dest_addr)
  388. return -PAL_ERROR_INVAL;
  389. if (bind_addr && bind_addr->sa_family != dest_addr->sa_family)
  390. return -PAL_ERROR_INVAL;
  391. #if ALLOW_BIND_ANY == 0
  392. /* the socket need to have a binding address, a null address or an
  393. any address is not allowed */
  394. if (bind_addr && addr_check_any(bind_addr))
  395. return -PAL_ERROR_INVAL;
  396. #endif
  397. fd = INLINE_SYSCALL(socket, 3, dest_addr->sa_family,
  398. SOCK_STREAM|SOCK_CLOEXEC|options, 0);
  399. if (IS_ERR(fd))
  400. return -PAL_ERROR_DENIED;
  401. if (bind_addr) {
  402. if (IS_ERR(ret)) {
  403. INLINE_SYSCALL(close, 1, fd);
  404. switch (ERRNO(ret)) {
  405. case EADDRINUSE:
  406. ret = -PAL_ERROR_STREAMEXIST;
  407. goto failed;
  408. case EADDRNOTAVAIL:
  409. ret = -PAL_ERROR_ADDRNOTEXIST;
  410. goto failed;
  411. default:
  412. ret = unix_to_pal_error(ERRNO(ret));
  413. goto failed;
  414. }
  415. }
  416. }
  417. ret = INLINE_SYSCALL(connect, 3, fd, dest_addr, dest_addrlen);
  418. if (IS_ERR(ret)) {
  419. ret = unix_to_pal_error(ERRNO(ret));
  420. goto failed;
  421. }
  422. if (!bind_addr) {
  423. /* save some space to get socket address */
  424. bind_addr = buffer + 2;
  425. bind_addrlen = sizeof(struct sockaddr);
  426. /* call getsockname to get socket address */
  427. if ((ret = INLINE_SYSCALL(getsockname, 3, fd,
  428. bind_addr, &bind_addrlen)) < 0)
  429. bind_addr = NULL;
  430. }
  431. *handle = socket_create_handle(pal_type_tcp, fd, options,
  432. bind_addr, bind_addrlen,
  433. dest_addr, dest_addrlen);
  434. if (!(*handle)) {
  435. ret = -PAL_ERROR_NOMEM;
  436. goto failed;
  437. }
  438. return 0;
  439. failed:
  440. INLINE_SYSCALL(close, 1, fd);
  441. return ret;
  442. }
  443. /* 'open' operation of tcp stream */
  444. static int tcp_open (PAL_HANDLE *handle, const char * type, const char * uri,
  445. int access, int share, int create, int options)
  446. {
  447. int uri_len = strlen(uri) + 1;
  448. if (uri_len > PAL_SOCKADDR_SIZE)
  449. return -PAL_ERROR_TOOLONG;
  450. char uri_buf[PAL_SOCKADDR_SIZE];
  451. memcpy(uri_buf, uri, uri_len);
  452. if (strcmp_static(type, "tcp.srv"))
  453. return tcp_listen(handle, uri_buf, options);
  454. if (strcmp_static(type, "tcp"))
  455. return tcp_connect(handle, uri_buf, options);
  456. return -PAL_ERROR_NOTSUPPORT;
  457. }
  458. /* 'read' operation of tcp stream */
  459. static int tcp_read (PAL_HANDLE handle, int offset, int len, void * buf)
  460. {
  461. if (!IS_HANDLE_TYPE(handle, tcp) || !handle->sock.conn)
  462. return -PAL_ERROR_NOTCONNECTION;
  463. if (handle->sock.fd == PAL_IDX_POISON)
  464. return -PAL_ERROR_ENDOFSTREAM;
  465. struct msghdr hdr;
  466. struct iovec iov;
  467. iov.iov_base = buf;
  468. iov.iov_len = len;
  469. hdr.msg_name = NULL;
  470. hdr.msg_namelen = 0;
  471. hdr.msg_iov = &iov;
  472. hdr.msg_iovlen = 1;
  473. hdr.msg_control = NULL;
  474. hdr.msg_controllen = 0;
  475. hdr.msg_flags = 0;
  476. int bytes = INLINE_SYSCALL(recvmsg, 3, handle->sock.fd, &hdr, 0);
  477. if (IS_ERR(bytes))
  478. switch (ERRNO(bytes)) {
  479. case EWOULDBLOCK:
  480. return -PAL_ERROR_TRYAGAIN;
  481. default:
  482. return unix_to_pal_error(ERRNO(bytes));
  483. }
  484. if (!bytes)
  485. return -PAL_ERROR_ENDOFSTREAM;
  486. return bytes;
  487. }
  488. /* write' operation of tcp stream */
  489. static int tcp_write (PAL_HANDLE handle, int offset, int len, const void * buf)
  490. {
  491. if (!IS_HANDLE_TYPE(handle, tcp) || !handle->sock.conn)
  492. return -PAL_ERROR_NOTCONNECTION;
  493. if (handle->sock.fd == PAL_IDX_POISON)
  494. return -PAL_ERROR_CONNFAILED;
  495. struct msghdr hdr;
  496. struct iovec iov;
  497. iov.iov_base = (void *) buf;
  498. iov.iov_len = len;
  499. hdr.msg_name = NULL;
  500. hdr.msg_namelen = 0;
  501. hdr.msg_iov = &iov;
  502. hdr.msg_iovlen = 1;
  503. hdr.msg_control = NULL;
  504. hdr.msg_controllen = 0;
  505. hdr.msg_flags = 0;
  506. int bytes = INLINE_SYSCALL(sendmsg, 3, handle->sock.fd, &hdr, MSG_NOSIGNAL);
  507. if (IS_ERR(bytes))
  508. switch(ERRNO(bytes)) {
  509. case ECONNRESET:
  510. case EPIPE:
  511. return -PAL_ERROR_CONNFAILED;
  512. case EWOULDBLOCK:
  513. handle->hdr.flags &= ~WRITEABLE(0);
  514. return -PAL_ERROR_TRYAGAIN;
  515. default:
  516. return unix_to_pal_error(ERRNO(bytes));
  517. }
  518. if (bytes == len)
  519. handle->hdr.flags |= WRITEABLE(0);
  520. else
  521. handle->hdr.flags &= ~WRITEABLE(0);
  522. return bytes;
  523. }
  524. /* used by 'open' operation of tcp stream for bound socket */
  525. static int udp_bind (PAL_HANDLE * handle, char * uri, int options)
  526. {
  527. struct sockaddr buffer, * bind_addr = &buffer;
  528. int bind_addrlen;
  529. int ret = 0, fd = -1;
  530. if ((ret = socket_parse_uri(uri, &bind_addr, &bind_addrlen,
  531. NULL, NULL)) < 0)
  532. return ret;
  533. assert(bind_addr);
  534. assert(bind_addrlen == addr_size(bind_addr));
  535. #if ALLOW_BIND_ANY == 0
  536. /* the socket need to have a binding address, a null address or an
  537. any address is not allowed */
  538. if (addr_check_any(bind_addr))
  539. return -PAL_ERROR_INVAL;
  540. #endif
  541. options = HOST_SOCKET_OPTIONS(options);
  542. fd = INLINE_SYSCALL(socket, 3, bind_addr->sa_family,
  543. SOCK_DGRAM|SOCK_CLOEXEC|options, 0);
  544. if (IS_ERR(fd))
  545. return -PAL_ERROR_DENIED;
  546. ret = INLINE_SYSCALL(bind, 3, fd, bind_addr, bind_addrlen);
  547. if (IS_ERR(ret)) {
  548. switch (ERRNO(ret)) {
  549. case EADDRINUSE:
  550. ret = -PAL_ERROR_STREAMEXIST;
  551. goto failed;
  552. case EADDRNOTAVAIL:
  553. ret = -PAL_ERROR_ADDRNOTEXIST;
  554. goto failed;
  555. default:
  556. ret = unix_to_pal_error(ERRNO(ret));
  557. goto failed;
  558. }
  559. }
  560. *handle = socket_create_handle(pal_type_udpsrv, fd, options,
  561. bind_addr, bind_addrlen, NULL, 0);
  562. if (!(*handle)) {
  563. ret = -ENOMEM;
  564. goto failed;
  565. }
  566. return 0;
  567. failed:
  568. INLINE_SYSCALL(close, 1, fd);
  569. return ret;
  570. }
  571. /* used by 'open' operation of tcp stream for connected socket */
  572. static int udp_connect (PAL_HANDLE * handle, char * uri, int options)
  573. {
  574. struct sockaddr buffer[2];
  575. struct sockaddr * bind_addr = buffer, * dest_addr = buffer + 1;
  576. int bind_addrlen, dest_addrlen;
  577. int ret, fd = -1;
  578. if ((ret = socket_parse_uri(uri, &bind_addr, &bind_addrlen,
  579. &dest_addr, &dest_addrlen)) < 0)
  580. return ret;
  581. #if ALLOW_BIND_ANY == 0
  582. /* the socket need to have a binding address, a null address or an
  583. any address is not allowed */
  584. if (bind_addr && addr_check_any(bind_addr))
  585. return -PAL_ERROR_INVAL;
  586. #endif
  587. options = HOST_SOCKET_OPTIONS(options);
  588. fd = INLINE_SYSCALL(socket, 3, dest_addr ? dest_addr->sa_family : AF_INET,
  589. SOCK_DGRAM|SOCK_CLOEXEC|options, 0);
  590. if (IS_ERR(fd))
  591. return -PAL_ERROR_DENIED;
  592. if (bind_addr) {
  593. ret = INLINE_SYSCALL(bind, 3, fd, bind_addr, bind_addrlen);
  594. if (IS_ERR(ret)) {
  595. switch (ERRNO(ret)) {
  596. case EADDRINUSE:
  597. ret = -PAL_ERROR_STREAMEXIST;
  598. goto failed;
  599. case EADDRNOTAVAIL:
  600. ret = -PAL_ERROR_ADDRNOTEXIST;
  601. goto failed;
  602. default:
  603. ret = unix_to_pal_error(ERRNO(ret));
  604. goto failed;
  605. }
  606. }
  607. }
  608. *handle = socket_create_handle(dest_addr ? pal_type_udp : pal_type_udpsrv,
  609. fd, options,
  610. bind_addr, bind_addrlen,
  611. dest_addr, dest_addrlen);
  612. if (!(*handle)) {
  613. ret = -ENOMEM;
  614. goto failed;
  615. }
  616. return 0;
  617. failed:
  618. INLINE_SYSCALL(close, 1, fd);
  619. return ret;
  620. }
  621. static int udp_open (PAL_HANDLE *hdl, const char * type, const char * uri,
  622. int access, int share, int create, int options)
  623. {
  624. char buf[PAL_SOCKADDR_SIZE];
  625. int len = strlen(uri);
  626. if (len >= PAL_SOCKADDR_SIZE)
  627. return -PAL_ERROR_TOOLONG;
  628. memcpy(buf, uri, len + 1);
  629. options &= PAL_OPTION_MASK;
  630. if (!strcmp_static(type, "udp.srv"))
  631. return udp_bind(hdl, buf, options);
  632. if (!strcmp_static(type, "udp"))
  633. return udp_connect(hdl, buf, options);
  634. return -PAL_ERROR_NOTSUPPORT;
  635. }
  636. static int udp_receive (PAL_HANDLE handle, int offset, int len, void * buf)
  637. {
  638. if (!IS_HANDLE_TYPE(handle, udp))
  639. return -PAL_ERROR_NOTCONNECTION;
  640. if (handle->sock.fd == PAL_IDX_POISON)
  641. return -PAL_ERROR_BADHANDLE;
  642. struct msghdr hdr;
  643. struct iovec iov;
  644. iov.iov_base = buf;
  645. iov.iov_len = len;
  646. hdr.msg_name = NULL;
  647. hdr.msg_namelen = 0;
  648. hdr.msg_iov = &iov;
  649. hdr.msg_iovlen = 1;
  650. hdr.msg_control = NULL;
  651. hdr.msg_controllen = 0;
  652. hdr.msg_flags = 0;
  653. int bytes = INLINE_SYSCALL(recvmsg, 3, handle->sock.fd, &hdr, 0);
  654. if (IS_ERR(bytes))
  655. switch(ERRNO(bytes)) {
  656. case EWOULDBLOCK:
  657. return -PAL_ERROR_TRYAGAIN;
  658. case EINTR:
  659. return -PAL_ERROR_INTERRUPTED;
  660. default:
  661. return unix_to_pal_error(ERRNO(bytes));
  662. }
  663. return bytes;
  664. }
  665. static int udp_receivebyaddr (PAL_HANDLE handle, int offset, int len,
  666. void * buf, char * addr, int addrlen)
  667. {
  668. if (!IS_HANDLE_TYPE(handle, udpsrv))
  669. return -PAL_ERROR_NOTCONNECTION;
  670. if (handle->sock.fd == PAL_IDX_POISON)
  671. return -PAL_ERROR_BADHANDLE;
  672. struct sockaddr conn_addr;
  673. socklen_t conn_addrlen = sizeof(struct sockaddr);
  674. struct msghdr hdr;
  675. struct iovec iov;
  676. iov.iov_base = buf;
  677. iov.iov_len = len;
  678. hdr.msg_name = &conn_addr;
  679. hdr.msg_namelen = conn_addrlen;
  680. hdr.msg_iov = &iov;
  681. hdr.msg_iovlen = 1;
  682. hdr.msg_control = NULL;
  683. hdr.msg_controllen = 0;
  684. hdr.msg_flags = 0;
  685. int bytes = INLINE_SYSCALL(recvmsg, 3, handle->sock.fd, &hdr, 0);
  686. if (IS_ERR(bytes))
  687. switch(ERRNO(bytes)) {
  688. case EWOULDBLOCK:
  689. return -PAL_ERROR_TRYAGAIN;
  690. case EINTR:
  691. return -PAL_ERROR_INTERRUPTED;
  692. case ECONNREFUSED:
  693. return -PAL_ERROR_STREAMNOTEXIST;
  694. default:
  695. return unix_to_pal_error(ERRNO(bytes));
  696. }
  697. char * tmp = strcpy_static(addr, "udp:", addrlen);
  698. if (!tmp)
  699. return -PAL_ERROR_OVERFLOW;
  700. inet_create_uri(tmp, addr + addrlen - tmp, &conn_addr, hdr.msg_namelen);
  701. return bytes;
  702. }
  703. static int udp_send (PAL_HANDLE handle, int offset, int len, const void * buf)
  704. {
  705. if (!IS_HANDLE_TYPE(handle, udp))
  706. return -PAL_ERROR_NOTCONNECTION;
  707. if (handle->sock.fd == PAL_IDX_POISON)
  708. return -PAL_ERROR_BADHANDLE;
  709. struct msghdr hdr;
  710. struct iovec iov;
  711. iov.iov_base = (void *) buf;
  712. iov.iov_len = len;
  713. hdr.msg_name = handle->sock.conn;
  714. hdr.msg_namelen = addr_size(handle->sock.conn);
  715. hdr.msg_iov = &iov;
  716. hdr.msg_iovlen = 1;
  717. hdr.msg_control = NULL;
  718. hdr.msg_controllen = 0;
  719. hdr.msg_flags = 0;
  720. int bytes = INLINE_SYSCALL(sendmsg, 3, handle->sock.fd, &hdr, MSG_NOSIGNAL);
  721. if (IS_ERR(bytes))
  722. switch(ERRNO(bytes)) {
  723. case EAGAIN:
  724. handle->hdr.flags &= ~WRITEABLE(0);
  725. return -PAL_ERROR_TRYAGAIN;
  726. case ECONNRESET:
  727. case EPIPE:
  728. return -PAL_ERROR_CONNFAILED;
  729. default:
  730. return unix_to_pal_error(ERRNO(bytes));
  731. }
  732. if (bytes == len)
  733. handle->hdr.flags |= WRITEABLE(0);
  734. else
  735. handle->hdr.flags &= ~WRITEABLE(0);
  736. return bytes;
  737. }
  738. static int udp_sendbyaddr (PAL_HANDLE handle, int offset, int len,
  739. const void * buf, const char * addr, int addrlen)
  740. {
  741. if (!IS_HANDLE_TYPE(handle, udpsrv))
  742. return -PAL_ERROR_NOTCONNECTION;
  743. if (handle->sock.fd == PAL_IDX_POISON)
  744. return -PAL_ERROR_BADHANDLE;
  745. if (strpartcmp_static(addr, "udp:"))
  746. return -PAL_ERROR_INVAL;
  747. addr += static_strlen("udp:");
  748. addrlen -= static_strlen("udp:");
  749. char * addrbuf = __alloca(addrlen + 1);
  750. memcpy(addrbuf, addr, addrlen + 1);
  751. struct sockaddr conn_addr;
  752. int conn_addrlen;
  753. int ret = inet_parse_uri(&addrbuf, &conn_addr, &conn_addrlen);
  754. if (ret < 0)
  755. return ret;
  756. struct msghdr hdr;
  757. struct iovec iov;
  758. iov.iov_base = (void *) buf;
  759. iov.iov_len = len;
  760. hdr.msg_name = &conn_addr;
  761. hdr.msg_namelen = conn_addrlen;
  762. hdr.msg_iov = &iov;
  763. hdr.msg_iovlen = 1;
  764. hdr.msg_control = NULL;
  765. hdr.msg_controllen = 0;
  766. hdr.msg_flags = 0;
  767. int bytes = INLINE_SYSCALL(sendmsg, 3, handle->sock.fd, &hdr, MSG_NOSIGNAL);
  768. if (IS_ERR(bytes))
  769. switch(ERRNO(bytes)) {
  770. case ECONNRESET:
  771. case EPIPE:
  772. return -PAL_ERROR_CONNFAILED;
  773. case EAGAIN:
  774. handle->hdr.flags &= ~WRITEABLE(0);
  775. default:
  776. return unix_to_pal_error(ERRNO(bytes));
  777. }
  778. if (bytes == len)
  779. handle->hdr.flags |= WRITEABLE(0);
  780. else
  781. handle->hdr.flags &= ~WRITEABLE(0);
  782. return bytes;
  783. }
  784. static int socket_delete (PAL_HANDLE handle, int access)
  785. {
  786. if (handle->sock.fd == PAL_IDX_POISON)
  787. return 0;
  788. if (!IS_HANDLE_TYPE(handle, tcp) && access)
  789. return -PAL_ERROR_INVAL;
  790. if (IS_HANDLE_TYPE(handle, tcp) || IS_HANDLE_TYPE(handle, tcpsrv)) {
  791. int shutdown;
  792. switch (access) {
  793. case 0:
  794. shutdown = SHUT_RDWR;
  795. break;
  796. case PAL_DELETE_RD:
  797. shutdown = SHUT_RD;
  798. break;
  799. case PAL_DELETE_WR:
  800. shutdown = SHUT_WR;
  801. break;
  802. default:
  803. return -PAL_ERROR_INVAL;
  804. }
  805. INLINE_SYSCALL(shutdown, 2, handle->sock.fd, shutdown);
  806. }
  807. return 0;
  808. }
  809. static int socket_close (PAL_HANDLE handle)
  810. {
  811. if (handle->sock.fd != PAL_IDX_POISON) {
  812. INLINE_SYSCALL(close, 1, handle->sock.fd);
  813. handle->sock.fd = PAL_IDX_POISON;
  814. }
  815. if (handle->sock.bind)
  816. handle->sock.bind = NULL;
  817. if (handle->sock.conn)
  818. handle->sock.conn = NULL;
  819. return 0;
  820. }
  821. static int socket_attrquerybyhdl (PAL_HANDLE handle, PAL_STREAM_ATTR * attr)
  822. {
  823. if (handle->sock.fd == PAL_IDX_POISON)
  824. return -PAL_ERROR_BADHANDLE;
  825. int fd = handle->sock.fd, ret, val;
  826. if (handle->sock.conn) {
  827. /* try use ioctl FIONEAD to get the size of socket */
  828. ret = INLINE_SYSCALL(ioctl, 3, fd, FIONREAD, &val);
  829. if (IS_ERR(ret))
  830. return unix_to_pal_error(ERRNO(ret));
  831. attr->pending_size = val;
  832. attr->readable = !!attr->pending_size > 0;
  833. } else {
  834. attr->readable = !attr->disconnected;
  835. }
  836. attr->handle_type = handle->hdr.type;
  837. attr->disconnected = handle->hdr.flags & ERROR(0);
  838. attr->nonblocking = handle->sock.nonblocking;
  839. attr->writeable = handle->hdr.flags & WRITEABLE(0);
  840. attr->socket.linger = handle->sock.linger;
  841. attr->socket.receivebuf = handle->sock.receivebuf;
  842. attr->socket.sendbuf = handle->sock.sendbuf;
  843. attr->socket.receivetimeout = handle->sock.receivetimeout;
  844. attr->socket.sendtimeout = handle->sock.sendtimeout;
  845. attr->socket.tcp_cork = handle->sock.tcp_cork;
  846. attr->socket.tcp_keepalive = handle->sock.tcp_keepalive;
  847. attr->socket.tcp_nodelay = handle->sock.tcp_nodelay;
  848. return 0;
  849. }
  850. static int socket_attrsetbyhdl (PAL_HANDLE handle, PAL_STREAM_ATTR * attr)
  851. {
  852. if (handle->sock.fd == PAL_IDX_POISON)
  853. return -PAL_ERROR_BADHANDLE;
  854. int fd = handle->sock.fd, ret, val;
  855. if (attr->nonblocking != handle->sock.nonblocking) {
  856. ret = INLINE_SYSCALL(fcntl, 3, fd, F_SETFL,
  857. attr->nonblocking ? O_NONBLOCK : 0);
  858. if (IS_ERR(ret))
  859. return unix_to_pal_error(ERRNO(ret));
  860. handle->sock.nonblocking = attr->nonblocking;
  861. }
  862. if (IS_HANDLE_TYPE(handle, tcpsrv)) {
  863. struct __kernel_linger {
  864. int l_onoff;
  865. int l_linger;
  866. };
  867. if (attr->socket.linger != handle->sock.linger) {
  868. struct __kernel_linger l;
  869. l.l_onoff = attr->socket.linger ? 1 : 0;
  870. l.l_linger = attr->socket.linger;
  871. ret = INLINE_SYSCALL(setsockopt, 5, fd, SOL_SOCKET, SO_LINGER,
  872. &l, sizeof(struct __kernel_linger));
  873. if (IS_ERR(ret))
  874. return unix_to_pal_error(ERRNO(ret));
  875. handle->sock.linger = attr->socket.linger;
  876. }
  877. if (attr->socket.receivebuf != handle->sock.receivebuf) {
  878. int val = attr->socket.receivebuf;
  879. ret = INLINE_SYSCALL(setsockopt, 5, fd, SOL_SOCKET, SO_RCVBUF,
  880. &val, sizeof(int));
  881. if (IS_ERR(ret))
  882. return unix_to_pal_error(ERRNO(ret));
  883. handle->sock.receivebuf = attr->socket.receivebuf;
  884. }
  885. if (attr->socket.sendbuf != handle->sock.sendbuf) {
  886. int val = attr->socket.sendbuf;
  887. ret = INLINE_SYSCALL(setsockopt, 5, fd, SOL_SOCKET, SO_SNDBUF,
  888. &val, sizeof(int));
  889. if (IS_ERR(ret))
  890. return unix_to_pal_error(ERRNO(ret));
  891. handle->sock.sendbuf = attr->socket.sendbuf;
  892. }
  893. if (attr->socket.receivetimeout != handle->sock.receivetimeout) {
  894. int val = attr->socket.receivetimeout;
  895. ret = INLINE_SYSCALL(setsockopt, 5, fd, SOL_SOCKET, SO_RCVTIMEO,
  896. &val, sizeof(int));
  897. if (IS_ERR(ret))
  898. return unix_to_pal_error(ERRNO(ret));
  899. handle->sock.receivetimeout = attr->socket.receivetimeout;
  900. }
  901. if (attr->socket.sendtimeout != handle->sock.sendtimeout) {
  902. int val = attr->socket.sendtimeout;
  903. ret = INLINE_SYSCALL(setsockopt, 5, fd, SOL_SOCKET, SO_SNDTIMEO,
  904. &val, sizeof(int));
  905. if (IS_ERR(ret))
  906. return unix_to_pal_error(ERRNO(ret));
  907. handle->sock.sendtimeout = attr->socket.sendtimeout;
  908. }
  909. }
  910. if (IS_HANDLE_TYPE(handle, tcp) || IS_HANDLE_TYPE(handle, tcpsrv)) {
  911. if (attr->socket.tcp_cork != handle->sock.tcp_cork) {
  912. val = attr->socket.tcp_cork ? 1 : 0;
  913. ret = INLINE_SYSCALL(setsockopt, 5, fd, SOL_TCP, TCP_CORK,
  914. &val, sizeof(int));
  915. if (IS_ERR(ret))
  916. return unix_to_pal_error(ERRNO(ret));
  917. handle->sock.tcp_cork = attr->socket.tcp_cork;
  918. }
  919. if (attr->socket.tcp_keepalive != handle->sock.tcp_keepalive) {
  920. val = attr->socket.tcp_keepalive ? 1 : 0;
  921. ret = INLINE_SYSCALL(setsockopt, 5, fd, SOL_SOCKET, SO_KEEPALIVE,
  922. &val, sizeof(int));
  923. if (IS_ERR(ret))
  924. return unix_to_pal_error(ERRNO(ret));
  925. handle->sock.tcp_keepalive = attr->socket.tcp_keepalive;
  926. }
  927. if (attr->socket.tcp_nodelay != handle->sock.tcp_nodelay) {
  928. val = attr->socket.tcp_nodelay ? 1 : 0;
  929. ret = INLINE_SYSCALL(setsockopt, 5, fd, SOL_TCP, TCP_NODELAY,
  930. &val, sizeof(int));
  931. if (IS_ERR(ret))
  932. return unix_to_pal_error(ERRNO(ret));
  933. handle->sock.tcp_nodelay = attr->socket.tcp_nodelay;
  934. }
  935. }
  936. return 0;
  937. }
  938. static int socket_getname (PAL_HANDLE handle, char * buffer, int count)
  939. {
  940. int old_count = count;
  941. int ret;
  942. const char * prefix = NULL;
  943. int prefix_len = 0;
  944. struct sockaddr * bind_addr = NULL, * dest_addr = NULL;
  945. switch (PAL_GET_TYPE(handle)) {
  946. case pal_type_tcpsrv:
  947. prefix_len = 7;
  948. prefix = "tcp.srv";
  949. bind_addr = handle->sock.bind;
  950. break;
  951. case pal_type_tcp:
  952. prefix_len = 3;
  953. prefix = "tcp";
  954. bind_addr = handle->sock.bind;
  955. dest_addr = handle->sock.conn;
  956. break;
  957. case pal_type_udpsrv:
  958. prefix_len = 7;
  959. prefix = "udp.srv";
  960. bind_addr = handle->sock.bind;
  961. break;
  962. case pal_type_udp:
  963. prefix_len = 3;
  964. prefix = "udp";
  965. bind_addr = handle->sock.bind;
  966. dest_addr = handle->sock.conn;
  967. break;
  968. default:
  969. return -PAL_ERROR_INVAL;
  970. }
  971. if (prefix_len >= count)
  972. return -PAL_ERROR_OVERFLOW;
  973. memcpy(buffer, prefix, prefix_len + 1);
  974. buffer += prefix_len;
  975. count -= prefix_len;
  976. for (int i = 0 ; i < 2 ; i++) {
  977. struct sockaddr * addr = i ? dest_addr : bind_addr;
  978. if (addr) {
  979. if (count <= 1)
  980. return -PAL_ERROR_OVERFLOW;
  981. buffer[0] = ':';
  982. buffer[1] = 0;
  983. buffer++;
  984. count--;
  985. if ((ret = inet_create_uri(buffer, count, addr,
  986. addr_size(addr))) < 0)
  987. return ret;
  988. buffer += ret;
  989. count -= ret;
  990. }
  991. }
  992. return old_count - count;
  993. }
  994. struct handle_ops tcp_ops = {
  995. .getname = &socket_getname,
  996. .open = &tcp_open,
  997. .waitforclient = &tcp_accept,
  998. .read = &tcp_read,
  999. .write = &tcp_write,
  1000. .delete = &socket_delete,
  1001. .close = &socket_close,
  1002. .attrquerybyhdl = &socket_attrquerybyhdl,
  1003. .attrsetbyhdl = &socket_attrsetbyhdl,
  1004. };
  1005. struct handle_ops udp_ops = {
  1006. .getname = &socket_getname,
  1007. .open = &udp_open,
  1008. .read = &udp_receive,
  1009. .write = &udp_send,
  1010. .delete = &socket_delete,
  1011. .close = &socket_close,
  1012. .attrquerybyhdl = &socket_attrquerybyhdl,
  1013. .attrsetbyhdl = &socket_attrsetbyhdl,
  1014. };
  1015. struct handle_ops udpsrv_ops = {
  1016. .getname = &socket_getname,
  1017. .open = &udp_open,
  1018. .readbyaddr = &udp_receivebyaddr,
  1019. .writebyaddr = &udp_sendbyaddr,
  1020. .delete = &socket_delete,
  1021. .close = &socket_close,
  1022. .attrquerybyhdl = &socket_attrquerybyhdl,
  1023. .attrsetbyhdl = &socket_attrsetbyhdl,
  1024. };
  1025. static int mcast_s (PAL_HANDLE handle, int port)
  1026. {
  1027. handle->mcast.srv = PAL_IDX_POISON;
  1028. int ret = 0;
  1029. int fd = INLINE_SYSCALL(socket, 3, AF_INET, SOCK_DGRAM, 0);
  1030. if (IS_ERR(fd))
  1031. return -PAL_ERROR_DENIED;
  1032. struct in_addr local;
  1033. local.s_addr = INADDR_ANY;
  1034. ret = INLINE_SYSCALL(setsockopt, 5, fd, IPPROTO_IP, IP_MULTICAST_IF,
  1035. &local, sizeof(local));
  1036. if (IS_ERR(ret))
  1037. return -PAL_ERROR_DENIED;
  1038. handle->hdr.flags |= WFD(1)|WRITEABLE(1);
  1039. handle->mcast.srv = fd;
  1040. return 0;
  1041. }
  1042. static int mcast_c (PAL_HANDLE handle, int port)
  1043. {
  1044. handle->mcast.cli = PAL_IDX_POISON;
  1045. int ret = 0;
  1046. int fd = INLINE_SYSCALL(socket, 3, AF_INET, SOCK_DGRAM, 0);
  1047. if (IS_ERR(fd))
  1048. return -PAL_ERROR_DENIED;
  1049. int reuse = 1;
  1050. INLINE_SYSCALL(setsockopt, 5, fd, SOL_SOCKET, SO_REUSEPORT,
  1051. &reuse, sizeof(reuse));
  1052. struct sockaddr_in addr;
  1053. addr.sin_family = AF_INET;
  1054. addr.sin_addr.s_addr = htonl(INADDR_ANY);
  1055. addr.sin_port = htons(port);
  1056. ret = INLINE_SYSCALL(bind, 3, fd, &addr, sizeof(addr));
  1057. if (IS_ERR(ret))
  1058. return -PAL_ERROR_DENIED;
  1059. struct in_addr local;
  1060. local.s_addr = INADDR_ANY;
  1061. ret = INLINE_SYSCALL(setsockopt, 5, fd, IPPROTO_IP, IP_MULTICAST_IF,
  1062. &local, sizeof(local));
  1063. if (IS_ERR(ret))
  1064. return -PAL_ERROR_DENIED;
  1065. struct ip_mreq group;
  1066. inet_pton4(GRAPHENE_MCAST_GROUP, sizeof(GRAPHENE_MCAST_GROUP) - 1,
  1067. &group.imr_multiaddr.s_addr);
  1068. group.imr_interface.s_addr = htonl(INADDR_ANY);
  1069. ret = INLINE_SYSCALL(setsockopt, 5, fd, IPPROTO_IP, IP_ADD_MEMBERSHIP,
  1070. &group, sizeof(group));
  1071. if (IS_ERR(ret))
  1072. return -PAL_ERROR_DENIED;
  1073. handle->mcast.cli = fd;
  1074. handle->mcast.nonblocking = PAL_FALSE;
  1075. return 0;
  1076. }
  1077. PAL_HANDLE _DkBroadcastStreamOpen (void)
  1078. {
  1079. if (!pal_sec.mcast_port) {
  1080. unsigned short mcast_port;
  1081. _DkFastRandomBitsRead(&mcast_port, sizeof(unsigned short));
  1082. if (mcast_port < 1024)
  1083. mcast_port += 1024;
  1084. pal_sec.mcast_port = mcast_port > 1024 ? mcast_port : mcast_port + 1204;
  1085. }
  1086. PAL_HANDLE hdl = malloc(HANDLE_SIZE(mcast));
  1087. SET_HANDLE_TYPE(hdl, mcast);
  1088. mcast_s(hdl, pal_sec.mcast_port);
  1089. mcast_c(hdl, pal_sec.mcast_port);
  1090. hdl->mcast.port = pal_sec.mcast_port;
  1091. return hdl;
  1092. }
  1093. static int mcast_send (PAL_HANDLE handle, int offset, int size,
  1094. const void * buf)
  1095. {
  1096. if (handle->mcast.srv == PAL_IDX_POISON)
  1097. return -PAL_ERROR_BADHANDLE;
  1098. struct sockaddr_in addr;
  1099. addr.sin_family = AF_INET;
  1100. inet_pton4(GRAPHENE_MCAST_GROUP, sizeof(GRAPHENE_MCAST_GROUP) - 1,
  1101. &addr.sin_addr.s_addr);
  1102. addr.sin_port = htons(handle->mcast.port);
  1103. struct msghdr hdr;
  1104. struct iovec iov;
  1105. iov.iov_base = (void *) buf;
  1106. iov.iov_len = size;
  1107. hdr.msg_name = &addr;
  1108. hdr.msg_namelen = sizeof(addr);
  1109. hdr.msg_iov = &iov;
  1110. hdr.msg_iovlen = 1;
  1111. hdr.msg_control = NULL;
  1112. hdr.msg_controllen = 0;
  1113. hdr.msg_flags = 0;
  1114. int bytes = INLINE_SYSCALL(sendmsg, 3, handle->mcast.srv, &hdr,
  1115. MSG_NOSIGNAL);
  1116. if (IS_ERR(bytes))
  1117. switch(ERRNO(bytes)) {
  1118. case ECONNRESET:
  1119. case EPIPE:
  1120. return -PAL_ERROR_CONNFAILED;
  1121. case EAGAIN:
  1122. handle->hdr.flags &= ~WRITEABLE(1);
  1123. default:
  1124. return unix_to_pal_error(ERRNO(bytes));
  1125. }
  1126. if (bytes == size)
  1127. handle->hdr.flags |= WRITEABLE(1);
  1128. else
  1129. handle->hdr.flags &= ~WRITEABLE(1);
  1130. return bytes;
  1131. }
  1132. static int mcast_receive (PAL_HANDLE handle, int offset, int size, void * buf)
  1133. {
  1134. if (handle->mcast.cli == PAL_IDX_POISON)
  1135. return -PAL_ERROR_BADHANDLE;
  1136. struct msghdr hdr;
  1137. struct iovec iov;
  1138. iov.iov_base = buf;
  1139. iov.iov_len = size;
  1140. hdr.msg_name = NULL;
  1141. hdr.msg_namelen = 0;
  1142. hdr.msg_iov = &iov;
  1143. hdr.msg_iovlen = 1;
  1144. hdr.msg_control = NULL;
  1145. hdr.msg_controllen = 0;
  1146. hdr.msg_flags = 0;
  1147. int bytes = INLINE_SYSCALL(recvmsg, 3, handle->mcast.cli, &hdr, 0);
  1148. if (IS_ERR(bytes))
  1149. return -PAL_ERROR_DENIED;
  1150. return bytes;
  1151. }
  1152. static int mcast_close (PAL_HANDLE handle)
  1153. {
  1154. if (handle->mcast.srv != PAL_IDX_POISON) {
  1155. INLINE_SYSCALL(close, 1, handle->mcast.srv);
  1156. handle->mcast.srv = PAL_IDX_POISON;
  1157. }
  1158. if (handle->mcast.cli != PAL_IDX_POISON) {
  1159. INLINE_SYSCALL(close, 1, handle->mcast.cli);
  1160. handle->mcast.cli = PAL_IDX_POISON;
  1161. }
  1162. return 0;
  1163. }
  1164. static int mcast_attrquerybyhdl (PAL_HANDLE handle, PAL_STREAM_ATTR * attr)
  1165. {
  1166. int ret, val;
  1167. if (handle->mcast.cli == PAL_IDX_POISON)
  1168. return -PAL_ERROR_BADHANDLE;
  1169. ret = INLINE_SYSCALL(ioctl, 3, handle->mcast.cli, FIONREAD, &val);
  1170. if (IS_ERR(ret))
  1171. return unix_to_pal_error(ERRNO(ret));
  1172. attr->handle_type = pal_type_mcast;
  1173. attr->disconnected = handle->hdr.flags & (ERROR(0)|ERROR(1));
  1174. attr->nonblocking = handle->mcast.nonblocking;
  1175. attr->readable = !!val;
  1176. attr->writeable = handle->hdr.flags & WRITEABLE(1);
  1177. attr->runnable = PAL_FALSE;
  1178. attr->pending_size = val;
  1179. return 0;
  1180. }
  1181. static int mcast_attrsetbyhdl (PAL_HANDLE handle, PAL_STREAM_ATTR * attr)
  1182. {
  1183. if (handle->mcast.cli == PAL_IDX_POISON)
  1184. return -PAL_ERROR_BADHANDLE;
  1185. int ret;
  1186. PAL_BOL * nonblocking = &handle->mcast.nonblocking;
  1187. if (attr->nonblocking != *nonblocking) {
  1188. ret = INLINE_SYSCALL(fcntl, 3, handle->mcast.cli, F_SETFL,
  1189. *nonblocking ? O_NONBLOCK : 0);
  1190. if (IS_ERR(ret))
  1191. return unix_to_pal_error(ERRNO(ret));
  1192. *nonblocking = attr->nonblocking;
  1193. }
  1194. return 0;
  1195. }
  1196. struct handle_ops mcast_ops = {
  1197. .write = &mcast_send,
  1198. .read = &mcast_receive,
  1199. .close = &mcast_close,
  1200. .attrquerybyhdl = &mcast_attrquerybyhdl,
  1201. .attrsetbyhdl = &mcast_attrsetbyhdl,
  1202. };