void* handle_file_receive(void* pdata)
{
struct info* pinfo = pdata;
int sock, client, addrlen, r;
struct sockaddr_in server_info;
struct sockaddr_in client_addr;
char port[6];
char bufin[RW_BUFFER];
FILE* file;
system("mkdir inbox 2>/dev/null");
memset(bufin, 0, RW_BUFFER);
pthread_mutex_lock( &(pinfo->mutex) );
sprintf(bufin, "./inbox/%s", pinfo->filename);
file = fopen(bufin, "wb+");
memset(bufin, 0, RW_BUFFER);
memset(port, 0, 6);
sprintf(port, "%d", pinfo->port);
pthread_mutex_unlock( &(pinfo->mutex) );
if(file == NULL)
{
printf("Error with fopen in handle_file_receive - directory inbox does not exist\n");
file = fopen("./RE216_pwd", "wb+");
if(file != NULL) fclose(file);
return NULL;
}
// Create a server
sock = do_socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
init_serv_addr(port, &server_info);
do_bind(sock, &server_info);
listen(sock, 1);
client = accept(sock, (struct sockaddr*)&client_addr, &addrlen);
int a = 0;
do {
r = read(client, bufin, 1024);
a = fwrite(bufin, 1, r, file);
} while(r != 0);
fclose(file);
printf("\n> File received\n> ");
fflush(stdout);
close(client);
close(sock);
}
void main(int argc, char** argv) {
if (argc != 2) {
fprintf(stderr, "please specify local port to bind to\n");
exit(-1);
}
int server_sock;
int max_fd;
struct sockaddr_in * serv_addr = malloc(sizeof(*serv_addr));
struct sockaddr_in * addr = malloc(sizeof(*addr));
char buffer[256];
int users[15];
int most_recent_user = 0;
fd_set rdfs;
fd_set rdfs_utd;
//init a fresh socket
server_sock = do_socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
max_fd = server_sock;
//init server address structure
init_serv_addr(argv[1],serv_addr);
//perform the binding
do_bind(server_sock, serv_addr, sizeof(*serv_addr));
//specify the socket to be a server socket and listen for at most 20 concurrent client
if(listen(server_sock, 20) == -1)
error("ERROR Listen");
//init the fdset
FD_ZERO(&rdfs);
FD_SET(server_sock, &rdfs);
rdfs_utd = rdfs;
//in an infinite loop, listen to the stored sockets
for (;;) {
//use select to listen to sockets
if(select(max_fd+1, &rdfs, NULL, NULL, NULL) == -1){
error("ERROR Server Select");
}
int current_fd;
for(current_fd = 3; current_fd <=max_fd; current_fd++){
// Implement the listening for incoming sockets
if(current_fd == server_sock){ // Activité sur socket Serveur = Nouveau client
if(FD_ISSET(server_sock, &rdfs)){
int new_sock = do_accept(server_sock, (struct sockaddr *) addr, (socklen_t) sizeof(addr));
if (++most_recent_user > 2){
do_write(new_sock, "SATURATION : we can't accept further communication. Try later\n");
fprintf(stdout, "Attempt of connection but saturation\n");
close(new_sock);
//on ferme tout et on envoie un message de saturation du serveur
}
// on rajoute une nouvelle socket pour un nouvel utilisateurs
FD_SET(new_sock, &rdfs_utd);
users[most_recent_user] = new_sock;
//Mettre à jour le plus grand descripteur à considérer
max_fd = new_sock > max_fd ? new_sock : max_fd;
// Nettoyer le buffer
memset(buffer, 0, 256);
// Read what the client has to say
int lu = read(new_sock, buffer, (int) 256);
//Test the value of the message
if (strcmp(buffer,"/quit\n") == 0){
fprintf(stdout, "[CLIENT] : %s", buffer);
do_write(new_sock, "You will be terminated\n");
FD_CLR(new_sock, &rdfs);
close(new_sock);
printf("The client n°%i is now disconnected\n", (current_fd-3));
max_fd -= 1;
}else{
// Display on the server monitor who is saying what
fprintf(stdout, "[CLIENT n° %i] : %s", new_sock-3, buffer);
// we write back to the client
do_write(new_sock, buffer);
}
}
}
// then implement the listening for already connected socket that write data to the server
else{
if(FD_ISSET(current_fd, &rdfs)){
// Nettoyer le buffer
memset(buffer, 0, 256);
// Read what the client have to say
int lu = read(current_fd, buffer, (int) 256);
//Test the value of the message
if (strcmp(buffer,"/quit\n") == 0){
fprintf(stdout, "[CLIENT] : %s", buffer);
do_write(current_fd, "You will be terminated\n");
FD_CLR(current_fd, &rdfs);
close(current_fd);
printf("The client n°%i is now disconnected\n", (current_fd-3));
max_fd -= 1;
}else{
// Display on the server monitor who is saying what
fprintf(stdout, "[CLIENT n° %i] : %s",(current_fd-3), buffer);
// we write back to the client
do_write(current_fd, buffer);
}
}
}
}
// Mettre à jour le fd d'activité
rdfs = rdfs_utd;
}
}
int main(int argc, char *argv[])
{
int listenfd = 0, connfd = 0;
struct sockaddr_in serv_addr;
init_serv_addr(&serv_addr);
char sendBuff[1025];
memset(sendBuff, '0', sizeof(sendBuff));
time_t ticks;
listenfd = socket(AF_INET, SOCK_STREAM, 0); //TCP
if (listenfd == -1)
{
printf("Cannot create socket\n");
return 1;
}
int bound = bind(listenfd, (struct sockaddr*)&serv_addr, sizeof(serv_addr));
if (bound == -1)
{
printf("Cannot bind socket\n");
return 1;
}
int listening = listen(listenfd, 10);
if (listening == -1)
{
printf("Cannot listen on socket\n");
return 1;
}
//while(1)
{
connfd = accept(listenfd, (struct sockaddr*)NULL, NULL);
snprintf(sendBuff, sizeof(sendBuff), "Welcome to the login page\nPlease provide the correct password: "******"Password is correct!\n");
else
printf("Incorrect password: %s\n", sendBuff);
close(connfd);
sleep(1);
}
}