From 3f913efda03fd840cd526ef72e6f397c7da61bd7 Mon Sep 17 00:00:00 2001
From: Uros Majstorovic <majstor@majstor.org>
Date: Tue, 9 Aug 2022 22:23:08 +0200
Subject: code layout

---
 fw/fe310/eos/net/sock.c | 29 +++++++++++++++++++----------
 1 file changed, 19 insertions(+), 10 deletions(-)

(limited to 'fw/fe310/eos/net/sock.c')

diff --git a/fw/fe310/eos/net/sock.c b/fw/fe310/eos/net/sock.c
index 7365c97..1db0cd9 100644
--- a/fw/fe310/eos/net/sock.c
+++ b/fw/fe310/eos/net/sock.c
@@ -4,7 +4,7 @@
 
 #include "eos.h"
 #include "event.h"
-#include "net.h"
+#include "dev/net.h"
 
 #include "sock.h"
 
@@ -34,7 +34,7 @@ static void sock_handle_msg(unsigned char type, unsigned char *buffer, uint16_t
     }
 }
 
-void eos_sock_netinit(void) {
+void eos_sock_init(void) {
     int i;
 
     for (i=0; i<EOS_SOCK_MAX_SOCK; i++) {
@@ -69,20 +69,29 @@ int eos_sock_open_udp(eos_evt_handler_t handler, unsigned char *buffer) {
     buffer[0] = EOS_SOCK_MTYPE_OPEN_DGRAM;
 
     rv = eos_net_xchg(&type, buffer, &len);
-    if (rv) return rv;
+    if (rv) goto sock_open_udp_fin;
 
-    if (type != EOS_NET_MTYPE_SOCK) return EOS_ERR_NET;
-    if (len < 2) return EOS_ERR_SIZE;
+    if (type != EOS_NET_MTYPE_SOCK) {
+        rv = EOS_ERR_NET;
+        goto sock_open_udp_fin;
+    }
+    if (len < 2) {
+        rv = EOS_ERR_SIZE;
+        goto sock_open_udp_fin;
+    }
 
     sock = buffer[1];
-    if (sock == 0) return EOS_ERR_NET;
-
-    if (do_release) {
-        eos_net_free(buffer, 1);
+    if (sock == 0) {
+        rv = EOS_ERR_NET;
+        goto sock_open_udp_fin;
     }
+
+    rv = sock;
     eos_sock_set_handler(sock, handler);
 
-    return sock;
+sock_open_udp_fin:
+    if (do_release) eos_net_free(buffer, 1);
+    return rv;
 }
 
 void eos_sock_close(unsigned char sock, unsigned char *buffer) {
-- 
cgit v1.2.3