vlad-refactor
This commit is contained in:
189
vlad/lab_8/server.c
Normal file
189
vlad/lab_8/server.c
Normal file
@@ -0,0 +1,189 @@
|
||||
// server_tcp_pairs.c
|
||||
// Лабораторная №8, протокол TCP (нечётные варианты, но здесь по заданию TCP).
|
||||
// Задача: во всех парах одинаковых символов второй символ заменить на пробел,
|
||||
// считать количество замен и сохранить обработанный текст в выходной файл.
|
||||
|
||||
#include <stdio.h> // printf, fprintf, perror, FILE, fopen, fclose, fwrite
|
||||
#include <stdlib.h> // exit, EXIT_FAILURE, strtol
|
||||
#include <string.h> // memset, strlen
|
||||
#include <errno.h> // errno
|
||||
#include <unistd.h> // close, read, write
|
||||
#include <arpa/inet.h> // sockaddr_in, inet_ntoa, htons, htonl, ntohs, ntohl
|
||||
#include <sys/socket.h> // socket, bind, listen, accept
|
||||
#include <sys/types.h> // типы сокетов
|
||||
#include <netinet/in.h> // структура sockaddr_in
|
||||
|
||||
#define BUF_SIZE 4096 // Размер буфера приёма из TCP-сокета.
|
||||
|
||||
// Функция обрабатывает блок данных, заменяя во всех парах одинаковых подряд
|
||||
// идущих символов второй символ пары на пробел. Пары могут пересекать границу
|
||||
// буферов, поэтому учитывается последний символ предыдущего блока.
|
||||
static void process_block_pairs(char *buf,
|
||||
ssize_t len,
|
||||
int *has_prev,
|
||||
unsigned char *prev_char,
|
||||
long long *repl_counter) {
|
||||
if (!buf || !has_prev || !prev_char || !repl_counter) {
|
||||
return; // Проверка входных указателей.
|
||||
}
|
||||
|
||||
for (ssize_t i = 0; i < len; i++) {
|
||||
unsigned char c = (unsigned char) buf[i];
|
||||
|
||||
if (*has_prev) {
|
||||
// Есть предыдущий символ из этого же потока.
|
||||
if (c == *prev_char) {
|
||||
// Найдена пара одинаковых подряд символов.
|
||||
buf[i] = ' '; // Заменяем второй символ пары на пробел.
|
||||
(*repl_counter)++; // Увеличиваем счётчик замен.
|
||||
}
|
||||
*has_prev = 0; // Сбрасываем флаг: пара уже обработана.
|
||||
} else {
|
||||
// Текущий символ станет предыдущим для возможной пары в следующем байте.
|
||||
*prev_char = c;
|
||||
*has_prev = 1;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
int main(int argc, char *argv[]) {
|
||||
// Ожидаемые аргументы:
|
||||
// argv[1] - порт TCP сервера (например, 5000).
|
||||
// argv[2] - имя выходного файла.
|
||||
if (argc < 3) {
|
||||
fprintf(stderr,
|
||||
"Usage: %s <port> <output_file>\n",
|
||||
argv[0]);
|
||||
fprintf(stderr,
|
||||
"Example: %s 5000 out.txt\n",
|
||||
argv[0]);
|
||||
return -1;
|
||||
}
|
||||
|
||||
// Разбор номера порта.
|
||||
char *endptr = NULL;
|
||||
errno = 0;
|
||||
long port_long = strtol(argv[1], &endptr, 10);
|
||||
if (errno != 0 || endptr == argv[1] || *endptr != '\0' ||
|
||||
port_long <= 0 || port_long > 65535) {
|
||||
fprintf(stderr, "ERROR: invalid port '%s'\n", argv[1]);
|
||||
return -1;
|
||||
}
|
||||
int port = (int) port_long;
|
||||
|
||||
const char *out_path = argv[2];
|
||||
|
||||
// Открываем выходной файл для записи (перезаписываем).
|
||||
FILE *fout = fopen(out_path, "w");
|
||||
if (!fout) {
|
||||
perror("fopen(output_file)");
|
||||
return -1;
|
||||
}
|
||||
|
||||
// Создаём TCP-сокет.
|
||||
int listen_fd = socket(AF_INET, SOCK_STREAM, 0);
|
||||
if (listen_fd < 0) {
|
||||
perror("socket");
|
||||
fclose(fout);
|
||||
return -1;
|
||||
}
|
||||
|
||||
// Включаем SO_REUSEADDR, чтобы можно было быстро перезапускать сервер.
|
||||
int optval = 1;
|
||||
if (setsockopt(listen_fd, SOL_SOCKET, SO_REUSEADDR,
|
||||
&optval, sizeof(optval)) < 0) {
|
||||
perror("setsockopt(SO_REUSEADDR)");
|
||||
close(listen_fd);
|
||||
fclose(fout);
|
||||
return -1;
|
||||
}
|
||||
|
||||
// Заполняем структуру адреса сервера.
|
||||
struct sockaddr_in servaddr;
|
||||
memset(&servaddr, 0, sizeof(servaddr));
|
||||
servaddr.sin_family = AF_INET;
|
||||
servaddr.sin_addr.s_addr = htonl(INADDR_ANY); // Принимать на всех интерфейсах.
|
||||
servaddr.sin_port = htons(port);
|
||||
|
||||
// Привязываем сокет к адресу и порту.
|
||||
if (bind(listen_fd, (struct sockaddr *) &servaddr, sizeof(servaddr)) < 0) {
|
||||
perror("bind");
|
||||
close(listen_fd);
|
||||
fclose(fout);
|
||||
return -1;
|
||||
}
|
||||
|
||||
// Переводим сокет в режим прослушивания.
|
||||
if (listen(listen_fd, 1) < 0) {
|
||||
perror("listen");
|
||||
close(listen_fd);
|
||||
fclose(fout);
|
||||
return -1;
|
||||
}
|
||||
|
||||
printf("TCP server (pairs) listening on port %d, output file: %s\n",
|
||||
port, out_path);
|
||||
|
||||
// Принимаем одно соединение от клиента.
|
||||
struct sockaddr_in cliaddr;
|
||||
socklen_t cli_len = sizeof(cliaddr);
|
||||
int conn_fd = accept(listen_fd, (struct sockaddr *) &cliaddr, &cli_len);
|
||||
if (conn_fd < 0) {
|
||||
perror("accept");
|
||||
close(listen_fd);
|
||||
fclose(fout);
|
||||
return -1;
|
||||
}
|
||||
|
||||
printf("Client connected from %s:%d\n",
|
||||
inet_ntoa(cliaddr.sin_addr),
|
||||
ntohs(cliaddr.sin_port));
|
||||
|
||||
close(listen_fd); // Больше новых клиентов не принимаем.
|
||||
|
||||
char buf[BUF_SIZE];
|
||||
long long repl_counter = 0; // Общее количество замен.
|
||||
int has_prev = 0; // Был ли предыдущий символ для пары.
|
||||
unsigned char prev_char = 0; // Предыдущий символ, если has_prev == 1.
|
||||
|
||||
for (;;) {
|
||||
// Считываем данные из TCP-сокета.
|
||||
ssize_t n = read(conn_fd, buf, sizeof(buf));
|
||||
if (n > 0) {
|
||||
// Обрабатываем блок на предмет пар одинаковых символов.
|
||||
process_block_pairs(buf, n, &has_prev, &prev_char, &repl_counter);
|
||||
|
||||
// Пишем обработанный блок в выходной файл.
|
||||
if (fwrite(buf, 1, (size_t) n, fout) != (size_t) n) {
|
||||
perror("fwrite");
|
||||
repl_counter = -1;
|
||||
break;
|
||||
}
|
||||
} else if (n == 0) {
|
||||
// Клиент корректно закрыл соединение.
|
||||
printf("Client disconnected.\n");
|
||||
break;
|
||||
} else {
|
||||
// Ошибка при чтении.
|
||||
perror("read");
|
||||
repl_counter = -1;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
// Закрываем соединение.
|
||||
close(conn_fd);
|
||||
|
||||
// Закрываем файл.
|
||||
if (fclose(fout) == EOF) {
|
||||
perror("fclose(output_file)");
|
||||
repl_counter = -1;
|
||||
}
|
||||
|
||||
printf("Total replacements (pairs): %lld\n", repl_counter);
|
||||
if (repl_counter < 0) {
|
||||
return -1;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
Reference in New Issue
Block a user