更新 linux_chat.c

linux_chat.c

@@ -1,103 +1,79 @@
 #include <stdio.h>
+#include <stdlib.h>
 #include <string.h>
 #include <unistd.h>
 #include <arpa/inet.h>
 #include <sys/socket.h>
-#include <sys/select.h>
+#include <sys/time.h>
 
 #include "common.h"
 
-/* 停等协议发送（带超时重传） */
-void send_with_stopwait(int sock,
-                        struct sockaddr_in6 *peer,
-                        struct packet *pkt)
-{
-    socklen_t len = sizeof(*peer);
+#define LOCAL_PORT 9000
+#define PEER_PORT  9001
+#define TIMEOUT_MS 500
 
-    while (1) {
-        sendto(sock, pkt, sizeof(*pkt), 0,
-               (struct sockaddr *)peer, len);
-
-        fd_set fds;
-        FD_ZERO(&fds);
-        FD_SET(sock, &fds);
-
-        struct timeval tv;
-        tv.tv_sec = TIMEOUT_SEC;
-        tv.tv_usec = 0;
-
-        int ret = select(sock + 1, &fds, NULL, NULL, &tv);
-        if (ret > 0) {
-            struct packet ack;
-            recvfrom(sock, &ack, sizeof(ack), 0, NULL, NULL);
-            if (ack.type == TYPE_ACK && ack.seq == pkt->seq) {
-                return; // 正确 ACK
-            }
-        }
-        printf("[timeout] retransmit\n");
-    }
-}
-
-int main(int argc, char *argv[])
-{
+int main(int argc, char *argv[]) {
     if (argc != 2) {
-        printf("用法: %s <对方IPv6地址>\n", argv[0]);
+        printf("Usage: %s <Windows IPv6>\n", argv[0]);
         return 1;
     }
 
     int sock = socket(AF_INET6, SOCK_DGRAM, 0);
+    if (sock < 0) {
+        perror("socket");
+        return 1;
+    }
 
-    /* 绑定本地 IPv6 端口 */
-    struct sockaddr_in6 local;
-    memset(&local, 0, sizeof(local));
+    struct sockaddr_in6 local = {0};
     local.sin6_family = AF_INET6;
+    local.sin6_port = htons(LOCAL_PORT);
     local.sin6_addr = in6addr_any;
-    local.sin6_port = htons(PORT);
-    bind(sock, (struct sockaddr *)&local, sizeof(local));
 
-    /* 对端地址 */
-    struct sockaddr_in6 peer;
-    memset(&peer, 0, sizeof(peer));
+    bind(sock, (struct sockaddr*)&local, sizeof(local));
+
+    struct sockaddr_in6 peer = {0};
     peer.sin6_family = AF_INET6;
-    peer.sin6_port = htons(PORT);
+    peer.sin6_port = htons(PEER_PORT);
     inet_pton(AF_INET6, argv[1], &peer.sin6_addr);
 
-    uint8_t send_seq = 0;
-    uint8_t recv_seq = 0;
-    struct packet pkt, rcv;
+    struct timeval tv;
+    tv.tv_sec = 0;
+    tv.tv_usec = TIMEOUT_MS * 1000;
+    setsockopt(sock, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof(tv));
 
-    printf("Linux UDP IPv6 Stop-Wait Chat\n");
+    uint8_t seq = 0;
+    char input[DATA_LEN];
 
     while (1) {
-        /* 非阻塞接收 */
-        fd_set fds;
-        FD_ZERO(&fds);
-        FD_SET(sock, &fds);
-        struct timeval tv = {0, 100000};
+        printf("You: ");
+        fflush(stdout);
+        fgets(input, sizeof(input), stdin);
 
-        if (select(sock + 1, &fds, NULL, NULL, &tv) > 0) {
-            recvfrom(sock, &rcv, sizeof(rcv), 0, NULL, NULL);
-            if (rcv.type == TYPE_DATA) {
-                if (rcv.seq == recv_seq) {
-                    printf("\npeer: %s\n", rcv.payload);
-                    recv_seq ^= 1;
-                }
-                rcv.type = TYPE_ACK;
-                sendto(sock, &rcv, sizeof(rcv), 0,
-                       (struct sockaddr *)&peer, sizeof(peer));
+        packet_t pkt = {0};
+        pkt.type = TYPE_DATA;
+        pkt.seq = seq;
+        pkt.length = strlen(input);
+        memcpy(pkt.data, input, pkt.length);
+
+        while (1) {
+            sendto(sock, &pkt, sizeof(pkt), 0,
+                   (struct sockaddr*)&peer, sizeof(peer));
+
+            packet_t ack;
+            struct sockaddr_in6 from;
+            socklen_t fromlen = sizeof(from);
+
+            int n = recvfrom(sock, &ack, sizeof(ack), 0,
+                             (struct sockaddr*)&from, &fromlen);
+
+            if (n > 0 && ack.type == TYPE_ACK && ack.seq == seq) {
+                break;
             }
+
+            usleep(TIMEOUT_MS * 1000);
         }
 
-        /* 发送 */
-        printf(">> ");
-        if (!fgets(pkt.payload, sizeof(pkt.payload), stdin))
-            break;
-
-        pkt.seq = send_seq;
-        pkt.type = TYPE_DATA;
-
-        send_with_stopwait(sock, &peer, &pkt);
-        send_seq ^= 1;
+        seq ^= 1;
     }
 
     close(sock);