update

wangzhengquan

2020-07-23 b6043642f60ef23a7a100418cd4fec1251a98ad9

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
#include "mod_socket.h"
#include "shm_mm.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_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_destroy();
 
  return 0;
}