更新 windows_chat.c

This commit is contained in:
2025-12-26 15:00:35 +08:00
parent a22d94d7db
commit ae50f15afe

View File

@@ -10,7 +10,7 @@
#define LOCAL_PORT 9001
#define PEER_PORT 9000
#define TIMEOUT_MS 500
#define TIMEOUT_MS 200
int main(int argc, char *argv[]) {
if (argc != 2) {
@@ -23,16 +23,13 @@ int main(int argc, char *argv[]) {
SOCKET sock = socket(AF_INET6, SOCK_DGRAM, 0);
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;
bind(sock, (struct sockaddr*)&local, sizeof(local));
struct sockaddr_in6 peer;
memset(&peer, 0, sizeof(peer));
struct sockaddr_in6 peer = {0};
peer.sin6_family = AF_INET6;
peer.sin6_port = htons(PEER_PORT);
inet_pton(AF_INET6, argv[1], &peer.sin6_addr);
@@ -41,42 +38,60 @@ int main(int argc, char *argv[]) {
setsockopt(sock, SOL_SOCKET, SO_RCVTIMEO,
(char*)&timeout, sizeof(timeout));
uint8_t seq = 0;
char input[DATA_LEN];
uint8_t send_seq = 0;
uint8_t waiting_ack = 0;
packet_t last_pkt;
while (1) {
printf("You: ");
fflush(stdout);
fgets(input, sizeof(input), stdin);
/* 尝试接收 */
packet_t pkt;
struct sockaddr_in6 from;
int fromlen = sizeof(from);
packet_t pkt = {0};
pkt.type = TYPE_DATA;
pkt.seq = seq;
pkt.length = strlen(input);
memcpy(pkt.data, input, pkt.length);
int n = recvfrom(sock, (char*)&pkt, sizeof(pkt), 0,
(struct sockaddr*)&from, &fromlen);
while (1) {
sendto(sock, (char*)&pkt, sizeof(pkt), 0,
(struct sockaddr*)&peer, sizeof(peer));
if (n > 0) {
if (pkt.type == TYPE_DATA) {
printf("\nPeer: %s", pkt.data);
packet_t ack;
struct sockaddr_in6 from;
int fromlen = sizeof(from);
packet_t ack = {0};
ack.type = TYPE_ACK;
ack.seq = pkt.seq;
int n = recvfrom(sock, (char*)&ack, sizeof(ack), 0,
(struct sockaddr*)&from, &fromlen);
if (n > 0 && ack.type == TYPE_ACK && ack.seq == seq) {
break;
sendto(sock, (char*)&ack, sizeof(ack), 0,
(struct sockaddr*)&peer, sizeof(peer));
}
else if (pkt.type == TYPE_ACK &&
waiting_ack && pkt.seq == send_seq) {
waiting_ack = 0;
send_seq ^= 1;
}
Sleep(TIMEOUT_MS);
}
seq ^= 1;
}
/* 如果不在等 ACK允许发送 */
if (!waiting_ack) {
char buf[DATA_LEN];
printf("You: ");
fflush(stdout);
closesocket(sock);
WSACleanup();
return 0;
if (fgets(buf, sizeof(buf), stdin) == NULL)
continue;
memset(&last_pkt, 0, sizeof(last_pkt));
last_pkt.type = TYPE_DATA;
last_pkt.seq = send_seq;
last_pkt.length = strlen(buf);
memcpy(last_pkt.data, buf, last_pkt.length);
sendto(sock, (char*)&last_pkt, sizeof(last_pkt), 0,
(struct sockaddr*)&peer, sizeof(peer));
waiting_ack = 1;
} else {
sendto(sock, (char*)&last_pkt, sizeof(last_pkt), 0,
(struct sockaddr*)&peer, sizeof(peer));
Sleep(TIMEOUT_MS);
}
}
}