Project

General

Profile

Statistics
| Revision:

root / trunk / code / projects / colonet / ColonetServer / ColonetServer.cpp @ 409

History | View | Annotate | Download (6.41 KB)

1
/**
2
 * @file ColonetServer.cpp
3
 *
4
 * @brief colonet_server - primary server application for Colonet
5
 *
6
 * @author Jason Knichel
7
 * @author Eugene Marinelli
8
 * @date 10/31/06
9
 */
10

    
11
#include <arpa/inet.h>
12
#include <fcntl.h>
13
#include <errno.h>
14
#include <netinet/in.h>
15
#include <string.h>
16
#include <unistd.h>
17
#include <sys/select.h>
18
#include <sys/socket.h>
19

    
20
#include <colonet_wireless.h>
21

    
22
#include <ColonetServer.h>
23
#include <ConnectionPool.h>
24
#include <wirelessMessageHandler.h>
25
#include <options.h>
26
#include <Log.h>
27

    
28
#define LISTEN_BACKLOG 5
29
#define LOG_BUFFER_LENGTH 128
30

    
31
#define u_int32_t unsigned
32

    
33
static ConnectionPool * connection_pool;
34

    
35
/**
36
 * @brief Default constructor for ColonetServer
37
 */
38
ColonetServer::ColonetServer(): logger("logFile.txt") {
39
  listen_socket = 0;
40
}
41

    
42
/**
43
 * @brief Destructor for ColonetServer
44
 */
45
ColonetServer::~ColonetServer() {
46
}
47

    
48
/**
49
 * @brief Initializes the various elements needed for the server to run
50
 *
51
 * @param argc The number of command line arguments passed to the program
52
 * @param argv The command line arguments passed to the program
53
 *
54
 * @return 0 on success, negative error code on failure
55
 */
56
int ColonetServer::initialize_server(int argc, char * argv[]) {
57
  parseCmdLine(argc, argv);
58

    
59
  if (initialize_connection(optionsG.listen_port) < 0) {
60
    return -1;
61
  }
62

    
63
  if (initialize_wireless() < 0) {
64
    fprintf(stderr, "%s: initWireless failed\n", __FUNCTION__);
65
    return -1;
66
  }
67

    
68
  return 0;
69
}
70

    
71
/**
72
 * @brief Starts the server listening on the socket that was opened for listening
73
 *
74
 * @return 0 on success, negative error code on failure
75
 */
76
int ColonetServer::start_listening() {
77
  if (listen(listen_socket, LISTEN_BACKLOG) < 0) {
78
    logger.log_error("\t\nThere was an error telling the socket to "
79
                     "listen for connections from clients.  Terminating Server...\n");
80
    return -1;
81
  }
82
  return 0;
83
}
84

    
85

    
86
/**
87
 * @brief Starts the server running (starts an infinite loop)
88
 */
89
int ColonetServer::run_server() {
90
  connection_pool.add_new_socket_to_pool(listen_socket);
91

    
92
  int accept_socket = 0;
93
  struct sockaddr_in client_addr;
94
  socklen_t client_addr_size = sizeof(client_addr);
95

    
96
  logger.log_message("Server initialized.  About to start listening for connections");
97

    
98
  while(1) {
99
    usleep(10000);
100

    
101
    connection_pool.perform_select(listen_socket);
102

    
103
    //either no descriptors are ready or there was an error
104
    if (connection_pool.get_number_clients_ready() <= 0) {
105
      continue;
106
    }
107

    
108
    if (connection_pool.is_socket_ready_to_read(listen_socket)) {
109
      printf("Something is trying to connect...\n");
110
      if ((accept_socket = accept(listen_socket, (struct sockaddr*) &client_addr, &client_addr_size)) < 0) {
111
        if (errno == EMFILE) {
112
          printf("\tWhen attempting to accept a connection, "
113
                 "reached the per process limit of file descriptors."
114
                 "  Dropping the new connection.\n");
115
          continue;
116
        } else {
117
          printf("\tThere was an error when attempting to accept a connection");
118
        }
119
        continue;
120
      }
121

    
122
      char log_buffer[LOG_BUFFER_LENGTH];
123
      snprintf(log_buffer, LOG_BUFFER_LENGTH, "Client at address %s attempting to connect.",
124
               inet_ntoa(client_addr.sin_addr));
125
      logger.log_string(LOG_TYPE_CONNECT, log_buffer);
126

    
127
      if (connection_pool.add_client(accept_socket) < 0) {
128
        printf("\tThere was an error when trying to add a client to the connection pool.");
129
        continue;
130
      }
131

    
132
      snprintf(log_buffer, LOG_BUFFER_LENGTH, "Client at address %s successfully added to connection pool.",
133
               inet_ntoa(client_addr.sin_addr));
134
      logger.log_string(LOG_TYPE_CONNECT, log_buffer);
135
    }
136

    
137
    if (connection_pool.check_clients() < 0) {
138
      printf("\tThere was an error trying to update the clients.");
139
      continue;
140
    }
141
  }
142

    
143
  return 0;
144
}
145

    
146
int ColonetServer::process_received_wireless_message(unsigned char type, short source, int dest,
147
                                                     unsigned char * data, int len) {
148
  if (type == COLONET_RESPONSE) {
149
    printf("response\n");
150

    
151
    //Greg's code
152
    char * buffer = (char *) &(data[5]);
153
    int buflen = strlen(buffer);
154
    buffer[buflen] = '\n';
155
    buflen++;
156

    
157
    if (connection_pool.write_to_client(dest, buffer, buflen) == ERROR_INVALID_CLIENT_ID) {
158
      printf("The robot wanted to pass the data to a client not in the pool.\n");
159
      return -1;
160
    }
161

    
162
    printf("Put data in write buffer for client.\n");
163
  } else {
164
    printf("not a response\n");
165
    return -1;
166
  }
167

    
168
  return 0;
169
}
170

    
171

    
172
/**
173
 * @brief Initializes the wireless
174
 *
175
 * @return 0 on success, negative error code on error
176
 */
177
int ColonetServer::initialize_wireless() {
178
  char* log_filename = NULL;
179

    
180
  if (optionsG.logging_enabled) {
181
    printf("Logging enabled. Log filename: %s\n", optionsG.log_filename);
182
    log_filename = optionsG.log_filename;
183
  } else {
184
    printf("Logging disabled.\n");
185
  }
186

    
187
  if (colonet_wl_init(optionsG.wireless_port, wirelessMessageHandler, log_filename) != 0) {
188
    fprintf(stderr, "ERROR - colonet_wl_init failed.\n");
189
    return -1;
190
  }
191

    
192
  if (colonet_wl_run_listener_thread()) {
193
    fprintf(stderr, "%s: colonet_wl_run_listener_thread failed.\n", __FUNCTION__);
194
    return -1;
195
  }
196

    
197
  return 0;
198
}
199

    
200
/**
201
 * @brief Initialize a connection to listen on
202
 *
203
 * @port The port to try to open to listen on
204
 *
205
 * @return 0 on success, negative error code on error
206
 */
207
int ColonetServer::initialize_connection(int port) {
208
  printf("Initializing connection that will be used to listen for clients...\n");
209
  int options = 1;
210
  struct sockaddr_in my_address;
211

    
212
  //get a socket fd
213
  if ((listen_socket = socket(AF_INET, SOCK_STREAM, 0)) < 0) {
214
    printf("\tThere was an error creating socket\n");
215
    return -1;
216
  }
217

    
218
  //set up the address struct
219
  memset(&my_address,'\0',sizeof(my_address));
220
  my_address.sin_family = AF_INET;
221
  my_address.sin_addr.s_addr = htonl(INADDR_ANY);
222
  my_address.sin_port = htons(port);
223

    
224
  setsockopt(listen_socket, SOL_SOCKET, SO_REUSEADDR, &options, sizeof(options));
225

    
226
  //get the current socket options
227
  if ((options = fcntl(listen_socket, F_GETFL)) < 0) {
228
    printf("\tThere was an error getting the socket options.\n");
229
    return -1;
230
  }
231

    
232
  //set the socket to non blocking
233
  options = (options | O_NONBLOCK);
234
  if (fcntl(listen_socket, F_SETFL, options) < 0) {
235
    printf("\tThere was an error setting the socket to be non blocking.\n");
236
    return -1;
237
  }
238

    
239
  //bind the socket to listen on the specified port
240
  if (bind(listen_socket, (struct sockaddr *) &my_address, sizeof(my_address)) < 0) {
241
    printf("\tThere was an error binding the socket\n");
242
    return -1;
243
  }
244

    
245
  return 0;
246
}