42/42_SIDE_exam_06_miniserv
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

wip tester

Browse Source
This commit is contained in:
asus
2023-02-17 12:44:05 +01:00
parent 84357f3462
commit b401ebb5ab
17 changed files with 648 additions and 185 deletions
Download Patch File Download Diff File Expand all files Collapse all files

177
mini_serv_01.c
View File

@@ -3,28 +3,39 @@
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include <stdio.h>
#include <stdio.h> //sprintf
#define BUFSIZE 42000
typedef struct s_client {
int id;
char msg[1024];
} t_client;
} t_client;
// GLOBALS
int server_fd;
int id;
int max_fd;
socklen_t addr_len;
fd_set fdset;
t_client clients[FD_SETSIZE];
void error(char *msg) {
write(2, msg, strlen(msg));
void error(char *str) {
write(2, str, strlen(str));
exit(1);
}
int init_socket(struct sockaddr_in *addr, int len, int port) {
int server_fd;
if ( (server_fd = socket(AF_INET, SOCK_STREAM, 0)) == -1)
error("Fatal error2\n");
bzero(addr, len);
addr->sin_family = AF_INET;
addr->sin_addr.s_addr = htonl(2130706433); //127.0.0.1
addr->sin_port = htons(port);
if (bind(server_fd, (const struct sockaddr *)addr, len) == -1)
error("Fatal error3\n");
if (listen(server_fd, 10) == -1)
error("Fatal error4\n");
return server_fd;
}
void broadcast(char *buf, fd_set *set, int max_fd, int server_fd, int sender_fd) {
for(int i = 0; i <= max_fd; ++i) {
if (i == server_fd)
@@ -36,89 +47,87 @@ void broadcast(char *buf, fd_set *set, int max_fd, int server_fd, int sender_fd)
}
}
void handle_socket(struct sockaddr *addr) {
fd_set rdset;
int client_fd;
int ret;
t_client client;
char buf[BUFSIZE];
char msg[BUFSIZE];
rdset = fdset;
select(max_fd + 1, &rdset, NULL, NULL, NULL);
if (FD_ISSET(server_fd, &rdset)) {
client_fd = accept(server_fd, addr, &addr_len);
FD_SET(client_fd, &fdset);
if (client_fd > max_fd)
max_fd = client_fd;
client = clients[client_fd];
client.id = id;
bzero(client.msg, strlen(client.msg));
sprintf(buf, "server: client %d arrived\n", id);
}
client_fd = 0;
while(client_fd < max_fd) {
if (FD_ISSET(client_fd, &rdset)) {
bzero(buf, BUFSIZE);
ret = recv(client_fd, buf, BUFSIZE, 0);
if (ret == 0) {
sprintf(msg, "server: client %d just left\n", clients[client_fd].id);
broadcast(msg, &fdset, max_fd, server_fd, client_fd);
FD_CLR(client_fd, &fdset);
}
else if (ret > 0) {
client = clients[client_fd];
for(int i = 0, j = strlen(client.msg); i < ret; ++i, ++j) {
client.msg[j] = buf[i];
if (buf[i] == '\n') {
client.msg[j] = '\0';
sprintf(msg, "client %d: %s\n", client.id, client.msg);
broadcast(msg, &fdset, max_fd, server_fd, client_fd);
bzero(client.msg, strlen(client.msg));
j = -1;
}
}
}
}
client_fd++;
}
}
int main(int ac, char **av) {
struct sockaddr_in addr;
int server_fd;
int client_fd;
int client_id;
int port;
int max_fd;
int ret;
t_client clients[FD_SETSIZE];
t_client client;
char buf[BUFSIZE];
char msg[BUFSIZE];
struct sockaddr_in addr;
socklen_t addr_len;
fd_set fdset;
fd_set rdset;
if (ac != 2)
error("Fatal error\n");
port = atoi(av[1]);
error("Wrong number of arguments\n");
if ( (port = atoi(av[1])) == -1)
error("Fatal error1\n");
addr_len = sizeof(addr);
// socket create and verification
if ( (server_fd = socket(AF_INET, SOCK_STREAM, 0)) == -1)
error("Fatal error\n");
bzero(&addr, addr_len);
// assign IP, PORT
addr.sin_family = AF_INET;
addr.sin_addr.s_addr = htonl(2130706433); //127.0.0.1
addr.sin_port = htons(port);
// Binding newly created socket to given IP and verification
if (bind(server_fd, (const struct sockaddr *)&addr, addr_len) == -1)
error("Fatal error\n");
if (listen(server_fd, 10) == -1)
error("Fatal error\n");
server_fd = init_socket(&addr, addr_len, port);
FD_ZERO(&fdset);
FD_SET(server_fd, &fdset);
max_fd = server_fd;
client_id = 0;
bzero(&clients, sizeof(clients));
while(1)
handle_socket((struct sockaddr *)&addr);
while (1) {
rdset = fdset;
select(max_fd + 1, &rdset, NULL, NULL, NULL);
// new connection
if (FD_ISSET(server_fd, &rdset)) {
client_fd = accept(server_fd, (struct sockaddr *)&addr, &addr_len);
FD_SET(client_fd, &fdset);
clients[client_fd].id = client_id;
if (client_fd > max_fd)
max_fd = client_fd;
sprintf(msg, "server: client %d just arrived\n", client_id);
broadcast(msg, &fdset, max_fd, server_fd, client_fd);
client_id++;
}
// new message
client_fd = 0;
while(client_fd <= max_fd) {
ret = 1;
if (FD_ISSET(client_fd, &rdset)) {
bzero(buf, BUFSIZE);
ret = recv(client_fd, buf, BUFSIZE, 0);
if (ret == 0) {
sprintf(msg, "server: client %d just left\n", clients[client_fd].id);
broadcast(msg, &fdset, max_fd, server_fd, client_fd);
FD_CLR(client_fd, &fdset);
}
else if (ret > 0) {
client = clients[client_fd];
for(int i = 0, j = strlen(client.msg); i < ret; ++i, ++j) {
client.msg[j] = buf[i];
if (buf[i] == '\n') {
client.msg[j] = '\0';
sprintf(msg, "client %d: %s\n", client.id, client.msg);
broadcast(msg, &fdset, max_fd, server_fd, client_fd);
bzero(client.msg, strlen(client.msg));
j = -1;
}
}
}
}
client_fd++;
}
}
return (0);
}