int ini_server(const char * url, Protocol * p, TCFBroadcastGroup * b) {
ChannelServer * serv = NULL;
PeerServer * ps = NULL;
Trap trap;
if (!set_trap(&trap)) {
bcg = NULL;
proto = NULL;
if (ps != NULL) peer_server_free(ps);
errno = trap.error;
return -1;
}
bcg = b;
proto = p;
ps = channel_peer_from_url(url);
if (ps == NULL) str_exception(ERR_OTHER, "Invalid server URL");
peer_server_addprop(ps, loc_strdup("Name"), loc_strdup(PROXY_NAME));
peer_server_addprop(ps, loc_strdup("Proxy"), loc_strdup(""));
SERVER_ADDPROP_HOOK;
serv = channel_server(ps);
if (serv == NULL) exception(errno);
serv->new_conn = channel_new_connection;
clear_trap(&trap);
add_channel_redirection_listener(channel_redirection_listener);
return 0;
}
static void connect_dest(void * x) {
Channel * c1 = (Channel *)x;
PeerServer * ps;
ConnectInfo * info;
ps = channel_peer_from_url(dest_url);
if (ps == NULL) {
trace(LOG_ALWAYS, "cannot parse peer url: %s", dest_url);
channel_close(c1);
return;
}
channel_lock(c1);
c1->state = ChannelStateRedirectReceived;
info = (ConnectInfo *)loc_alloc_zero(sizeof(ConnectInfo));
info->ps = ps;
info->c1 = c1;
channel_connect(ps, connect_done, info);
}
static int lua_peer_server_from_url(lua_State *L)
{
PeerServer *ps;
const char *url;
assert(L == luastate);
if(lua_gettop(L) != 1 || !lua_isstring(L, 1)) {
luaL_error(L, "wrong number or type of arguments");
}
url = lua_tostring(L, 1);
ps = channel_peer_from_url(url);
trace(LOG_LUA, "lua_peer_server_from_url %s %p", url, ps);
if(ps == NULL) luaL_error(L, "cannot parse url: %s", lua_tostring(L, 1));
if(ps->id == NULL) {
peer_server_addprop(ps, loc_strdup("ID"), loc_strdup(lua_tostring(L, 1)));
}
lua_alloc_pse(L, ps);
return 1;
}
int tcf_log(void) {
#else
int main(int argc, char ** argv) {
int c;
int ind;
const char * log_name = "-";
#endif
const char * url = "TCP:";
PeerServer * ps;
ChannelServer * serv;
int print_server_properties = 0;
ini_mdep();
ini_trace();
ini_events_queue();
ini_asyncreq();
log_mode = LOG_TCFLOG;
#if defined(_WRS_KERNEL)
progname = "tcf";
open_log_file("-");
#else
progname = argv[0];
/* Parse arguments */
for (ind = 1; ind < argc; ind++) {
char * s = argv[ind];
if (*s++ != '-') break;
while (s && (c = *s++) != '\0') {
switch (c) {
case 'h':
show_help();
exit(0);
case 'n':
auto_redirect = 0;
break;
case 'S':
print_server_properties = 1;
break;
#if ENABLE_Trace
case 'l':
#endif
case 'L':
case 's':
case 'f':
if (*s == '\0') {
if (++ind >= argc) {
fprintf(stderr, "%s: error: no argument given to option '%c'\n", progname, c);
exit(1);
}
s = argv[ind];
}
switch (c) {
#if ENABLE_Trace
case 'l':
if (parse_trace_mode(s, &log_mode) != 0) {
fprintf(stderr, "Cannot parse log level: %s\n", s);
exit(1);
}
break;
#endif
case 'L':
log_name = s;
break;
case 's':
url = s;
break;
case 'f':
if (filter_add_message_filter(s) != 0) {
fprintf(stderr, "Cannot parse filter level: %s\n", s);
exit(1);
}
break;
default:
fprintf(stderr, "%s: error: illegal option '%c'\n", progname, c);
show_help();
exit(1);
}
s = NULL;
break;
default:
ILLEGAL_OPTION_HOOK;
fprintf(stderr, "%s: error: illegal option '%c'\n", progname, c);
show_help();
exit(1);
}
}
}
open_log_file(log_name);
if (ind < argc) {
dest_url = argv[ind++];
if (!auto_redirect) {
fprintf(stderr, "Automatic redirect disabled: argument '%s' ignored\n", dest_url);
dest_url = NULL;
}
}
#endif
bcg = broadcast_group_alloc();
/* Default filters (use "-fi" to disable). */
filter_add_message_filter("o,E,Locator,peerHeartBeat");
filter_add_message_filter("o,E,Locator,peerAdded");
filter_add_message_filter("o,E,Locator,peerRemoved");
filter_add_message_filter("o,E,Locator,peerChanged");
filter_add_message_filter("o,C,Locator,getPeers");
filter_add_message_filter("tr10,C,FileSystem,read");
filter_add_message_filter("t10,C,FileSystem,write");
set_proxy_log_filter_listener2(filter_is_log_filtered);
ps = channel_peer_from_url(url);
if (ps == NULL) {
fprintf(stderr, "%s: invalid server URL (-s option value): %s\n", progname, url);
exit(1);
}
peer_server_addprop(ps, loc_strdup("Name"), loc_strdup("TCF Protocol Logger"));
peer_server_addprop(ps, loc_strdup("Proxy"), loc_strdup(""));
SERVER_ADDPROP_HOOK;
serv = channel_server(ps);
if (serv == NULL) {
fprintf(stderr, "%s: cannot create TCF server: %s\n", progname, errno_to_str(errno));
exit(1);
}
serv->new_conn = channel_new_connection;
discovery_start();
if (print_server_properties) {
ChannelServer * s;
char * server_properties;
assert(!list_is_empty(&channel_server_root));
s = servlink2channelserverp(channel_server_root.next);
server_properties = channel_peer_to_json(s->ps);
printf("Server-Properties: %s\n", server_properties);
fflush(stdout);
trace(LOG_ALWAYS, "Server-Properties: %s", server_properties);
loc_free(server_properties);
}
/* Process events - must run on the initial thread since ptrace()
* returns ECHILD otherwise, thinking we are not the owner. */
run_event_loop();
return 0;
}
int tcf_log(void) {
#else
int main(int argc, char ** argv) {
int c;
int ind;
const char * log_name = "-";
#endif
const char * url = "TCP:";
PeerServer * ps;
ChannelServer * serv;
ini_mdep();
ini_trace();
ini_events_queue();
ini_asyncreq();
log_mode = LOG_TCFLOG;
#if defined(_WRS_KERNEL)
progname = "tcf";
open_log_file("-");
#else
progname = argv[0];
/* Parse arguments */
for (ind = 1; ind < argc; ind++) {
const char * s = argv[ind];
if (*s != '-') {
break;
}
s++;
while ((c = *s++) != '\0') {
switch (c) {
case 'h':
show_help();
exit (0);
case 'l':
case 'L':
case 's':
if (*s == '\0') {
if (++ind >= argc) {
fprintf(stderr, "%s: error: no argument given to option '%c'\n", progname, c);
exit(1);
}
s = argv[ind];
}
switch (c) {
case 'l':
if (parse_trace_mode(s, &log_mode) != 0) {
fprintf(stderr, "Cannot parse log level: %s\n", s);
exit(1);
}
break;
case 'L':
log_name = s;
break;
case 's':
url = s;
break;
default:
fprintf(stderr, "%s: error: illegal option '%c'\n", progname, c);
show_help();
exit(1);
}
s = "";
break;
default:
fprintf(stderr, "%s: error: illegal option '%c'\n", progname, c);
show_help();
exit(1);
}
}
}
open_log_file(log_name);
if (ind < argc) {
dest_url = argv[ind++];
}
#endif
ps = channel_peer_from_url(url);
if (ps == NULL) {
fprintf(stderr, "%s: invalid server URL (-s option value): %s\n", progname, url);
exit(1);
}
peer_server_addprop(ps, loc_strdup("Name"), loc_strdup("TCF Protocol Logger"));
peer_server_addprop(ps, loc_strdup("Proxy"), loc_strdup(""));
serv = channel_server(ps);
if (serv == NULL) {
fprintf(stderr, "%s: cannot create TCF server: %s\n", progname, errno_to_str(errno));
exit(1);
}
serv->new_conn = channel_new_connection;
discovery_start();
/* Process events - must run on the initial thread since ptrace()
* returns ECHILD otherwise, thinking we are not the owner. */
run_event_loop();
return 0;
}
