vconn bugfix

1 files changed, 33 insertions, 36 deletions

diff --git a/ecp/src/platform/fe310/transport.c b/ecp/src/platform/fe310/transport.c
index b551c51..2640ecf 100644
--- a/ecp/src/platform/fe310/transport.c
+++ b/ecp/src/platform/fe310/transport.c
@@ -1,5 +1,6 @@
 #include <stdlib.h>
 #include <string.h>
+#include <stdio.h>
 
 #include <core.h>
 #include <tr.h>
@@ -7,43 +8,42 @@
 #include <eos/eos.h>
 #include <eos/net.h>
 
-static unsigned char _flags = 0;
-
 ECPSocket *_ecp_tr_sock = NULL;
+unsigned char pld_buf[ECP_MAX_PLD];
 
 static void packet_handler(unsigned char type, unsigned char *buffer, uint16_t len) {
-    ecp_tr_addr_t addr;
-
     ECP2Buffer bufs;
     ECPBuffer packet;
     ECPBuffer payload;
-    unsigned char pld_buf[ECP_MAX_PLD];
+    ecp_tr_addr_t addr;
+    ssize_t rv;
 
     bufs.packet = &packet;
     bufs.payload = &payload;
 
     packet.buffer = buffer+EOS_SOCK_SIZE_UDP_HDR;
-    packet.size = ECP_MAX_PKT;
+    packet.size = EOS_NET_MTU-EOS_SOCK_SIZE_UDP_HDR;
     payload.buffer = pld_buf;
     payload.size = ECP_MAX_PLD;
 
-    if ((buffer == NULL) || (len < EOS_SOCK_SIZE_UDP_HDR)) {
+    if (len < EOS_SOCK_SIZE_UDP_HDR) {
         eos_net_free(buffer, 0);
         return;
     }
 
-    eos_sock_getfrom(buffer, &addr);
-    ssize_t rv = ecp_pkt_handle(_ecp_tr_sock, &addr, NULL, &bufs, len-EOS_SOCK_SIZE_UDP_HDR);
+    eos_sock_recvfrom(buffer, len, NULL, 0, &addr);
+
+    rv = ecp_pkt_handle(_ecp_tr_sock, NULL, &addr, &bufs, len-EOS_SOCK_SIZE_UDP_HDR);
 #ifdef ECP_DEBUG
     if (rv < 0) {
-        char b[16];
-        puts("ERR:");
-        puts(itoa(rv, b, 10));
-        puts("\n");
+        printf("RCV ERR:%d\n", rv);
     }
 #endif
-    if (bufs.packet->buffer) eos_net_free(buffer, 0);
-    eos_net_release();
+    if (bufs.packet->buffer) {
+        eos_net_free(buffer, 0);
+    } else {
+        eos_net_release();
+    }
 }
 
 int ecp_tr_init(ECPContext *ctx) {
@@ -66,38 +66,43 @@ int ecp_tr_addr_set(ecp_tr_addr_t *addr, void *addr_s) {
 }
 
 int ecp_tr_open(ECPSocket *sock, void *addr_s) {
-    sock->sock = eos_sock_open_udp(packet_handler);
+    sock->sock = eos_sock_open_udp(packet_handler, NULL);
     if (sock->sock < 0) {
         sock->sock = 0;
-        return ECP_ERR_SEND;
+        return ECP_ERR;
     }
     _ecp_tr_sock = sock;
+
     return ECP_OK;
 }
 
 void ecp_tr_close(ECPSocket *sock) {
-    eos_sock_close(sock->sock);
+    eos_sock_close(sock->sock, NULL);
     _ecp_tr_sock = NULL;
 }
 
-ssize_t ecp_tr_send(ECPSocket *sock, ECPBuffer *packet, size_t msg_size, ecp_tr_addr_t *addr, unsigned char flags) {
+ssize_t ecp_tr_send(ECPSocket *sock, ECPBuffer *packet, size_t pkt_size, ecp_tr_addr_t *addr, unsigned char flags) {
     unsigned char *buf = NULL;
+    int reply;
     int rv;
 
-    flags |= _flags;
-    if (packet && packet->buffer) {
-        if (flags & ECP_SEND_FLAG_REPLY) {
+    reply = 0;
+    if (flags & ECP_SEND_FLAG_REPLY) {
+        if (flags & ECP_SEND_FLAG_MORE) return ECP_ERR;
+        if (packet->buffer) {
             buf = packet->buffer-EOS_SOCK_SIZE_UDP_HDR;
             packet->buffer = NULL;
-        } else {
-            buf = eos_net_alloc();
-            memcpy(buf+EOS_SOCK_SIZE_UDP_HDR, packet->buffer, msg_size);
+            reply = 1;
         }
+    } else {
+        buf = eos_net_alloc();
     }
     if (buf == NULL) return ECP_ERR;
-    rv = eos_sock_sendto(sock->sock, buf, msg_size+EOS_SOCK_SIZE_UDP_HDR, flags & ECP_SEND_FLAG_MORE, addr);
+
+    rv = eos_sock_sendto_async(sock->sock, reply ? NULL : packet->buffer, pkt_size, addr, buf, !!(flags & ECP_SEND_FLAG_MORE));
     if (rv) return ECP_ERR_SEND;
-    return msg_size;
+
+    return pkt_size;
 }
 
 ssize_t ecp_tr_recv(ECPSocket *sock, ECPBuffer *packet, ecp_tr_addr_t *addr, int timeout) {
@@ -105,18 +110,10 @@ ssize_t ecp_tr_recv(ECPSocket *sock, ECPBuffer *packet, ecp_tr_addr_t *addr, int
 }
 
 void ecp_tr_release(ECPBuffer *packet, unsigned char more) {
-    if (packet && packet->buffer) {
+    if (packet->buffer) {
         eos_net_free(packet->buffer-EOS_SOCK_SIZE_UDP_HDR, more);
         packet->buffer = NULL;
     } else if (!more) {
         eos_net_release();
     }
 }
-
-void ecp_tr_flag_set(unsigned char flags) {
-    _flags |= flags;
-}
-
-void ecp_tr_flag_clear(unsigned char flags) {
-    _flags &= ~flags;
-}