xinghenLuyus/stopwait_chat_ipv6
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

更新 linux_chat.c

Browse Source
This commit is contained in:
xinghenLuyus
2025-12-26 15:00:11 +08:00
parent a9939d43d4
commit a22d94d7db
1 changed files with 49 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

85
linux_chat.c
View File

@@ -10,7 +10,7 @@
#define LOCAL_PORT 9000 #define LOCAL_PORT 9000
#define PEER_PORT 9001 #define PEER_PORT 9001
#define TIMEOUT_MS 500 #define TIMEOUT_MS 200
int main(int argc, char *argv[]) { int main(int argc, char *argv[]) {
if (argc != 2) { if (argc != 2) {
@@ -19,16 +19,11 @@ int main(int argc, char *argv[]) {
} }
int sock = socket(AF_INET6, SOCK_DGRAM, 0); int sock = socket(AF_INET6, SOCK_DGRAM, 0);
if (sock < 0) {
perror("socket");
return 1;
}
struct sockaddr_in6 local = {0}; struct sockaddr_in6 local = {0};
local.sin6_family = AF_INET6; local.sin6_family = AF_INET6;
local.sin6_port = htons(LOCAL_PORT); local.sin6_port = htons(LOCAL_PORT);
local.sin6_addr = in6addr_any; local.sin6_addr = in6addr_any;
bind(sock, (struct sockaddr*)&local, sizeof(local)); bind(sock, (struct sockaddr*)&local, sizeof(local));
struct sockaddr_in6 peer = {0}; struct sockaddr_in6 peer = {0};
@@ -36,46 +31,64 @@ int main(int argc, char *argv[]) {
peer.sin6_port = htons(PEER_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);
struct timeval tv; struct timeval tv = {0, TIMEOUT_MS * 1000};
tv.tv_sec = 0;
tv.tv_usec = TIMEOUT_MS * 1000;
setsockopt(sock, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof(tv)); setsockopt(sock, SOL_SOCKET, SO_RCVTIMEO, &tv, sizeof(tv));
uint8_t seq = 0; uint8_t send_seq = 0;
char input[DATA_LEN]; uint8_t waiting_ack = 0;
packet_t last_pkt;
while (1) { while (1) {
printf("You: "); /* 尝试接收 */
fflush(stdout); packet_t pkt;
fgets(input, sizeof(input), stdin); struct sockaddr_in6 from;
socklen_t fromlen = sizeof(from);
packet_t pkt = {0}; int n = recvfrom(sock, &pkt, sizeof(pkt), 0,
pkt.type = TYPE_DATA; (struct sockaddr*)&from, &fromlen);
pkt.seq = seq;
pkt.length = strlen(input);
memcpy(pkt.data, input, pkt.length);
while (1) { if (n > 0) {
sendto(sock, &pkt, sizeof(pkt), 0, if (pkt.type == TYPE_DATA) {
(struct sockaddr*)&peer, sizeof(peer)); printf("\nPeer: %s", pkt.data);
packet_t ack; packet_t ack = {0};
struct sockaddr_in6 from; ack.type = TYPE_ACK;
socklen_t fromlen = sizeof(from); ack.seq = pkt.seq;
int n = recvfrom(sock, &ack, sizeof(ack), 0, sendto(sock, &ack, sizeof(ack), 0,
(struct sockaddr*)&from, &fromlen); (struct sockaddr*)&peer, sizeof(peer));
}
if (n > 0 && ack.type == TYPE_ACK && ack.seq == seq) { else if (pkt.type == TYPE_ACK &&
break; waiting_ack && pkt.seq == send_seq) {
waiting_ack = 0;
send_seq ^= 1;
} }
usleep(TIMEOUT_MS * 1000);
} }
seq ^= 1; /* 如果不在等 ACK，则允许发送 */
} if (!waiting_ack) {
char buf[DATA_LEN];
printf("You: ");
fflush(stdout);
close(sock); if (fgets(buf, sizeof(buf), stdin) == NULL)
return 0; 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, &last_pkt, sizeof(last_pkt), 0,
(struct sockaddr*)&peer, sizeof(peer));
waiting_ack = 1;
} else {
/* 超时重传 */
sendto(sock, &last_pkt, sizeof(last_pkt), 0,
(struct sockaddr*)&peer, sizeof(peer));
usleep(TIMEOUT_MS * 1000);
}
}
} }