#include "socket.h"


void server(int port) {
	void *socket = shm_open_socket(REQ_REP);
	shm_bind(socket, port);
	shm_listen(socket);
	int size;
	void *recvbuf;
	char sendbuf[512];
	while(true) {
		shm_recv(socket, &recvbuf, &size);
		sprintf(sendbuf, "SERVER RECEIVED: %s", recvbuf);
		puts(sendbuf);
		shm_send(socket, sendbuf, strlen(sendbuf)+1) ;
		shm_free(recvbuf);

	}
	shm_close_socket(socket);
}

void client(int port) {
	void *socket = shm_open_socket(REQ_REP);
	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);
		shm_free(recvbuf);

	}
	shm_close_socket(socket);
}

int main(int argc, char *argv[]) {
  shm_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);
 shm_destroy();
 // fprintf(stderr, "Usage: reqrep %s|%s <URL> ...\n", "server", "client");
  return 0;
}