modusocket.c 18 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547
  1. /*
  2. * This file is part of the MicroPython project, http://micropython.org/
  3. *
  4. * The MIT License (MIT)
  5. *
  6. * Copyright (c) 2018 Armink (armink.ztl@gmail.com)
  7. *
  8. * Permission is hereby granted, free of charge, to any person obtaining a copy
  9. * of this software and associated documentation files (the "Software"), to deal
  10. * in the Software without restriction, including without limitation the rights
  11. * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  12. * copies of the Software, and to permit persons to whom the Software is
  13. * furnished to do so, subject to the following conditions:
  14. *
  15. * The above copyright notice and this permission notice shall be included in
  16. * all copies or substantial portions of the Software.
  17. *
  18. * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  19. * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  20. * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  21. * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  22. * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  23. * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  24. * THE SOFTWARE.
  25. */
  26. #include "py/mpconfig.h"
  27. #if MICROPY_PY_USOCKET
  28. #include <stdio.h>
  29. #include <string.h>
  30. #include <stdio.h>
  31. #include <netdb.h>
  32. #include <sys/socket.h>
  33. #include <dfs_posix.h>
  34. #include "py/objtuple.h"
  35. #include "py/objlist.h"
  36. #include "py/runtime.h"
  37. #include "py/mperrno.h"
  38. #include "py/stream.h"
  39. #include "py/objstr.h"
  40. #include "py/builtin.h"
  41. #include "lib/netutils/netutils.h"
  42. #include "modnetwork.h"
  43. /******************************************************************************/
  44. // socket class
  45. STATIC const mp_obj_type_t socket_type;
  46. typedef struct _posix_socket_obj_t {
  47. mp_obj_base_t base;
  48. int fd;
  49. } posix_socket_obj_t;
  50. // constructor socket(family=AF_INET, type=SOCK_STREAM, proto=0)
  51. STATIC mp_obj_t socket_make_new(const mp_obj_type_t *type, size_t n_args, size_t n_kw, const mp_obj_t *args) {
  52. uint8_t domain = AF_INET;
  53. uint8_t style = SOCK_STREAM;
  54. uint8_t proto = 0;
  55. int fd = -1;
  56. posix_socket_obj_t *s = MP_OBJ_NULL;
  57. mp_arg_check_num(n_args, n_kw, 0, 4, false);
  58. if (n_args >= 1) {
  59. domain = mp_obj_get_int(args[0]);
  60. if (n_args >= 2) {
  61. style = mp_obj_get_int(args[1]);
  62. }
  63. }
  64. fd = socket(domain, style, proto);
  65. if (fd != -1) {
  66. s = m_new_obj_with_finaliser(posix_socket_obj_t);
  67. if (s) {
  68. s->base.type = (mp_obj_t) &socket_type;
  69. s->fd = fd;
  70. } else {
  71. rt_kprintf("m_new_obj_with_finaliser error.\n");
  72. mp_raise_OSError(MP_EAGAIN);
  73. }
  74. } else {
  75. rt_kprintf("Get fd error fd = -1\n");
  76. mp_raise_OSError(MP_EAGAIN);
  77. }
  78. return s;
  79. }
  80. // method socket.close()
  81. STATIC mp_obj_t socket_close(mp_obj_t self_in) {
  82. posix_socket_obj_t *self = self_in;
  83. if (fd_get(self->fd)) {
  84. closesocket(self->fd);
  85. } else {
  86. rt_kprintf("fd error,fd == NULL\n");
  87. }
  88. return mp_const_none;
  89. }
  90. STATIC MP_DEFINE_CONST_FUN_OBJ_1(socket_close_obj, socket_close);
  91. // method socket.bind(address)
  92. STATIC mp_obj_t socket_bind(mp_obj_t self_in, mp_obj_t addr_in) {
  93. posix_socket_obj_t *self = self_in;
  94. int _errno;
  95. struct sockaddr_in sockaddr;
  96. mp_obj_t *items;
  97. mp_obj_get_array_fixed_n(addr_in, 2, &items);
  98. sockaddr.sin_family = AF_INET;
  99. sockaddr.sin_port = htons(mp_obj_get_int(items[1]));
  100. const char *strip = mp_obj_str_get_str(items[0]);
  101. inet_aton((char * )strip, (struct in_addr* )&(sockaddr.sin_addr));
  102. memset(&(sockaddr.sin_zero), 0, sizeof(sockaddr.sin_zero));
  103. if ((_errno = bind(self->fd, (struct sockaddr *) &sockaddr, sizeof(struct sockaddr))) < 0) {
  104. mp_raise_OSError(_errno);
  105. }
  106. return mp_const_none;
  107. }
  108. STATIC MP_DEFINE_CONST_FUN_OBJ_2(socket_bind_obj, socket_bind);
  109. // method socket.listen(backlog)
  110. STATIC mp_obj_t socket_listen(mp_obj_t self_in, mp_obj_t backlog) {
  111. posix_socket_obj_t *self = self_in;
  112. int _errno;
  113. if ((_errno = listen(self->fd, mp_obj_get_int(backlog))) < 0) {
  114. mp_raise_OSError(_errno);
  115. }
  116. return mp_const_none;
  117. }
  118. STATIC MP_DEFINE_CONST_FUN_OBJ_2(socket_listen_obj, socket_listen);
  119. // method socket.accept()
  120. STATIC mp_obj_t socket_accept(mp_obj_t self_in) {
  121. posix_socket_obj_t *self = self_in;
  122. // create new socket object
  123. posix_socket_obj_t *socket2 = m_new_obj_with_finaliser(posix_socket_obj_t);
  124. socket2->base.type = (mp_obj_t) &socket_type;
  125. // accept incoming connection
  126. int new_client;
  127. struct sockaddr_in addr;
  128. socklen_t addrlen;
  129. if ((new_client = accept(self->fd, (struct sockaddr *) &addr, &addrlen)) < 0) {
  130. mp_raise_OSError(new_client);
  131. }
  132. socket2->fd = new_client;
  133. // make the return value
  134. mp_obj_tuple_t *client = mp_obj_new_tuple(2, NULL);
  135. client->items[0] = socket2;
  136. mp_obj_t tuple[2] = {
  137. tuple[0] = netutils_format_ipv4_addr((uint8_t *)&addr.sin_addr, NETUTILS_BIG),
  138. tuple[1] = mp_obj_new_int(ntohs(addr.sin_port)),
  139. };
  140. client->items[1] = mp_obj_new_tuple(2, tuple);
  141. return client;
  142. }
  143. STATIC MP_DEFINE_CONST_FUN_OBJ_1(socket_accept_obj, socket_accept);
  144. // method socket.connect(address)
  145. STATIC mp_obj_t socket_connect(mp_obj_t self_in, mp_obj_t addr_in) {
  146. posix_socket_obj_t *self = self_in;
  147. int _errno;
  148. mp_obj_t *items;
  149. struct sockaddr_in sockaddr;
  150. mp_obj_get_array_fixed_n(addr_in, 2, &items);
  151. sockaddr.sin_family = AF_INET;
  152. sockaddr.sin_port = htons(mp_obj_get_int(items[1]));
  153. const char *strip = mp_obj_str_get_str(items[0]);
  154. inet_aton((char * )strip, (struct in_addr* )&(sockaddr.sin_addr));
  155. memset(&(sockaddr.sin_zero), 0, sizeof(sockaddr.sin_zero));
  156. if ((_errno = connect(self->fd, (struct sockaddr *) &sockaddr, sizeof(struct sockaddr))) < 0) {
  157. mp_raise_OSError(_errno);
  158. }
  159. return mp_const_none;
  160. }
  161. STATIC MP_DEFINE_CONST_FUN_OBJ_2(socket_connect_obj, socket_connect);
  162. // method socket.send(bytes)
  163. STATIC mp_obj_t socket_send(mp_obj_t self_in, mp_obj_t buf_in) {
  164. posix_socket_obj_t *self = self_in;
  165. mp_buffer_info_t bufinfo;
  166. mp_get_buffer_raise(buf_in, &bufinfo, MP_BUFFER_READ);
  167. int _errno;
  168. _errno = send(self->fd, bufinfo.buf, bufinfo.len, 0);
  169. if (_errno == -1) {
  170. mp_raise_OSError(_errno);
  171. }
  172. return mp_obj_new_int_from_uint(_errno);
  173. }
  174. STATIC MP_DEFINE_CONST_FUN_OBJ_2(socket_send_obj, socket_send);
  175. //socket.write(buf)
  176. //Write the buffer of bytes to the socket.
  177. //This function will try to write all data to a socket (no short writes ).
  178. //This may be not possible with a non-blocking socket though, and returned value will be less than the length of buf.
  179. //Return value: number of bytes written.
  180. STATIC mp_obj_t socket_write(mp_obj_t self_in, mp_obj_t buf_in) {
  181. posix_socket_obj_t *self = self_in;
  182. mp_buffer_info_t bufinfo;
  183. mp_get_buffer_raise(buf_in, &bufinfo, MP_BUFFER_READ);
  184. int _errno;
  185. _errno = send(self->fd, bufinfo.buf, bufinfo.len, 0);
  186. if (_errno == -1) {
  187. mp_raise_OSError(_errno);
  188. }
  189. return mp_obj_new_int_from_uint(_errno);
  190. }
  191. STATIC MP_DEFINE_CONST_FUN_OBJ_2(socket_write_obj, socket_write);
  192. // method socket.recv(bufsize)
  193. STATIC mp_obj_t socket_recv(mp_obj_t self_in, mp_obj_t len_in) {
  194. posix_socket_obj_t *self = self_in;
  195. mp_int_t len = mp_obj_get_int(len_in);
  196. vstr_t vstr;
  197. vstr_init_len(&vstr, len);
  198. int ret;
  199. ret = recv(self->fd, (byte*)vstr.buf, len, 0);
  200. if (ret == -1) {
  201. mp_raise_OSError(ret);
  202. }
  203. if (ret == 0) {
  204. return mp_const_empty_bytes;
  205. }
  206. vstr.len = ret;
  207. return mp_obj_new_str_from_vstr(&mp_type_bytes, &vstr);
  208. }
  209. STATIC MP_DEFINE_CONST_FUN_OBJ_2(socket_recv_obj, socket_recv);
  210. // method socket.sendto(bytes, address)
  211. STATIC mp_obj_t socket_sendto(mp_obj_t self_in, mp_obj_t data_in, mp_obj_t addr_in) {
  212. posix_socket_obj_t *self = self_in;
  213. // get the data
  214. mp_buffer_info_t bufinfo;
  215. mp_get_buffer_raise(data_in, &bufinfo, MP_BUFFER_READ);
  216. mp_obj_t *items;
  217. struct sockaddr_in sockaddr;
  218. mp_obj_get_array_fixed_n(addr_in, 2, &items);
  219. sockaddr.sin_family = AF_INET;
  220. sockaddr.sin_port = htons(mp_obj_get_int(items[1]));
  221. const char *strip = mp_obj_str_get_str(items[0]);
  222. inet_aton((char * )strip, (struct in_addr* )&(sockaddr.sin_addr));
  223. memset(&(sockaddr.sin_zero), 0, sizeof(sockaddr.sin_zero));
  224. mp_int_t ret = sendto(self->fd, bufinfo.buf, bufinfo.len, MSG_DONTWAIT, (struct sockaddr *) &sockaddr,
  225. sizeof(struct sockaddr));
  226. if (ret == -1) {
  227. mp_raise_OSError(ret);
  228. }
  229. return mp_obj_new_int(ret);
  230. }
  231. STATIC MP_DEFINE_CONST_FUN_OBJ_3(socket_sendto_obj, socket_sendto);
  232. // method socket.recvfrom(bufsize)
  233. STATIC mp_obj_t socket_recvfrom(mp_obj_t self_in, mp_obj_t len_in) {
  234. posix_socket_obj_t *self = self_in;
  235. vstr_t vstr;
  236. vstr_init_len(&vstr, mp_obj_get_int(len_in));
  237. byte ip[4];
  238. mp_uint_t port;
  239. int ret;
  240. struct sockaddr_in sockaddr;
  241. uint32_t addr_len;
  242. ret = recvfrom(self->fd, vstr.buf, vstr.len, MSG_DONTWAIT, (struct sockaddr *)&sockaddr, &addr_len);
  243. if (ret == -1) {
  244. mp_raise_OSError(ret);
  245. }
  246. mp_obj_t tuple[2];
  247. if (ret == 0) {
  248. tuple[0] = mp_const_empty_bytes;
  249. } else {
  250. vstr.len = ret;
  251. tuple[0] = mp_obj_new_str_from_vstr(&mp_type_bytes, &vstr);
  252. }
  253. tuple[1] = netutils_format_inet_addr(ip, port, NETUTILS_BIG);
  254. return mp_obj_new_tuple(2, tuple);
  255. }
  256. STATIC MP_DEFINE_CONST_FUN_OBJ_2(socket_recvfrom_obj, socket_recvfrom);
  257. // method socket.setsockopt(level, optname, value)
  258. STATIC mp_obj_t socket_setsockopt(size_t n_args, const mp_obj_t *args) {
  259. posix_socket_obj_t *self = args[0];
  260. mp_int_t level = mp_obj_get_int(args[1]);
  261. mp_int_t opt = mp_obj_get_int(args[2]);
  262. const void *optval;
  263. mp_uint_t optlen;
  264. mp_int_t val;
  265. if (mp_obj_is_integer(args[3])) {
  266. val = mp_obj_get_int_truncated(args[3]);
  267. optval = &val;
  268. optlen = sizeof(val);
  269. } else {
  270. mp_buffer_info_t bufinfo;
  271. mp_get_buffer_raise(args[3], &bufinfo, MP_BUFFER_READ);
  272. optval = bufinfo.buf;
  273. optlen = bufinfo.len;
  274. }
  275. int _errno;
  276. if ((_errno = setsockopt(self->fd, level, opt, optval, optlen)) < 0) {
  277. mp_raise_OSError(_errno);
  278. }
  279. return mp_const_none;
  280. }
  281. STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(socket_setsockopt_obj, 4, 4, socket_setsockopt);
  282. // method socket.settimeout(value)
  283. // timeout=0 means non-blocking
  284. // timeout=None means blocking
  285. // otherwise, timeout is in seconds
  286. STATIC mp_obj_t socket_settimeout(mp_obj_t self_in, mp_obj_t timeout_in) {
  287. posix_socket_obj_t *self = self_in;
  288. int timeout;
  289. if (timeout_in == mp_const_none) {
  290. timeout = -1;
  291. } else {
  292. #if MICROPY_PY_BUILTINS_FLOAT
  293. timeout = 1000 * mp_obj_get_float(timeout_in);
  294. #else
  295. timeout = 1000 * mp_obj_get_int(timeout_in);
  296. #endif
  297. }
  298. int _errno;
  299. int flags = ioctl(self->fd, F_GETFL, NULL);
  300. if (timeout < 0) {
  301. _errno = ioctl(self->fd, F_SETFL, (void *)(flags & (~O_NONBLOCK)));
  302. } else if (timeout > 0) {
  303. //TODO
  304. MP_RTT_NOT_IMPL_PRINT;
  305. } else {
  306. _errno = ioctl(self->fd, F_SETFL, (void *)(flags | O_NONBLOCK));
  307. }
  308. if (_errno < 0) {
  309. mp_raise_OSError(_errno);
  310. }
  311. return mp_const_none;
  312. }
  313. STATIC MP_DEFINE_CONST_FUN_OBJ_2(socket_settimeout_obj, socket_settimeout);
  314. // method socket.setblocking(flag)
  315. STATIC mp_obj_t socket_setblocking(mp_obj_t self_in, mp_obj_t blocking) {
  316. if (mp_obj_is_true(blocking)) {
  317. return socket_settimeout(self_in, mp_const_none);
  318. } else {
  319. return socket_settimeout(self_in, MP_OBJ_NEW_SMALL_INT(0));
  320. }
  321. }
  322. STATIC MP_DEFINE_CONST_FUN_OBJ_2(socket_setblocking_obj, socket_setblocking);
  323. STATIC const mp_rom_map_elem_t socket_locals_dict_table[] = {
  324. { MP_ROM_QSTR(MP_QSTR___del__), MP_ROM_PTR(&socket_close_obj) },
  325. { MP_ROM_QSTR(MP_QSTR_close), MP_ROM_PTR(&socket_close_obj) },
  326. { MP_ROM_QSTR(MP_QSTR_bind), MP_ROM_PTR(&socket_bind_obj) },
  327. { MP_ROM_QSTR(MP_QSTR_listen), MP_ROM_PTR(&socket_listen_obj) },
  328. { MP_ROM_QSTR(MP_QSTR_accept), MP_ROM_PTR(&socket_accept_obj) },
  329. { MP_ROM_QSTR(MP_QSTR_connect), MP_ROM_PTR(&socket_connect_obj) },
  330. { MP_ROM_QSTR(MP_QSTR_send), MP_ROM_PTR(&socket_send_obj) },
  331. { MP_ROM_QSTR(MP_QSTR_recv), MP_ROM_PTR(&socket_recv_obj) },
  332. { MP_ROM_QSTR(MP_QSTR_sendto), MP_ROM_PTR(&socket_sendto_obj) },
  333. { MP_ROM_QSTR(MP_QSTR_recvfrom), MP_ROM_PTR(&socket_recvfrom_obj) },
  334. { MP_ROM_QSTR(MP_QSTR_setsockopt), MP_ROM_PTR(&socket_setsockopt_obj) },
  335. { MP_ROM_QSTR(MP_QSTR_settimeout), MP_ROM_PTR(&socket_settimeout_obj) },
  336. { MP_ROM_QSTR(MP_QSTR_setblocking), MP_ROM_PTR(&socket_setblocking_obj) },
  337. { MP_ROM_QSTR(MP_QSTR_write), MP_ROM_PTR(&mp_stream_write_obj) },
  338. { MP_ROM_QSTR(MP_QSTR_readline), MP_ROM_PTR(&mp_stream_unbuffered_readline_obj) },
  339. { MP_ROM_QSTR(MP_QSTR_read), MP_ROM_PTR(&mp_stream_read_obj) },
  340. { MP_ROM_QSTR(MP_QSTR_readinto), MP_ROM_PTR(&mp_stream_readinto_obj) },
  341. };
  342. STATIC MP_DEFINE_CONST_DICT(socket_locals_dict, socket_locals_dict_table);
  343. STATIC mp_uint_t socket_ioctl(mp_obj_t self_in, mp_uint_t request, mp_uint_t arg, int *errcode) {
  344. posix_socket_obj_t * socket = self_in;
  345. if (request == MP_STREAM_POLL) {
  346. fd_set rfds; FD_ZERO(&rfds);
  347. fd_set wfds; FD_ZERO(&wfds);
  348. fd_set efds; FD_ZERO(&efds);
  349. struct timeval timeout = { .tv_sec = 0, .tv_usec = 0 };
  350. if (arg & MP_STREAM_POLL_RD) FD_SET(socket->fd, &rfds);
  351. if (arg & MP_STREAM_POLL_WR) FD_SET(socket->fd, &wfds);
  352. if (arg & MP_STREAM_POLL_HUP) FD_SET(socket->fd, &efds);
  353. int r = select((socket->fd)+1, &rfds, &wfds, &efds, &timeout);
  354. if (r < 0) {
  355. *errcode = MP_EIO;
  356. return MP_STREAM_ERROR;
  357. }
  358. mp_uint_t ret = 0;
  359. if (FD_ISSET(socket->fd, &rfds)) ret |= MP_STREAM_POLL_RD;
  360. if (FD_ISSET(socket->fd, &wfds)) ret |= MP_STREAM_POLL_WR;
  361. if (FD_ISSET(socket->fd, &efds)) ret |= MP_STREAM_POLL_HUP;
  362. return ret;
  363. }else{
  364. return ioctl(socket->fd, request, (void *) arg);
  365. }
  366. *errcode = MP_EINVAL;
  367. return MP_STREAM_ERROR;
  368. }
  369. STATIC mp_uint_t stream_read(mp_obj_t self_in, void *buf, mp_uint_t size, int *errcode) {
  370. posix_socket_obj_t *self = self_in;
  371. int err = recv(self->fd, buf, size, 0);
  372. if (err < 0) {
  373. *errcode = err;
  374. return -1;
  375. }
  376. return err;
  377. }
  378. STATIC mp_uint_t steam_write(mp_obj_t self_in, const void *buf, mp_uint_t size, int *errcode) {
  379. posix_socket_obj_t *self = self_in;
  380. int err = send(self->fd, buf, size, 0);
  381. if (err < 0) {
  382. *errcode = err;
  383. return -1;
  384. }
  385. return err;
  386. }
  387. STATIC const mp_stream_p_t socket_stream_p = {
  388. .read = stream_read,
  389. .write = steam_write,
  390. .ioctl = socket_ioctl,
  391. .is_text = false, };
  392. STATIC const mp_obj_type_t socket_type = {
  393. { &mp_type_type },
  394. .name = MP_QSTR_socket,
  395. .make_new = socket_make_new,
  396. .protocol = &socket_stream_p,
  397. .locals_dict = (mp_obj_dict_t*)&socket_locals_dict,
  398. };
  399. /******************************************************************************/
  400. // usocket module
  401. // function usocket.getaddrinfo(host, port)
  402. STATIC mp_obj_t mod_usocket_getaddrinfo(uint n_args, const mp_obj_t *arg) {
  403. if (n_args == 3) {
  404. rt_kprintf("Usage: usocket.getaddrinfo(\"host\",port,x,x). n_args must be 2 or 4.\n");
  405. rt_kprintf("argument num/types mismatch\n");
  406. mp_raise_OSError(MP_EAGAIN);
  407. return mp_const_none;
  408. } else if (n_args == 2 || n_args == 4) {
  409. size_t hlen;
  410. int ret;
  411. const char *host = mp_obj_str_get_data(arg[0], &hlen);
  412. mp_int_t port = mp_obj_get_int(arg[1]);
  413. struct addrinfo hint, *res = NULL;
  414. memset(&hint, 0, sizeof(hint));
  415. ret = getaddrinfo(host, NULL, &hint, &res);
  416. if (ret != 0) {
  417. rt_kprintf("getaddrinfo err: %d '%s'\n", ret, host);
  418. nlr_raise(mp_obj_new_exception_msg(&mp_type_OSError, "no available netif"));
  419. }
  420. mp_obj_tuple_t *tuple = mp_obj_new_tuple(5, NULL);
  421. tuple->items[0] = MP_OBJ_NEW_SMALL_INT(MOD_NETWORK_AF_INET);
  422. tuple->items[1] = MP_OBJ_NEW_SMALL_INT(MOD_NETWORK_SOCK_STREAM);
  423. tuple->items[2] = MP_OBJ_NEW_SMALL_INT(0);
  424. tuple->items[3] = MP_OBJ_NEW_QSTR(MP_QSTR_);
  425. mp_obj_t tuple_addr[2] = {
  426. tuple_addr[0] = netutils_format_ipv4_addr(((res->ai_addr->sa_data) + 2),NETUTILS_BIG),
  427. tuple_addr[1] = mp_obj_new_int(port),
  428. };
  429. tuple->items[4] = mp_obj_new_tuple(2, tuple_addr);
  430. freeaddrinfo(res);
  431. return mp_obj_new_list(1, (mp_obj_t*) &tuple);
  432. }
  433. }
  434. STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(mod_usocket_getaddrinfo_obj, 2, 4, mod_usocket_getaddrinfo);
  435. STATIC const mp_rom_map_elem_t mp_module_usocket_globals_table[] = {
  436. { MP_ROM_QSTR(MP_QSTR___name__), MP_ROM_QSTR(MP_QSTR_usocket) },
  437. { MP_ROM_QSTR(MP_QSTR_socket), MP_ROM_PTR(&socket_type) },
  438. { MP_ROM_QSTR(MP_QSTR_getaddrinfo), MP_ROM_PTR(&mod_usocket_getaddrinfo_obj) },
  439. // class constants
  440. { MP_ROM_QSTR(MP_QSTR_AF_INET), MP_ROM_INT(MOD_NETWORK_AF_INET) },
  441. { MP_ROM_QSTR(MP_QSTR_AF_INET6), MP_ROM_INT(MOD_NETWORK_AF_INET6) },
  442. { MP_ROM_QSTR(MP_QSTR_SOCK_STREAM), MP_ROM_INT(MOD_NETWORK_SOCK_STREAM) },
  443. { MP_ROM_QSTR(MP_QSTR_SOCK_DGRAM), MP_ROM_INT(MOD_NETWORK_SOCK_DGRAM) },
  444. { MP_ROM_QSTR(MP_QSTR_SOCK_RAW), MP_ROM_INT(MOD_NETWORK_SOCK_RAW) },
  445. { MP_ROM_QSTR(MP_QSTR_SOL_SOCKET), MP_ROM_INT(SOL_SOCKET) },
  446. { MP_ROM_QSTR(MP_QSTR_SO_REUSEADDR), MP_ROM_INT(SO_REUSEADDR) },
  447. /*
  448. { MP_ROM_QSTR(MP_QSTR_IPPROTO_IP), MP_ROM_INT(MOD_NETWORK_IPPROTO_IP) },
  449. { MP_ROM_QSTR(MP_QSTR_IPPROTO_ICMP), MP_ROM_INT(MOD_NETWORK_IPPROTO_ICMP) },
  450. { MP_ROM_QSTR(MP_QSTR_IPPROTO_IPV4), MP_ROM_INT(MOD_NETWORK_IPPROTO_IPV4) },
  451. { MP_ROM_QSTR(MP_QSTR_IPPROTO_TCP), MP_ROM_INT(MOD_NETWORK_IPPROTO_TCP) },
  452. { MP_ROM_QSTR(MP_QSTR_IPPROTO_UDP), MP_ROM_INT(MOD_NETWORK_IPPROTO_UDP) },
  453. { MP_ROM_QSTR(MP_QSTR_IPPROTO_IPV6), MP_ROM_INT(MOD_NETWORK_IPPROTO_IPV6) },
  454. { MP_ROM_QSTR(MP_QSTR_IPPROTO_RAW), MP_ROM_INT(MOD_NETWORK_IPPROTO_RAW) },
  455. */
  456. };
  457. STATIC MP_DEFINE_CONST_DICT(mp_module_usocket_globals, mp_module_usocket_globals_table);
  458. const mp_obj_module_t mp_module_usocket = {
  459. .base = { &mp_type_module },
  460. .globals = (mp_obj_dict_t*)&mp_module_usocket_globals,
  461. };
  462. #endif // MICROPY_PY_USOCKET