expaned protocall support

Author: dreamer-coding

code/logic/fossil/io/network.h

@@ -32,6 +32,12 @@
     #define FOSSIL_IO_INVALID_SOCKET (-1)
 #endif
 
+#define FOSSIL_IO_SOCKET_TYPE_TCP SOCK_STREAM
+#define FOSSIL_IO_SOCKET_TYPE_UDP SOCK_DGRAM
+#define FOSSIL_IO_SOCKET_TYPE_RAW SOCK_RAW
+#define FOSSIL_IO_SOCKET_TYPE_RDM SOCK_RDM
+#define FOSSIL_IO_SOCKET_TYPE_SEQPACKET SOCK_SEQPACKET
+
 #ifdef __cplusplus
 extern "C" {
 #endif
@@ -51,7 +57,7 @@ void fossil_io_network_destroy(void);
  * Create a new TCP socket.
  * Returns a valid socket on success or FOSSIL_IO_INVALID_SOCKET on failure.
  */
-fossil_io_socket_t fossil_io_network_create_socket(void);
+fossil_io_socket_t fossil_io_network_create_socket(int type);
 
 /**
  * Bind a socket to a specific port (IPv4/IPv6).
@@ -94,8 +100,21 @@ int fossil_io_network_receive(fossil_io_socket_t sock, void *buffer, size_t len)
  */
 void fossil_io_network_close(fossil_io_socket_t sock);
 
+/**
+ * Send data to a specific IP address and port.
+ * Returns the number of bytes sent, or -1 on failure.
+ */
+int fossil_io_network_sendto(fossil_io_socket_t sock, const void *data, size_t len, const char *ip, uint16_t port);
+
+/**
+ * Receive data from a specific IP address and port.
+ * Returns the number of bytes received, or -1 on failure.
+ */
+int fossil_io_network_recvfrom(fossil_io_socket_t sock, void *buffer, size_t len, char *ip, uint16_t *port);
+
 #ifdef __cplusplus
 }
+
 /**
  * C++ wrapper for the output functions.
  */
@@ -128,8 +147,8 @@ namespace fossil {
              * Create a new TCP socket.
              * Returns a valid socket on success or FOSSIL_IO_INVALID_SOCKET on failure.
              */
-            static fossil_io_socket_t create_socket(void) {
-                return fossil_io_network_create_socket();
+            static fossil_io_socket_t create_socket(int type) {
+                return fossil_io_network_create_socket(type);
             }
 
             /**
@@ -187,6 +206,22 @@ namespace fossil {
                 fossil_io_network_close(sock);
             }
 
+            /**
+             * Send data to a specific IP address and port.
+             * Returns the number of bytes sent, or -1 on failure.
+             */
+            static int sendto(fossil_io_socket_t sock, const void *data, size_t len, const char *ip, uint16_t port) {
+                return fossil_io_network_sendto(sock, data, len, ip, port);
+            }
+
+            /**
+             * Receive data from a specific IP address and port.
+             * Returns the number of bytes received, or -1 on failure.
+             */
+            static int recvfrom(fossil_io_socket_t sock, void *buffer, size_t len, char *ip, uint16_t *port) {
+                return fossil_io_network_recvfrom(sock, buffer, len, ip, port);
+            }
+
         };
     }
 }

code/logic/network.c

@@ -35,8 +35,8 @@ void fossil_io_network_destroy(void) {
 #endif
 }
 
-fossil_io_socket_t fossil_io_network_create_socket(void) {
-    fossil_io_socket_t sock = socket(AF_INET, SOCK_STREAM, 0);
+fossil_io_socket_t fossil_io_network_create_socket(int type) {
+    fossil_io_socket_t sock = socket(AF_INET, type, 0);
     if (sock == FOSSIL_IO_INVALID_SOCKET) {
 #ifdef _WIN32
         fprintf(stderr, "Socket creation failed with error: %d\n", WSAGetLastError());
@@ -138,6 +138,43 @@ int fossil_io_network_receive(fossil_io_socket_t sock, void *buffer, size_t len)
     return bytes_received;
 }
 
+int fossil_io_network_sendto(fossil_io_socket_t sock, const void *data, size_t len, const char *ip, uint16_t port) {
+    struct sockaddr_in dest_addr;
+    dest_addr.sin_family = AF_INET;
+    dest_addr.sin_port = htons(port);
+    dest_addr.sin_addr.s_addr = inet_addr(ip);
+
+    int bytes_sent = sendto(sock, data, (int)len, 0, (struct sockaddr*)&dest_addr, sizeof(dest_addr));
+    if (bytes_sent == -1) {
+#ifdef _WIN32
+        fprintf(stderr, "Sendto failed with error: %d\n", WSAGetLastError());
+#else
+        perror("Sendto failed");
+#endif
+    }
+    return bytes_sent;
+}
+
+int fossil_io_network_recvfrom(fossil_io_socket_t sock, void *buffer, size_t len, char *ip, uint16_t *port) {
+    struct sockaddr_in src_addr;
+    socklen_t addr_len = sizeof(src_addr);
+
+    int bytes_received = recvfrom(sock, buffer, (int)len, 0, (struct sockaddr*)&src_addr, &addr_len);
+    if (bytes_received == -1) {
+#ifdef _WIN32
+        fprintf(stderr, "Recvfrom failed with error: %d\n", WSAGetLastError());
+#else
+        perror("Recvfrom failed");
+#endif
+    } else if (ip) {
+        strcpy(ip, inet_ntoa(src_addr.sin_addr));
+        if (port) {
+            *port = ntohs(src_addr.sin_port);
+        }
+    }
+    return bytes_received;
+}
+
 void fossil_io_network_close(fossil_io_socket_t sock) {
 #ifdef _WIN32
     if (closesocket(sock) == SOCKET_ERROR) {