/*==============================================================================
* Name : void Client_start_handle(CONN_INFO* conn)
* Abstr : Start the connection with db server
* Params : CONN_INFO* conn : connection with db server
* Return :
* Modify :
*=============================================================================*/
void Client_start_handle(CONN_INFO* conn)
{
int8 err;
struct sockaddr_in addr; /* servant address info */
err = EXEC_SUCS;
memset(&addr, 0x00, sizeof(struct sockaddr_in));
/* initialize the socket fd */
err = Socket_init(&conn->fd);
if (!err) {
/* bind the address and port */
addr.sin_family = AF_INET;
addr.sin_port = htons(conn->port);
addr.sin_addr.s_addr = inet_addr(&conn->addr[0]);
/* start connection */
err = connect(conn->fd, (struct sockaddr *)(&addr), sizeof(struct sockaddr));
if ((err) && (errno != EINPROGRESS)) {
log_fatal(log_cat, "%s: connect failed [s_fd=%d] [%s]",
&conn->name[0], conn->fd, strerror(errno));
err = EXEC_FAIL;
} else {
err = EXEC_SUCS;
}
}
/* check the result of connect */
if (!err) {
Client_switch_to_test(conn);
} else {
Client_switch_to_stop(conn);
}
}
int echo_entry(config_opt config_opts[])
{
int i, port, socket_type;
char proto[5];
int worker_model = DISPATCHER_WORKER_MODEL_SINGLE;
int num_workers = 5;
for(i=0;;i++)
{
if (!strcmp(config_opts[i].name, ""))
{
// End of options
break;
}
if (!strcmp(config_opts[i].name, "port"))
{
port = atoi(config_opts[i].value);
if (port <= 0 || port >= 65535)
{
fprintf(stderr, "Invalid port specification: %s\n", config_opts[i].value);
return 1;
}
}
else if (!strcmp(config_opts[i].name, "proto"))
{
strncpy(proto, config_opts[i].value, sizeof(proto));
proto[sizeof(proto)-1] = 0;
if (!strcmp(proto, "tcp"))
{
socket_type = SOCK_STREAM;
}
else if (!strcmp(proto, "udp"))
{
socket_type = SOCK_DGRAM;
}
else
{
fprintf(stderr, "Invalid proto specification '%s'. Must be one of: tcp, udp\n", config_opts[i].value);
return 1;
}
}
else if (!strcmp(config_opts[i].name, "worker_model"))
{
if (!strcmp(config_opts[i].value, "single"))
{
worker_model = DISPATCHER_WORKER_MODEL_SINGLE;
}
else if (!strcmp(config_opts[i].value, "postfork"))
{
worker_model = DISPATCHER_WORKER_MODEL_POSTFORK;
}
else if (!strcmp(config_opts[i].value, "prefork"))
{
worker_model = DISPATCHER_WORKER_MODEL_PREFORK;
}
else if (!strcmp(config_opts[i].value, "thread"))
{
worker_model = DISPATCHER_WORKER_MODEL_THREAD;
}
else
{
fprintf(stderr, "Invalid worker_model specification: %s\n", config_opts[i].value);
return 1;
}
}
else if (!strcmp(config_opts[i].name, "num_workers"))
{
num_workers = atoi(config_opts[i].value);
if (num_workers <= 0 || num_workers > 200)
{
fprintf(stderr, "Invalid num_workers specification: %s\n", config_opts[i].value);
return 1;
}
}
else
{
fprintf(stderr, "Unsupported option: %s\n", config_opts[i].name);
return 1;
}
}
Socket *sock = Socket_init(0);
sock->create(sock, AF_INET, socket_type);
if (sock == NULL)
{
fprintf(stderr, "Failed to create socket\n");
return 1;
}
sock->set_flag(sock, O_NONBLOCK);
if (sock->bind(sock, "0.0.0.0", port) < 0)
{
fprintf(stderr, "Bind to socket failed\n");
return 1;
}
if (socket_type == SOCK_STREAM)
{
if (sock->listen(sock, 5) < 0)
{
fprintf(stderr, "Listen on socket failed\n");
return 1;
}
}
Dispatcher * disp = Dispatcher_init(worker_model, num_workers);
disp->add_listener(disp, sock, echo_dispatcher_poll_callback, echo_dispatcher_run_callback, echo_dispatcher_cleanup_callback);
disp->run(disp);
disp->destroy(disp);
sock->destroy(sock);
return 0;
}
