#include "mod_socket.h"
|
#include "shm_mm_wraper.h"
|
#include "usg_common.h"
|
|
#define SERVER "server"
|
#define CLIENT "client"
|
#define DATE "DATE"
|
|
char *date(void) {
|
time_t now = time(&now);
|
struct tm *info = localtime(&now);
|
char *text = asctime(info);
|
text[strlen(text) - 1] = '\0'; // remove '\n'
|
return (text);
|
}
|
|
int server(const int port) {
|
int rv;
|
|
void *socket = mod_open_socket(PUB_SUB);
|
mod_socket_bind(socket, port);
|
if ((rv = mod_listen(socket)) != 0) {
|
printf("mod_listen");
|
}
|
for (;;) {
|
printf("SERVER: SENDING DATE SURVEY REQUEST\n");
|
if ((rv = mod_send(socket, DATE, strlen(DATE) + 1)) != 0) {
|
printf("mod_send");
|
}
|
|
for (;;) {
|
char *buf = NULL;
|
int sz;
|
rv = mod_recv(socket, (void **)&buf, &sz);
|
|
if (rv != 0) {
|
printf("mod_recv");
|
}
|
printf("SERVER: RECEIVED \"%s\" SURVEY RESPONSE\n", buf);
|
mod_free(buf, sz);
|
}
|
|
printf("SERVER: SURVEY COMPLETE\n");
|
}
|
}
|
|
int client(const int port, const char *name) {
|
int rv;
|
|
void *socket = mod_open_socket(PUB_SUB);
|
if ((rv = mod_connect(socket, port)) != 0) {
|
printf("mod_connect");
|
}
|
for (;;) {
|
char *buf = NULL;
|
int sz;
|
if ((rv = mod_recv(socket, (void **)&buf, &sz)) == 0) {
|
printf("CLIENT (%s): RECEIVED \"%s\" SURVEY REQUEST\n", name, buf);
|
mod_free(buf, sz);
|
char *d = date();
|
printf("CLIENT (%s): SENDING DATE SURVEY RESPONSE\n", name);
|
if ((rv = mod_send(socket, d, strlen(d) + 1, 0)) != 0) {
|
printf("mod_send");
|
}
|
}
|
}
|
}
|
|
int main(const int argc, const char **argv) {
|
shm_mm_wrapper_init(512);
|
int port;
|
if ((argc >= 2) && (strcmp(SERVER, argv[1]) == 0)) {
|
port = atoi(argv[2]);
|
(server(port));
|
}
|
else if ((argc >= 3) && (strcmp(CLIENT, argv[1]) == 0)) {
|
port = atoi(argv[2]);
|
(client(port, argv[3]));
|
} else {
|
fprintf(stderr, "Usage: survey %s|%s <URL> <ARG> ...\n", SERVER, CLIENT);
|
}
|
|
shm_mm_wrapper_destroy();
|
|
return 0;
|
}
|
