更新 windows_chat.c

This commit is contained in:
2025-12-26 14:54:51 +08:00
parent 03b2cf8c8a
commit a9939d43d4

View File

@@ -2,98 +2,78 @@
#include <string.h> #include <string.h>
#include <winsock2.h> #include <winsock2.h>
#include <ws2tcpip.h> #include <ws2tcpip.h>
#include <windows.h>
#include "common.h"
#pragma comment(lib, "ws2_32.lib") #pragma comment(lib, "ws2_32.lib")
void send_with_stopwait(SOCKET sock, #include "common.h"
struct sockaddr_in6 *peer,
struct packet *pkt)
{
int len = sizeof(*peer);
while (1) { #define LOCAL_PORT 9001
sendto(sock, (char *)pkt, sizeof(*pkt), 0, #define PEER_PORT 9000
(struct sockaddr *)peer, len); #define TIMEOUT_MS 500
fd_set fds; int main(int argc, char *argv[]) {
FD_ZERO(&fds);
FD_SET(sock, &fds);
TIMEVAL tv;
tv.tv_sec = TIMEOUT_SEC;
tv.tv_usec = 0;
int ret = select(0, &fds, NULL, NULL, &tv);
if (ret > 0) {
struct packet ack;
recvfrom(sock, (char *)&ack, sizeof(ack), 0, NULL, NULL);
if (ack.type == TYPE_ACK && ack.seq == pkt->seq) {
return;
}
}
printf("[timeout] retransmit\n");
}
}
int main(int argc, char *argv[])
{
if (argc != 2) { if (argc != 2) {
printf("用法: %s <对方IPv6地址>\n", argv[0]); printf("Usage: %s <Ubuntu IPv6>\n", argv[0]);
return 1; return 1;
} }
WSADATA wsa; WSADATA wsa;
WSAStartup(MAKEWORD(2, 2), &wsa); WSAStartup(MAKEWORD(2,2), &wsa);
SOCKET sock = socket(AF_INET6, SOCK_DGRAM, 0); SOCKET sock = socket(AF_INET6, SOCK_DGRAM, 0);
struct sockaddr_in6 local = {0}; struct sockaddr_in6 local;
memset(&local, 0, sizeof(local));
local.sin6_family = AF_INET6; local.sin6_family = AF_INET6;
local.sin6_port = htons(LOCAL_PORT);
local.sin6_addr = in6addr_any; local.sin6_addr = in6addr_any;
local.sin6_port = htons(PORT);
bind(sock, (struct sockaddr *)&local, sizeof(local));
struct sockaddr_in6 peer = {0}; bind(sock, (struct sockaddr*)&local, sizeof(local));
struct sockaddr_in6 peer;
memset(&peer, 0, sizeof(peer));
peer.sin6_family = AF_INET6; 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); inet_pton(AF_INET6, argv[1], &peer.sin6_addr);
uint8_t send_seq = 0; int timeout = TIMEOUT_MS;
uint8_t recv_seq = 0; setsockopt(sock, SOL_SOCKET, SO_RCVTIMEO,
struct packet pkt, rcv; (char*)&timeout, sizeof(timeout));
printf("Windows UDP IPv6 Stop-Wait Chat\n"); uint8_t seq = 0;
char input[DATA_LEN];
while (1) { while (1) {
fd_set fds; printf("You: ");
FD_ZERO(&fds); fflush(stdout);
FD_SET(sock, &fds); fgets(input, sizeof(input), stdin);
TIMEVAL tv = {0, 100000};
if (select(0, &fds, NULL, NULL, &tv) > 0) { packet_t pkt = {0};
recvfrom(sock, (char *)&rcv, sizeof(rcv), 0, NULL, NULL); pkt.type = TYPE_DATA;
if (rcv.type == TYPE_DATA) { pkt.seq = seq;
if (rcv.seq == recv_seq) { pkt.length = strlen(input);
printf("\npeer: %s\n", rcv.payload); memcpy(pkt.data, input, pkt.length);
recv_seq ^= 1;
} while (1) {
rcv.type = TYPE_ACK; sendto(sock, (char*)&pkt, sizeof(pkt), 0,
sendto(sock, (char *)&rcv, sizeof(rcv), 0, (struct sockaddr*)&peer, sizeof(peer));
(struct sockaddr *)&peer, sizeof(peer));
packet_t ack;
struct sockaddr_in6 from;
int fromlen = sizeof(from);
int n = recvfrom(sock, (char*)&ack, sizeof(ack), 0,
(struct sockaddr*)&from, &fromlen);
if (n > 0 && ack.type == TYPE_ACK && ack.seq == seq) {
break;
} }
Sleep(TIMEOUT_MS);
} }
printf(">> "); seq ^= 1;
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;
} }
closesocket(sock); closesocket(sock);