В настоящее время я работаю над программой «Клиент-сервер UDP», в которой постепенно внедряю некоторые функции TCP, помимо текущих UDP. Важно отметить, что этот код предназначен для работы с одним клиентом (реализация для большего количества клиентов, вероятно, придет позже).
Мне удалось правильно настроить трехстороннее рукопожатие, и сообщенияправильно обмениваются между клиентом и сервером. Я использую сообщение SYN-ACK для извлечения нового номера порта с сервера, который клиент будет использовать для обмена данными (в данном случае это просто сообщения). Однако после завершения синхронизации и изменения клиентской части порта я не могу продолжать обмен сообщениями, и сервер никогда не получает информацию, отправленную клиентом.
Вот код сервера:
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <errno.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/select.h>
#include <sys/time.h>
#include <netdb.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#define DOMAIN AF_INET
#define TYPE SOCK_DGRAM
#define PROTOCOL 0
#define BUFFER_SIZE 1025
#define TIMEOUT_SECONDS 1
#define TIMEOUT_MICRO 0
int main(int argc, char* argv[])
{
// INITIALIZATION
int port_sync, port_data;
if (argc < 3)
{
printf("Too few arguments given.\n");
printf("Format: ./server <port_sync> <port_data>\n");
exit(1);
}
else if (argc > 3)
{
printf("Too many arguments given.\n");
printf("Format: ./server <port_sync> <port_data>\n");
exit(1);
}
else
{
port_sync = atoi(argv[1]);
port_data = atoi(argv[2]);
}
// SELECT RETURN
int select_return;
// SOCKET OPTION
int opt = 1;
// FILE DESCRIPTOR
fd_set read_fd_set;
// SYNCHRONIZATION SOCKET
struct sockaddr_in server_addr;
memset((char*)&server_addr, 0, sizeof(server_addr));
int socket_fd;
if ((socket_fd = socket(DOMAIN, TYPE, PROTOCOL)) < 0)
{
perror("[-] Connection error");
exit(1);
}
printf("[+] Synchronization socket created.\n");
if(setsockopt(socket_fd, SOL_SOCKET, SO_REUSEADDR, (char *)&opt, sizeof(opt)) < 0)
{
perror("[-] setsockopt error");
exit(1);
}
server_addr.sin_family = DOMAIN;
server_addr.sin_port = htons(port_sync);
server_addr.sin_addr.s_addr = INADDR_ANY;
int socket_length = sizeof(server_addr);
// SYNCHRONIZATION SOCKET BIND
if (bind(socket_fd, (struct sockaddr*) &server_addr, sizeof(server_addr)) < 0)
{
perror("[-] Bind error");
exit(1);
}
printf("[+] Binded successfully to port %d.\n", port_sync);
// DATA SOCKET
struct sockaddr_in server_data_addr;
memset((char*)&server_data_addr, 0, sizeof(server_data_addr));
int socket_data_fd;
if ((socket_data_fd = socket(DOMAIN, TYPE, PROTOCOL)) < 0)
{
perror("[-] Connection error");
exit(1);
}
printf("[+] Data socket created.\n");
if(setsockopt(socket_data_fd, SOL_SOCKET, SO_REUSEADDR, (char *)&opt, sizeof(opt)) < 0)
{
perror("[-] setsockopt error");
exit(1);
}
server_data_addr.sin_family = DOMAIN;
server_data_addr.sin_port = htons(port_data);
server_data_addr.sin_addr.s_addr = INADDR_ANY;
int socket_data_length = sizeof(server_data_addr);
// DATA SOCKET BIND
if (bind(socket_data_fd, (struct sockaddr*) &server_data_addr, sizeof(server_data_addr)) < 0)
{
perror("[-] Bind error");
exit(1);
}
printf("[+] Binded successfully to port %d.\n", port_data);
// CLIENT SYNCHRONIZATION
struct sockaddr_in client_addr;
socklen_t client_length;
// CLIENT DATA
struct sockaddr_in client_data_addr;
socklen_t client_data_length;
char* client_data_IP;
int client_data_port;
// CONNECTION
int online = 1;
int sync_done = 0;
char buffer[BUFFER_SIZE];
while(online)
{
FD_ZERO(&read_fd_set);
if (!sync_done)
{
FD_SET(socket_fd, &read_fd_set);
}
FD_SET(socket_data_fd, &read_fd_set);
if (((select_return = select(10, &read_fd_set, NULL, NULL, NULL))) < 0 && (errno!=EINTR))
{
perror("[-] Select error\n");
exit(1);
}
if (FD_ISSET(socket_fd, &read_fd_set))
{
client_length = sizeof(client_addr);
bzero(buffer, sizeof(buffer));
recvfrom(socket_fd, buffer, BUFFER_SIZE, 0, (struct sockaddr*) &client_addr, &client_length);
if (strstr(buffer, "SYN") != NULL)
{
memset(buffer, 0, sizeof(buffer));
sprintf(buffer, "SYN-ACK-%i", port_data);
sendto(socket_fd, buffer, BUFFER_SIZE, 0, (struct sockaddr*) &client_addr, sizeof(client_addr));
recvfrom(socket_fd, buffer, BUFFER_SIZE, 0, (struct sockaddr*) & client_addr, &client_length);
if (strstr(buffer, "ACK"))
{
memset(buffer, 0, sizeof(buffer));
}
}
printf("Finished synchronization\n");
FD_CLR(socket_fd, &read_fd_set);
sync_done = 1;
}
if (FD_ISSET(socket_data_fd, &read_fd_set))
{
int recvfrom_return = recvfrom(socket_data_fd, buffer, BUFFER_SIZE, 0, (struct sockaddr*) & client_data_addr, &client_data_length);
if (strcmp(buffer, "") != 0)
{
client_data_IP = inet_ntoa(client_data_addr.sin_addr);
client_data_port = ntohs(client_data_addr.sin_port);
if (strcmp(buffer, ":exit") == 0)
{
printf("[-] Disconnected from %s:%d\n", client_data_IP, client_data_port);
close(socket_data_fd);
break;
}
else
{
printf("UDP Client: %s\n", buffer);
sendto(socket_data_fd, buffer, recvfrom_return, 0, (struct sockaddr*) & client_data_addr, client_data_length);
memset(buffer, 0, sizeof(buffer));
bzero(buffer, sizeof(buffer));
}
}
}
}
return 0;
}
и вот код клиента:
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <errno.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netdb.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#define DOMAIN AF_INET
#define TYPE SOCK_DGRAM
#define PROTOCOL 0
#define BUFFER_SIZE 1025
#define TIMEOUT_SECONDS 1
#define TIMEOUT_MICRO 0
int get_data_port(char* buffer)
{
char delim[] = "-";
char port[20];
char* ptr = strtok(buffer, delim); // ptr = "SYN"
ptr = strtok(NULL, delim); // ptr = "ACK"
ptr = strtok(NULL, delim); // ptr = port_data
strcpy(port, ptr);
int port_data = atoi(port);
return port_data;
}
int main(int argc, char* argv[])
{
// INITIALIZATION
int port_sync, port_data;
char addr[15];
if (argc < 3)
{
printf("Too few arguments given.\n");
printf("Format: ./client <adresseIP> <port>\n");
exit(1);
}
else if (argc > 3)
{
printf("Too many arguments given.\n");
printf("Format: ./client <adresseIP> <port>\n");
exit(1);
}
else
{
strcpy(addr, argv[1]);
port_sync = atoi(argv[2]);
}
// SOCKET OPTION
int opt = 1;
// SYNCHRONIZATION SOCKET
struct sockaddr_in client_addr;
memset((char*)&client_addr, 0, sizeof(client_addr));
int socket_fd;
if ((socket_fd = socket(DOMAIN, TYPE, PROTOCOL)) < 0)
{
perror("[-] Connection error");
exit(1);
}
printf("[+] Synchronization socket created.\n");
if(setsockopt(socket_fd, SOL_SOCKET, SO_REUSEADDR, (char *)&opt, sizeof(opt)) < 0)
{
perror("[-] setsockopt error");
exit(1);
}
client_addr.sin_family = DOMAIN;
client_addr.sin_port = htons(port_sync);
client_addr.sin_addr.s_addr = INADDR_ANY;
socklen_t socket_length = sizeof(client_addr);
// CONNECTION
int online = 1;
int sync_done = 0;
char buffer[BUFFER_SIZE];
while (online)
{
if (!sync_done)
{
memset(buffer, 0, sizeof(buffer));
sprintf(buffer, "SYN");
sendto(socket_fd, buffer, strlen(buffer), 0, (struct sockaddr*) & client_addr, sizeof(client_addr));
recvfrom(socket_fd, buffer, BUFFER_SIZE, 0, (struct sockaddr*) & client_addr, &socket_length);
if (strstr(buffer, "SYN-ACK") != NULL)
{
port_data = get_data_port(buffer);
printf("New port: %d\n", port_data);
memset(buffer, 0, sizeof(buffer));
sprintf(buffer, "ACK");
sendto(socket_fd, buffer, strlen(buffer), 0, (struct sockaddr*) & client_addr, sizeof(client_addr));
}
memset(buffer, 0, sizeof(buffer));
printf("Finished synchronization\n");
client_addr.sin_port = htons(port_data);
sync_done = 1;
socket_length = sizeof(client_addr);
}
else
{
printf("UDP Client: ");
fgets(buffer, BUFFER_SIZE, stdin);
buffer[strcspn(buffer, "\n")] = 0;
sendto(socket_fd, buffer, sizeof(buffer), 0, (struct sockaddr*)& client_addr, sizeof(client_addr));
if (strcmp(buffer, ":exit") == 0)
{
close(socket_fd);
printf("[-] Disconnected from server\n");
exit(1);
}
int recvfrom_return;
if ((recvfrom_return = recvfrom(socket_fd, buffer, sizeof(buffer), 0, (struct sockaddr*) & client_addr, &socket_length)) < 0)
{
perror("[-] Data reception error");
}
else
{
buffer[recvfrom_return] = 0;
printf("Server: %s\n", buffer);
}
}
}
return 0;
}
Я проверил многочисленные онлайн-примеры и учебные пособия о том, как изменить порт с помощью сокетной связи, и мне кажется, что я следовал всемрекомендации, но явно я что-то упускаю.
РЕДАКТИРОВАТЬ
Моя проблема заключается в том, что сервер никогда не получает клиентские сообщения после того, как я изменил порт с помощью client_addr.sin_port = htons(port_data);
. Я проверил на стороне сервера, и код вводит второй оператор if(FD_ISSET...)
. Мне не удалось точно определить причину, по которой сообщение не приходит.
Заранее спасибо и не стесняйтесь указывать, является ли это дубликатом (и связать дубликат вопроса)!