modnwcc3k.c 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600
  1. /*
  2. * This file is part of the MicroPython project, http://micropython.org/
  3. *
  4. * The MIT License (MIT)
  5. *
  6. * Copyright (c) 2014 Damien P. George
  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 <string.h>
  27. #include <stdarg.h>
  28. // CC3000 defines its own ENOBUFS (different to standard one!)
  29. #undef ENOBUFS
  30. #include "py/objtuple.h"
  31. #include "py/objlist.h"
  32. #include "py/stream.h"
  33. #include "py/runtime.h"
  34. #include "py/mperrno.h"
  35. #include "py/mphal.h"
  36. #include "lib/netutils/netutils.h"
  37. #include "modnetwork.h"
  38. #include "pin.h"
  39. #include "spi.h"
  40. #include "hci.h"
  41. #include "socket.h"
  42. #include "inet_ntop.h"
  43. #include "inet_pton.h"
  44. #include "ccspi.h"
  45. #include "wlan.h"
  46. #include "nvmem.h"
  47. #include "netapp.h"
  48. #include "patch_prog.h"
  49. #define MAX_ADDRSTRLEN (128)
  50. #define MAX_RX_PACKET (CC3000_RX_BUFFER_SIZE-CC3000_MINIMAL_RX_SIZE-1)
  51. #define MAX_TX_PACKET (CC3000_TX_BUFFER_SIZE-CC3000_MINIMAL_TX_SIZE-1)
  52. #define MAKE_SOCKADDR(addr, ip, port) \
  53. sockaddr addr; \
  54. addr.sa_family = AF_INET; \
  55. addr.sa_data[0] = port >> 8; \
  56. addr.sa_data[1] = port; \
  57. addr.sa_data[2] = ip[0]; \
  58. addr.sa_data[3] = ip[1]; \
  59. addr.sa_data[4] = ip[2]; \
  60. addr.sa_data[5] = ip[3];
  61. #define UNPACK_SOCKADDR(addr, ip, port) \
  62. port = (addr.sa_data[0] << 8) | addr.sa_data[1]; \
  63. ip[0] = addr.sa_data[2]; \
  64. ip[1] = addr.sa_data[3]; \
  65. ip[2] = addr.sa_data[4]; \
  66. ip[3] = addr.sa_data[5];
  67. STATIC int cc3k_socket_ioctl(mod_network_socket_obj_t *socket, mp_uint_t request, mp_uint_t arg, int *_errno);
  68. int CC3000_EXPORT(errno); // for cc3000 driver
  69. STATIC volatile uint32_t fd_closed_state = 0;
  70. STATIC volatile bool wlan_connected = false;
  71. STATIC volatile bool ip_obtained = false;
  72. STATIC int cc3k_get_fd_closed_state(int fd) {
  73. return fd_closed_state & (1 << fd);
  74. }
  75. STATIC void cc3k_set_fd_closed_state(int fd) {
  76. fd_closed_state |= 1 << fd;
  77. }
  78. STATIC void cc3k_reset_fd_closed_state(int fd) {
  79. fd_closed_state &= ~(1 << fd);
  80. }
  81. STATIC void cc3k_callback(long event_type, char *data, unsigned char length) {
  82. switch (event_type) {
  83. case HCI_EVNT_WLAN_UNSOL_CONNECT:
  84. wlan_connected = true;
  85. break;
  86. case HCI_EVNT_WLAN_UNSOL_DISCONNECT:
  87. // link down
  88. wlan_connected = false;
  89. ip_obtained = false;
  90. break;
  91. case HCI_EVNT_WLAN_UNSOL_DHCP:
  92. ip_obtained = true;
  93. break;
  94. case HCI_EVNT_BSD_TCP_CLOSE_WAIT:
  95. // mark socket for closure
  96. cc3k_set_fd_closed_state(data[0]);
  97. break;
  98. }
  99. }
  100. STATIC int cc3k_gethostbyname(mp_obj_t nic, const char *name, mp_uint_t len, uint8_t *out_ip) {
  101. uint32_t ip;
  102. // CC3000 gethostbyname is unreliable and usually returns -95 on first call
  103. for (int retry = 5; CC3000_EXPORT(gethostbyname)((char*)name, len, &ip) < 0; retry--) {
  104. if (retry == 0 || CC3000_EXPORT(errno) != -95) {
  105. return CC3000_EXPORT(errno);
  106. }
  107. mp_hal_delay_ms(50);
  108. }
  109. if (ip == 0) {
  110. // unknown host
  111. return -2;
  112. }
  113. out_ip[0] = ip >> 24;
  114. out_ip[1] = ip >> 16;
  115. out_ip[2] = ip >> 8;
  116. out_ip[3] = ip;
  117. return 0;
  118. }
  119. STATIC int cc3k_socket_socket(mod_network_socket_obj_t *socket, int *_errno) {
  120. if (socket->u_param.domain != MOD_NETWORK_AF_INET) {
  121. *_errno = MP_EAFNOSUPPORT;
  122. return -1;
  123. }
  124. mp_uint_t type;
  125. switch (socket->u_param.type) {
  126. case MOD_NETWORK_SOCK_STREAM: type = SOCK_STREAM; break;
  127. case MOD_NETWORK_SOCK_DGRAM: type = SOCK_DGRAM; break;
  128. case MOD_NETWORK_SOCK_RAW: type = SOCK_RAW; break;
  129. default: *_errno = MP_EINVAL; return -1;
  130. }
  131. // open socket
  132. int fd = CC3000_EXPORT(socket)(AF_INET, type, 0);
  133. if (fd < 0) {
  134. *_errno = CC3000_EXPORT(errno);
  135. return -1;
  136. }
  137. // clear socket state
  138. cc3k_reset_fd_closed_state(fd);
  139. // store state of this socket
  140. socket->u_state = fd;
  141. // make accept blocking by default
  142. int optval = SOCK_OFF;
  143. socklen_t optlen = sizeof(optval);
  144. CC3000_EXPORT(setsockopt)(socket->u_state, SOL_SOCKET, SOCKOPT_ACCEPT_NONBLOCK, &optval, optlen);
  145. return 0;
  146. }
  147. STATIC void cc3k_socket_close(mod_network_socket_obj_t *socket) {
  148. CC3000_EXPORT(closesocket)(socket->u_state);
  149. }
  150. STATIC int cc3k_socket_bind(mod_network_socket_obj_t *socket, byte *ip, mp_uint_t port, int *_errno) {
  151. MAKE_SOCKADDR(addr, ip, port)
  152. int ret = CC3000_EXPORT(bind)(socket->u_state, &addr, sizeof(addr));
  153. if (ret != 0) {
  154. *_errno = ret;
  155. return -1;
  156. }
  157. return 0;
  158. }
  159. STATIC int cc3k_socket_listen(mod_network_socket_obj_t *socket, mp_int_t backlog, int *_errno) {
  160. int ret = CC3000_EXPORT(listen)(socket->u_state, backlog);
  161. if (ret != 0) {
  162. *_errno = ret;
  163. return -1;
  164. }
  165. return 0;
  166. }
  167. STATIC int cc3k_socket_accept(mod_network_socket_obj_t *socket, mod_network_socket_obj_t *socket2, byte *ip, mp_uint_t *port, int *_errno) {
  168. // accept incoming connection
  169. int fd;
  170. sockaddr addr;
  171. socklen_t addr_len = sizeof(addr);
  172. if ((fd = CC3000_EXPORT(accept)(socket->u_state, &addr, &addr_len)) < 0) {
  173. if (fd == SOC_IN_PROGRESS) {
  174. *_errno = MP_EAGAIN;
  175. } else {
  176. *_errno = -fd;
  177. }
  178. return -1;
  179. }
  180. // clear socket state
  181. cc3k_reset_fd_closed_state(fd);
  182. // store state in new socket object
  183. socket2->u_state = fd;
  184. // return ip and port
  185. // it seems CC3000 returns little endian for accept??
  186. //UNPACK_SOCKADDR(addr, ip, *port);
  187. *port = (addr.sa_data[1] << 8) | addr.sa_data[0];
  188. ip[3] = addr.sa_data[2];
  189. ip[2] = addr.sa_data[3];
  190. ip[1] = addr.sa_data[4];
  191. ip[0] = addr.sa_data[5];
  192. return 0;
  193. }
  194. STATIC int cc3k_socket_connect(mod_network_socket_obj_t *socket, byte *ip, mp_uint_t port, int *_errno) {
  195. MAKE_SOCKADDR(addr, ip, port)
  196. int ret = CC3000_EXPORT(connect)(socket->u_state, &addr, sizeof(addr));
  197. if (ret != 0) {
  198. *_errno = CC3000_EXPORT(errno);
  199. return -1;
  200. }
  201. return 0;
  202. }
  203. STATIC mp_uint_t cc3k_socket_send(mod_network_socket_obj_t *socket, const byte *buf, mp_uint_t len, int *_errno) {
  204. if (cc3k_get_fd_closed_state(socket->u_state)) {
  205. CC3000_EXPORT(closesocket)(socket->u_state);
  206. *_errno = MP_EPIPE;
  207. return -1;
  208. }
  209. // CC3K does not handle fragmentation, and will overflow,
  210. // split the packet into smaller ones and send them out.
  211. mp_int_t bytes = 0;
  212. while (bytes < len) {
  213. int n = MIN((len - bytes), MAX_TX_PACKET);
  214. n = CC3000_EXPORT(send)(socket->u_state, (uint8_t*)buf + bytes, n, 0);
  215. if (n <= 0) {
  216. *_errno = CC3000_EXPORT(errno);
  217. return -1;
  218. }
  219. bytes += n;
  220. }
  221. return bytes;
  222. }
  223. STATIC mp_uint_t cc3k_socket_recv(mod_network_socket_obj_t *socket, byte *buf, mp_uint_t len, int *_errno) {
  224. // check the socket is open
  225. if (cc3k_get_fd_closed_state(socket->u_state)) {
  226. // socket is closed, but CC3000 may have some data remaining in buffer, so check
  227. fd_set rfds;
  228. FD_ZERO(&rfds);
  229. FD_SET(socket->u_state, &rfds);
  230. cc3000_timeval tv;
  231. tv.tv_sec = 0;
  232. tv.tv_usec = 1;
  233. int nfds = CC3000_EXPORT(select)(socket->u_state + 1, &rfds, NULL, NULL, &tv);
  234. if (nfds == -1 || !FD_ISSET(socket->u_state, &rfds)) {
  235. // no data waiting, so close socket and return 0 data
  236. CC3000_EXPORT(closesocket)(socket->u_state);
  237. return 0;
  238. }
  239. }
  240. // cap length at MAX_RX_PACKET
  241. len = MIN(len, MAX_RX_PACKET);
  242. // do the recv
  243. int ret = CC3000_EXPORT(recv)(socket->u_state, buf, len, 0);
  244. if (ret < 0) {
  245. *_errno = CC3000_EXPORT(errno);
  246. return -1;
  247. }
  248. return ret;
  249. }
  250. STATIC mp_uint_t cc3k_socket_sendto(mod_network_socket_obj_t *socket, const byte *buf, mp_uint_t len, byte *ip, mp_uint_t port, int *_errno) {
  251. MAKE_SOCKADDR(addr, ip, port)
  252. int ret = CC3000_EXPORT(sendto)(socket->u_state, (byte*)buf, len, 0, (sockaddr*)&addr, sizeof(addr));
  253. if (ret < 0) {
  254. *_errno = CC3000_EXPORT(errno);
  255. return -1;
  256. }
  257. return ret;
  258. }
  259. STATIC mp_uint_t cc3k_socket_recvfrom(mod_network_socket_obj_t *socket, byte *buf, mp_uint_t len, byte *ip, mp_uint_t *port, int *_errno) {
  260. sockaddr addr;
  261. socklen_t addr_len = sizeof(addr);
  262. mp_int_t ret = CC3000_EXPORT(recvfrom)(socket->u_state, buf, len, 0, &addr, &addr_len);
  263. if (ret < 0) {
  264. *_errno = CC3000_EXPORT(errno);
  265. return -1;
  266. }
  267. UNPACK_SOCKADDR(addr, ip, *port);
  268. return ret;
  269. }
  270. STATIC int cc3k_socket_setsockopt(mod_network_socket_obj_t *socket, mp_uint_t level, mp_uint_t opt, const void *optval, mp_uint_t optlen, int *_errno) {
  271. int ret = CC3000_EXPORT(setsockopt)(socket->u_state, level, opt, optval, optlen);
  272. if (ret < 0) {
  273. *_errno = CC3000_EXPORT(errno);
  274. return -1;
  275. }
  276. return 0;
  277. }
  278. STATIC int cc3k_socket_settimeout(mod_network_socket_obj_t *socket, mp_uint_t timeout_ms, int *_errno) {
  279. int ret;
  280. if (timeout_ms == 0 || timeout_ms == -1) {
  281. int optval;
  282. socklen_t optlen = sizeof(optval);
  283. if (timeout_ms == 0) {
  284. // set non-blocking mode
  285. optval = SOCK_ON;
  286. } else {
  287. // set blocking mode
  288. optval = SOCK_OFF;
  289. }
  290. ret = CC3000_EXPORT(setsockopt)(socket->u_state, SOL_SOCKET, SOCKOPT_RECV_NONBLOCK, &optval, optlen);
  291. if (ret == 0) {
  292. ret = CC3000_EXPORT(setsockopt)(socket->u_state, SOL_SOCKET, SOCKOPT_ACCEPT_NONBLOCK, &optval, optlen);
  293. }
  294. } else {
  295. // set timeout
  296. socklen_t optlen = sizeof(timeout_ms);
  297. ret = CC3000_EXPORT(setsockopt)(socket->u_state, SOL_SOCKET, SOCKOPT_RECV_TIMEOUT, &timeout_ms, optlen);
  298. }
  299. if (ret != 0) {
  300. *_errno = CC3000_EXPORT(errno);
  301. return -1;
  302. }
  303. return 0;
  304. }
  305. STATIC int cc3k_socket_ioctl(mod_network_socket_obj_t *socket, mp_uint_t request, mp_uint_t arg, int *_errno) {
  306. mp_uint_t ret;
  307. if (request == MP_STREAM_POLL) {
  308. mp_uint_t flags = arg;
  309. ret = 0;
  310. int fd = socket->u_state;
  311. // init fds
  312. fd_set rfds, wfds, xfds;
  313. FD_ZERO(&rfds);
  314. FD_ZERO(&wfds);
  315. FD_ZERO(&xfds);
  316. // set fds if needed
  317. if (flags & MP_STREAM_POLL_RD) {
  318. FD_SET(fd, &rfds);
  319. // A socked that just closed is available for reading. A call to
  320. // recv() returns 0 which is consistent with BSD.
  321. if (cc3k_get_fd_closed_state(fd)) {
  322. ret |= MP_STREAM_POLL_RD;
  323. }
  324. }
  325. if (flags & MP_STREAM_POLL_WR) {
  326. FD_SET(fd, &wfds);
  327. }
  328. if (flags & MP_STREAM_POLL_HUP) {
  329. FD_SET(fd, &xfds);
  330. }
  331. // call cc3000 select with minimum timeout
  332. cc3000_timeval tv;
  333. tv.tv_sec = 0;
  334. tv.tv_usec = 1;
  335. int nfds = CC3000_EXPORT(select)(fd + 1, &rfds, &wfds, &xfds, &tv);
  336. // check for error
  337. if (nfds == -1) {
  338. *_errno = CC3000_EXPORT(errno);
  339. return -1;
  340. }
  341. // check return of select
  342. if (FD_ISSET(fd, &rfds)) {
  343. ret |= MP_STREAM_POLL_RD;
  344. }
  345. if (FD_ISSET(fd, &wfds)) {
  346. ret |= MP_STREAM_POLL_WR;
  347. }
  348. if (FD_ISSET(fd, &xfds)) {
  349. ret |= MP_STREAM_POLL_HUP;
  350. }
  351. } else {
  352. *_errno = MP_EINVAL;
  353. ret = -1;
  354. }
  355. return ret;
  356. }
  357. /******************************************************************************/
  358. // MicroPython bindings; CC3K class
  359. typedef struct _cc3k_obj_t {
  360. mp_obj_base_t base;
  361. } cc3k_obj_t;
  362. STATIC const cc3k_obj_t cc3k_obj = {{(mp_obj_type_t*)&mod_network_nic_type_cc3k}};
  363. // \classmethod \constructor(spi, pin_cs, pin_en, pin_irq)
  364. // Initialise the CC3000 using the given SPI bus and pins and return a CC3K object.
  365. //
  366. // Note: pins were originally hard-coded to:
  367. // PYBv1.0: init(pyb.SPI(2), pyb.Pin.board.Y5, pyb.Pin.board.Y4, pyb.Pin.board.Y3)
  368. // [SPI on Y position; Y6=B13=SCK, Y7=B14=MISO, Y8=B15=MOSI]
  369. //
  370. // STM32F4DISC: init(pyb.SPI(2), pyb.Pin.cpu.A15, pyb.Pin.cpu.B10, pyb.Pin.cpu.B11)
  371. STATIC mp_obj_t cc3k_make_new(const mp_obj_type_t *type, size_t n_args, size_t n_kw, const mp_obj_t *args) {
  372. // check arguments
  373. mp_arg_check_num(n_args, n_kw, 4, 4, false);
  374. // set the pins to use
  375. SpiInit(
  376. spi_from_mp_obj(args[0])->spi,
  377. pin_find(args[1]),
  378. pin_find(args[2]),
  379. pin_find(args[3])
  380. );
  381. // initialize and start the module
  382. wlan_init(cc3k_callback, NULL, NULL, NULL,
  383. ReadWlanInterruptPin, SpiResumeSpi, SpiPauseSpi, WriteWlanPin);
  384. if (wlan_start(0) != 0) {
  385. nlr_raise(mp_obj_new_exception_msg(&mp_type_OSError, "failed to init CC3000 module"));
  386. }
  387. // set connection policy. this should be called explicitly by the user
  388. // wlan_ioctl_set_connection_policy(0, 0, 0);
  389. // Mask out all non-required events from the CC3000
  390. wlan_set_event_mask(HCI_EVNT_WLAN_KEEPALIVE|
  391. HCI_EVNT_WLAN_UNSOL_INIT|
  392. HCI_EVNT_WLAN_ASYNC_PING_REPORT|
  393. HCI_EVNT_WLAN_ASYNC_SIMPLE_CONFIG_DONE);
  394. // register with network module
  395. mod_network_register_nic((mp_obj_t)&cc3k_obj);
  396. return (mp_obj_t)&cc3k_obj;
  397. }
  398. // method connect(ssid, key=None, *, security=WPA2, bssid=None)
  399. STATIC mp_obj_t cc3k_connect(size_t n_args, const mp_obj_t *pos_args, mp_map_t *kw_args) {
  400. static const mp_arg_t allowed_args[] = {
  401. { MP_QSTR_ssid, MP_ARG_REQUIRED | MP_ARG_OBJ, {.u_obj = MP_OBJ_NULL} },
  402. { MP_QSTR_key, MP_ARG_OBJ, {.u_obj = mp_const_none} },
  403. { MP_QSTR_security, MP_ARG_KW_ONLY | MP_ARG_INT, {.u_int = WLAN_SEC_WPA2} },
  404. { MP_QSTR_bssid, MP_ARG_KW_ONLY | MP_ARG_OBJ, {.u_obj = mp_const_none} },
  405. };
  406. // parse args
  407. mp_arg_val_t args[MP_ARRAY_SIZE(allowed_args)];
  408. mp_arg_parse_all(n_args - 1, pos_args + 1, kw_args, MP_ARRAY_SIZE(allowed_args), allowed_args, args);
  409. // get ssid
  410. size_t ssid_len;
  411. const char *ssid = mp_obj_str_get_data(args[0].u_obj, &ssid_len);
  412. // get key and sec
  413. size_t key_len = 0;
  414. const char *key = NULL;
  415. mp_uint_t sec = WLAN_SEC_UNSEC;
  416. if (args[1].u_obj != mp_const_none) {
  417. key = mp_obj_str_get_data(args[1].u_obj, &key_len);
  418. sec = args[2].u_int;
  419. }
  420. // get bssid
  421. const char *bssid = NULL;
  422. if (args[3].u_obj != mp_const_none) {
  423. bssid = mp_obj_str_get_str(args[3].u_obj);
  424. }
  425. // connect to AP
  426. if (wlan_connect(sec, (char*)ssid, ssid_len, (uint8_t*)bssid, (uint8_t*)key, key_len) != 0) {
  427. nlr_raise(mp_obj_new_exception_msg_varg(&mp_type_OSError, "could not connect to ssid=%s, sec=%d, key=%s\n", ssid, sec, key));
  428. }
  429. return mp_const_none;
  430. }
  431. STATIC MP_DEFINE_CONST_FUN_OBJ_KW(cc3k_connect_obj, 1, cc3k_connect);
  432. STATIC mp_obj_t cc3k_disconnect(mp_obj_t self_in) {
  433. // should we check return value?
  434. wlan_disconnect();
  435. return mp_const_none;
  436. }
  437. STATIC MP_DEFINE_CONST_FUN_OBJ_1(cc3k_disconnect_obj, cc3k_disconnect);
  438. STATIC mp_obj_t cc3k_isconnected(mp_obj_t self_in) {
  439. return mp_obj_new_bool(wlan_connected && ip_obtained);
  440. }
  441. STATIC MP_DEFINE_CONST_FUN_OBJ_1(cc3k_isconnected_obj, cc3k_isconnected);
  442. STATIC mp_obj_t cc3k_ifconfig(mp_obj_t self_in) {
  443. tNetappIpconfigRetArgs ipconfig;
  444. netapp_ipconfig(&ipconfig);
  445. // render MAC address
  446. VSTR_FIXED(mac_vstr, 18);
  447. const uint8_t *mac = ipconfig.uaMacAddr;
  448. vstr_printf(&mac_vstr, "%02x:%02x:%02x:%02x:%02x:%02x", mac[5], mac[4], mac[3], mac[2], mac[1], mac[0]);
  449. // create and return tuple with ifconfig info
  450. mp_obj_t tuple[7] = {
  451. netutils_format_ipv4_addr(ipconfig.aucIP, NETUTILS_LITTLE),
  452. netutils_format_ipv4_addr(ipconfig.aucSubnetMask, NETUTILS_LITTLE),
  453. netutils_format_ipv4_addr(ipconfig.aucDefaultGateway, NETUTILS_LITTLE),
  454. netutils_format_ipv4_addr(ipconfig.aucDNSServer, NETUTILS_LITTLE),
  455. netutils_format_ipv4_addr(ipconfig.aucDHCPServer, NETUTILS_LITTLE),
  456. mp_obj_new_str(mac_vstr.buf, mac_vstr.len),
  457. mp_obj_new_str((const char*)ipconfig.uaSSID, strlen((const char*)ipconfig.uaSSID)),
  458. };
  459. return mp_obj_new_tuple(MP_ARRAY_SIZE(tuple), tuple);
  460. }
  461. STATIC MP_DEFINE_CONST_FUN_OBJ_1(cc3k_ifconfig_obj, cc3k_ifconfig);
  462. STATIC mp_obj_t cc3k_patch_version(mp_obj_t self_in) {
  463. uint8_t pver[2];
  464. mp_obj_tuple_t *t_pver;
  465. nvmem_read_sp_version(pver);
  466. t_pver = mp_obj_new_tuple(2, NULL);
  467. t_pver->items[0] = mp_obj_new_int(pver[0]);
  468. t_pver->items[1] = mp_obj_new_int(pver[1]);
  469. return t_pver;
  470. }
  471. STATIC MP_DEFINE_CONST_FUN_OBJ_1(cc3k_patch_version_obj, cc3k_patch_version);
  472. STATIC mp_obj_t cc3k_patch_program(mp_obj_t self_in, mp_obj_t key_in) {
  473. const char *key = mp_obj_str_get_str(key_in);
  474. if (key[0] == 'p' && key[1] == 'g' && key[2] == 'm' && key[3] == '\0') {
  475. patch_prog_start();
  476. } else {
  477. mp_print_str(&mp_plat_print, "pass 'pgm' as argument in order to program\n");
  478. }
  479. return mp_const_none;
  480. }
  481. STATIC MP_DEFINE_CONST_FUN_OBJ_2(cc3k_patch_program_obj, cc3k_patch_program);
  482. STATIC const mp_rom_map_elem_t cc3k_locals_dict_table[] = {
  483. { MP_ROM_QSTR(MP_QSTR_connect), MP_ROM_PTR(&cc3k_connect_obj) },
  484. { MP_ROM_QSTR(MP_QSTR_disconnect), MP_ROM_PTR(&cc3k_disconnect_obj) },
  485. { MP_ROM_QSTR(MP_QSTR_isconnected), MP_ROM_PTR(&cc3k_isconnected_obj) },
  486. { MP_ROM_QSTR(MP_QSTR_ifconfig), MP_ROM_PTR(&cc3k_ifconfig_obj) },
  487. { MP_ROM_QSTR(MP_QSTR_patch_version), MP_ROM_PTR(&cc3k_patch_version_obj) },
  488. { MP_ROM_QSTR(MP_QSTR_patch_program), MP_ROM_PTR(&cc3k_patch_program_obj) },
  489. // class constants
  490. { MP_ROM_QSTR(MP_QSTR_WEP), MP_ROM_INT(WLAN_SEC_WEP) },
  491. { MP_ROM_QSTR(MP_QSTR_WPA), MP_ROM_INT(WLAN_SEC_WPA) },
  492. { MP_ROM_QSTR(MP_QSTR_WPA2), MP_ROM_INT(WLAN_SEC_WPA2) },
  493. };
  494. STATIC MP_DEFINE_CONST_DICT(cc3k_locals_dict, cc3k_locals_dict_table);
  495. const mod_network_nic_type_t mod_network_nic_type_cc3k = {
  496. .base = {
  497. { &mp_type_type },
  498. .name = MP_QSTR_CC3K,
  499. .make_new = cc3k_make_new,
  500. .locals_dict = (mp_obj_dict_t*)&cc3k_locals_dict,
  501. },
  502. .gethostbyname = cc3k_gethostbyname,
  503. .socket = cc3k_socket_socket,
  504. .close = cc3k_socket_close,
  505. .bind = cc3k_socket_bind,
  506. .listen = cc3k_socket_listen,
  507. .accept = cc3k_socket_accept,
  508. .connect = cc3k_socket_connect,
  509. .send = cc3k_socket_send,
  510. .recv = cc3k_socket_recv,
  511. .sendto = cc3k_socket_sendto,
  512. .recvfrom = cc3k_socket_recvfrom,
  513. .setsockopt = cc3k_socket_setsockopt,
  514. .settimeout = cc3k_socket_settimeout,
  515. .ioctl = cc3k_socket_ioctl,
  516. };