extmod/modussl: address comments

tve · tve · commit 40a4b51649b6 · 2020-09-17T09:59:20.000-07:00
diff --git a/extmod/modussl_axtls.c b/extmod/modussl_axtls.c
@@ -241,7 +241,7 @@ STATIC mp_uint_t ussl_socket_read(mp_obj_t o_in, void *buf, mp_uint_t size, int
 }
 
 STATIC mp_obj_t ussl_socket_recv(mp_obj_t self_in, mp_obj_t len_in) {
-    size_t len = mp_obj_get_int(len_in);
+    size_t len = mp_obj_int_get_uint_checked(len_in);
     vstr_t vstr;
     vstr_init_len(&vstr, len);
 
@@ -267,8 +267,8 @@ STATIC mp_uint_t ussl_socket_write(mp_obj_t o_in, const void *buf, mp_uint_t siz
     mp_int_t r;
 eagain:
     r = ssl_write(o->ssl_sock, buf, size);
-    if (r == SSL_OK) {
-        // see comment in read method
+    if (r == 0) {
+        // see comment in ussl_socket_read above
         if (o->blocking) {
             goto eagain;
         } else {
diff --git a/ports/esp32/modsocket.c b/ports/esp32/modsocket.c
@@ -667,12 +667,10 @@ STATIC mp_uint_t socket_stream_write(mp_obj_t self_in, const void *buf, mp_uint_
         int r = lwip_write(sock->fd, buf, size);
         MP_THREAD_GIL_ENTER();
         if (r > 0) {
-            printf("socket_stream_write wrote %d\n", r);
             return r;
         }
         // lwip returns MP_EINPROGRESS when trying to write right after a non-blocking connect
         if (r < 0 && errno != MP_EWOULDBLOCK && errno != MP_EINPROGRESS) {
-            printf("socket_stream_write error %d\n", errno);
             *errcode = errno;
             return MP_STREAM_ERROR;
         }

Original file line number	Diff line number	Diff line change
`@@ -667,12 +667,10 @@ STATIC mp_uint_t socket_stream_write(mp_obj_t self_in, const void *buf, mp_uint_`
`667`	`667`	`int r = lwip_write(sock->fd, buf, size);`
`668`	`668`	`MP_THREAD_GIL_ENTER();`
`669`	`669`	`if (r > 0) {`
`670`		`- printf("socket_stream_write wrote %d\n", r);`
`671`	`670`	`return r;`
`672`	`671`	`}`
`673`	`672`	`// lwip returns MP_EINPROGRESS when trying to write right after a non-blocking connect`
`674`	`673`	`if (r < 0 && errno != MP_EWOULDBLOCK && errno != MP_EINPROGRESS) {`
`675`		`- printf("socket_stream_write error %d\n", errno);`
`676`	`674`	`*errcode = errno;`
`677`	`675`	`return MP_STREAM_ERROR;`
`678`	`676`	`}`