#include "shm_socket.h"
#include "usg_common.h"
#include "shm_mm_wraper.h"
typedef struct Targ {
	int port;
	int id;
}Targ;

void * precess_client(void *_socket) {
	pthread_detach(pthread_self());
	shm_socket_t *socket = (shm_socket_t *)_socket;
	int size;
	void *recvbuf;
	char sendbuf[512];
	while (shm_recv(socket, &recvbuf, &size) == 0 ) {
		sprintf(sendbuf, "SERVER RECEIVED: %s", recvbuf);
		puts(sendbuf);
		shm_send(socket, sendbuf, strlen(sendbuf)+1);
		free(recvbuf);
	}
	shm_close_socket(socket);
}

void server(int port) {
	pthread_t tid;
	shm_socket_t *socket = shm_open_socket();
	shm_socket_bind(socket, port);
	shm_listen(socket);
	shm_socket_t *client_socket;
	while(true) {
		client_socket = shm_accept(socket);
// printf("server messageQueue = %p\n", client_socket->messageQueue);
		pthread_create(&tid, NULL, precess_client , (void *)client_socket);
	}

	
}

void client(int port) {
	shm_socket_t *socket = shm_open_socket();
	shm_connect(socket, port);
	int size;
	void *recvbuf;
	char sendbuf[512];
	while(true) {
		printf("request: ");
		scanf("%s", sendbuf);
		shm_send(socket, sendbuf, strlen(sendbuf)+1) ;
		shm_recv(socket, &recvbuf, &size);
		printf("reply: %s\n", (char *)recvbuf);
		free(recvbuf);

	}
	shm_close_socket(socket);
}

void client_send(shm_socket_t *socket, char *sendbuf) {
	
	int size;
	void *recvbuf;
	printf("requst:%s\n", sendbuf);
	shm_send(socket, sendbuf, strlen(sendbuf)+1) ;
	shm_recv(socket, &recvbuf, &size);
	printf("reply: %s\n", (char *)recvbuf);
	free(recvbuf);

	
}


void multyProcessorsClient(int port) {

	int status, i = 0, processors = 4, scale = 100000;
	pid_t productors[processors];
	pid_t pid;
	char sendbuf[512];
	for ( i = 0; i < processors; i++) {
		if ((productors[i] = fork()) == 0)     /* Child runs user job */
		{
			shm_socket_t *socket = shm_open_socket();
			shm_connect(socket, port);
			while( scale-- > 0) {
				sprintf(sendbuf, "processor(%d) %d", i, scale);
				client_send(socket, sendbuf);
			}
			shm_close_socket(socket);
			exit(0);
		}
	}

	while ((pid = waitpid(-1, &status, 0)) > 0) {
		if(WIFEXITED(status)) {
			//fprintf(stderr, "child %d terminated normally with exit status=%d\n", pid, WEXITSTATUS(status));
		}else
			fprintf(stderr, "child %d terminated abnormally\n", pid);
	}

	if (errno != ECHILD)
		perror("waitpid error");
}

void *threadrun(void *arg) {
	Targ * targ = ( Targ * )arg;
	int port = targ->port;
	char sendbuf[512];
	int scale = 100000;
	int i;
	shm_socket_t *socket = shm_open_socket();
	shm_connect(socket, port);
	for( i = 0; i<scale; i++) {
		sprintf(sendbuf, "thread(%d) %d", targ->id, i);
		
		client_send(socket, sendbuf);
	}
	shm_close_socket(socket);
	return (void*)i;
}

void multyThreadClient(int port) {

	int status, i = 0, processors = 4;
	void *res[processors];
	Targ *targs= (Targ*)calloc(processors, sizeof(Targ));
	pthread_t tids[processors];
	char sendbuf[512];
	for ( i = 0; i < processors; i++) {
		targs[i].port = port;
		targs[i].id = i;
		pthread_create(&tids[i], NULL, threadrun, (void *)&targs[i]);
		
	}

	for (i = 0; i< processors; i++) {
	    if(pthread_join(tids[i], &res[i])!=0) {
	      perror("multyThreadClient pthread_join");
	    } else {
	      fprintf(stderr, "client(%d) 写入 %ld 条数据\n", i, (long)res[i]);
	    }
	  }

	
}

int main(int argc, char *argv[]) {
  shm_mm_wrapper_init(512);
  int port;
  if (argc < 3) {
  	 fprintf(stderr, "Usage: reqrep %s|%s <PORT> ...\n", "server", "client");
  	 return 1;
  }

  port = atoi(argv[2]);

  if (strcmp("server", argv[1]) == 0 ) {
     server(port);
  }

  if (strcmp("client", argv[1]) == 0)
     client(port);

 if (strcmp("mclient", argv[1]) == 0)
     multyThreadClient(port);

  shm_mm_wrapper_destroy();
 // fprintf(stderr, "Usage: reqrep %s|%s <URL> ...\n", "server", "client");
  return 0;
}