int main(int argc, char* argv[])
{
// expect one argument
if (argc != 2)
{
std::cerr << "usage: " << argv[0] << " url-list-file" << std::endl;
return 1;
}
// this example program downloads all urls in the text file argv[1] to the current directory
char* url_file_name = argv[1];
// start by creating an io_service object
boost::asio::io_service io_service;
// construct an instance of curl::multi
curl::multi manager(io_service);
// treat each line in url_file_name as url and start a download from it
std::ifstream url_file(url_file_name);
while (!url_file.eof())
{
std::string url;
std::getline(url_file, url);
start_download(manager, url);
}
// let Boost.Asio do its magic
io_service.run();
std::cout << "All downloads completed" << std::endl;
return 0;
}
/**
* \brief Callback of the HTTP client.
*
* \param[in] module_inst Module instance of HTTP client module.
* \param[in] type Type of event.
* \param[in] data Data structure of the event. \refer http_client_data
*/
static void http_client_callback(struct http_client_module *module_inst, int type, union http_client_data *data)
{
switch (type) {
case HTTP_CLIENT_CALLBACK_SOCK_CONNECTED:
printf("Http client socket connected\r\n");
break;
case HTTP_CLIENT_CALLBACK_REQUESTED:
printf("Request completed\r\n");
add_state(GET_REQUESTED);
break;
case HTTP_CLIENT_CALLBACK_RECV_RESPONSE:
printf("Received response %u data size %u\r\n",
(unsigned int)data->recv_response.response_code,
(unsigned int)data->recv_response.content_length);
if ((unsigned int)data->recv_response.response_code == 200) {
http_file_size = data->recv_response.content_length;
received_file_size = 0;
}
else {
add_state(CANCELED);
return;
}
break;
case HTTP_CLIENT_CALLBACK_RECV_CHUNKED_DATA:
store_file_packet(data->recv_chunked_data.data, data->recv_chunked_data.length);
if (data->recv_chunked_data.is_complete) {
add_state(COMPLETED);
}
break;
case HTTP_CLIENT_CALLBACK_DISCONNECTED:
printf("Disconnected reason:%d\r\n", data->disconnected.reason);
/* If disconnect reason is equal to -ECONNRESET(-104),
* It means the server has closed the connection (timeout).
* This is normal operation.
*/
if (data->disconnected.reason == -EAGAIN) {
/* Server has not responded. Retry immediately. */
if (is_state_set(DOWNLOADING)) {
f_close(&file_object);
clear_state(DOWNLOADING);
}
if (is_state_set(GET_REQUESTED)) {
clear_state(GET_REQUESTED);
}
start_download();
}
break;
}
}
void* start_download_thread(void* param)
{
DownloadParam* downloadParam = NULL;
while (true) {
int semWaitRet = sem_wait(g_pSem);
if (semWaitRet < 0) {
// error
break;
}
pthread_mutex_lock(&g_downloadMutex);
if (g_downloadQueue.empty()) {
pthread_mutex_unlock(&g_downloadMutex);
if (g_exitThread) {
g_exitThread = false;
break;
} else {
continue;
}
} else {
downloadParam = g_downloadQueue.front();
g_downloadQueue.pop();
pthread_mutex_unlock(&g_downloadMutex);
}
if (!downloadParam) {
continue;
}
if (g_downEvent) {
g_downEvent->InvokeDownStart(downloadParam->taskId, 1);
}
start_download(downloadParam->url.c_str(), downloadParam->saving_path.c_str(), downloadParam->max_count, downloadParam->max_speed, downloadParam->taskId);
if (g_downEvent) {
g_downEvent->InvokeDownOver(downloadParam->taskId, downloadParam->url.c_str(), downloadParam->saving_path.c_str());
}
delete downloadParam;
downloadParam = NULL;
}
if (g_pSem != NULL) {
#ifdef __APPLE__
sem_unlink(kSemName);
sem_close(g_pSem);
#else
sem_destroy(g_pSem);
#endif
g_pSem = NULL;
}
return NULL;
}
// Start another HTTP download if possible.
static void start_next_download (void) {
dlqueue_t *q;
if (!cls.download.pending || curl_handles >= cl_http_max_connections->integer) {
return;
}
//not enough downloads running, queue some more!
FOR_EACH_DLQ (q) {
if (q->state == DL_RUNNING) {
if (q->type == DL_PAK)
break; // hack for pak file single downloading
} else if (q->state == DL_PENDING) {
dlhandle_t *dl = get_free_handle();
if (dl)
start_download (q, dl);
break;
}
}
}
/**
* \brief Callback to get the Wi-Fi status update.
*
* \param[in] u8MsgType type of Wi-Fi notification. Possible types are:
* - [M2M_WIFI_RESP_CURRENT_RSSI](@ref M2M_WIFI_RESP_CURRENT_RSSI)
* - [M2M_WIFI_RESP_CON_STATE_CHANGED](@ref M2M_WIFI_RESP_CON_STATE_CHANGED)
* - [M2M_WIFI_RESP_CONNTION_STATE](@ref M2M_WIFI_RESP_CONNTION_STATE)
* - [M2M_WIFI_RESP_SCAN_DONE](@ref M2M_WIFI_RESP_SCAN_DONE)
* - [M2M_WIFI_RESP_SCAN_RESULT](@ref M2M_WIFI_RESP_SCAN_RESULT)
* - [M2M_WIFI_REQ_WPS](@ref M2M_WIFI_REQ_WPS)
* - [M2M_WIFI_RESP_IP_CONFIGURED](@ref M2M_WIFI_RESP_IP_CONFIGURED)
* - [M2M_WIFI_RESP_IP_CONFLICT](@ref M2M_WIFI_RESP_IP_CONFLICT)
* - [M2M_WIFI_RESP_P2P](@ref M2M_WIFI_RESP_P2P)
* - [M2M_WIFI_RESP_AP](@ref M2M_WIFI_RESP_AP)
* - [M2M_WIFI_RESP_CLIENT_INFO](@ref M2M_WIFI_RESP_CLIENT_INFO)
* \param[in] pvMsg A pointer to a buffer containing the notification parameters
* (if any). It should be casted to the correct data type corresponding to the
* notification type. Existing types are:
* - tstrM2mWifiStateChanged
* - tstrM2MWPSInfo
* - tstrM2MP2pResp
* - tstrM2MAPResp
* - tstrM2mScanDone
* - tstrM2mWifiscanResult
*/
static void wifi_cb(uint8_t u8MsgType, void *pvMsg)
{
switch (u8MsgType) {
case M2M_WIFI_RESP_CON_STATE_CHANGED:
{
tstrM2mWifiStateChanged *pstrWifiState = (tstrM2mWifiStateChanged *)pvMsg;
if (pstrWifiState->u8CurrState == M2M_WIFI_CONNECTED) {
printf("Wi-Fi connected\r\n");
m2m_wifi_request_dhcp_client();
} else if (pstrWifiState->u8CurrState == M2M_WIFI_DISCONNECTED) {
printf("Wi-Fi disconnected\r\n");
clear_state(WIFI_CONNECTED);
if (is_state_set(DOWNLOADING)) {
f_close(&file_object);
clear_state(DOWNLOADING);
}
if (is_state_set(GET_REQUESTED)) {
clear_state(GET_REQUESTED);
}
m2m_wifi_connect((char *)MAIN_WLAN_SSID, sizeof(MAIN_WLAN_SSID),
MAIN_WLAN_AUTH, (char *)MAIN_WLAN_PSK, M2M_WIFI_CH_ALL);
}
break;
}
case M2M_WIFI_REQ_DHCP_CONF:
{
uint8_t *pu8IPAddress = (uint8_t *)pvMsg;
printf("Wi-Fi IP is %u.%u.%u.%u\r\n", pu8IPAddress[0], pu8IPAddress[1], pu8IPAddress[2], pu8IPAddress[3]);
add_state(WIFI_CONNECTED);
start_download();
break;
}
default:
break;
}
}
static INT_PTR CALLBACK installer_proc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam)
{
switch(msg) {
case WM_INITDIALOG:
ShowWindow(GetDlgItem(hwnd, ID_DWL_PROGRESS), SW_HIDE);
install_dialog = hwnd;
return TRUE;
case WM_NOTIFY:
switch (((NMHDR *)lParam)->code)
{
case NM_CLICK:
case NM_RETURN:
if (wParam == ID_DWL_STATUS)
run_winebrowser(((NMLINK*)lParam)->item.szUrl);
break;
}
break;
case WM_COMMAND:
switch(wParam) {
case IDCANCEL:
if(dwl_binding)
IBinding_Abort(dwl_binding);
EndDialog(hwnd, 0);
return FALSE;
case ID_DWL_INSTALL:
ShowWindow(GetDlgItem(hwnd, ID_DWL_PROGRESS), SW_SHOW);
EnableWindow(GetDlgItem(hwnd, ID_DWL_INSTALL), 0);
if(!start_download())
EndDialog(install_dialog, 0);
}
}
return FALSE;
}
/* Main server module common for all transports
* This module will do the initial transport setups.
* Then it receives the command from client in CDC format
* and transmits the response back to the client.
* In the case of socket, it receives the command from the client
* and sends the response directly to the client via TCP socket.
*
* In the case of serial & wifi , it receives the command from the driver
* and sends the response to the driver.
*/
int
remote_server_exec(int argc, char **argv, void *wl)
{
int err;
int transport_descriptor;
char *async_cmd_flag = NULL;
int skip;
int download_flag = 0;
#if defined(LINUX) || defined(vxworks) || defined(OLYMPIC_RWL)
char old_intf_name[IFNAMSIZ];
#endif
#ifdef WIN32
char shell_fname[MAX_SHELL_FILE_LENGTH];
DWORD dwlen;
#endif
#ifdef RWL_DONGLE
int uart_enable = 1;
/* To set dongle flag when dongle server starts */
if ((err = rwl_var_setbuf(wl, dongleset, &uart_enable,
sizeof(int))) < 0) {
DPRINT_INFO(OUTPUT, "Unable to send to wl driver,error=%d\n", err);
}
#endif
if (rwl_iovar_check (wl) < 0) {
DPRINT_ERR(ERR, "wl_server: RWL_WIFI/RWL_DONGLE not defined ");
DPRINT_ERR(ERR, "Or In-Dongle mode enabled\n");
exit(0);
}
/* Initialise for all the transports - socket, serial, and wifi
* In Socket transport, main socket handler will be returned.
*/
if ((transport_descriptor = rwl_transport_setup(argc, argv)) < 0)
return BCME_ERROR;
#ifdef RWL_WIFI
remote_wifi_ser_init_cmds(wl);
#endif
/* Create a directory /tmp/RWL for the shell response files */
if (rwl_create_dir() < 0)
return BCME_ERROR;
#ifdef RWLASD
/* DUT initialization function */
wfa_dut_init(&trafficBuf, &respBuf, &parmsVal, &xcCmdBuf, &toutvalp);
#endif
#if defined(LINUX) || defined(vxworks)
/* Copy old interface name to restore it */
store_old_interface(wl, old_intf_name);
#endif /* defined(LINUX) || defined(vxworks) */
while (1) {
uchar *buf_ptr = NULL;
#ifdef VISTA_SERVER
int index;
char *vista_buf[MAX_VISTA_ARGC];
#endif
#ifdef RWL_SERIAL
g_rwl_hndle = transport_descriptor;
#else
g_rwl_hndle = -1;
#endif
#ifdef RWL_DONGLE
if (set_ctrlc) {
uart_enable = 0;
if ((err = rwl_var_setbuf(wl, dongleset, &uart_enable,
sizeof(int))) < 0) {
DPRINT_INFO(OUTPUT, "Unable to send to wl driver,error=%d\n", err);
}
set_ctrlc = 0;
exit(0);
}
#endif /* RWL_DONGLE */
/* Receive the CDC header */
if ((remote_rx_header(wl, transport_descriptor)) == BCME_ERROR) {
DPRINT_DBG(OUTPUT, "\n Waiting for client to transmit command\n");
continue;
}
DPRINT_INFO(OUTPUT, "REC : cmd %d\t msg len %d msg flag %d\t msg status %d\n",
g_rem_ptr->msg.cmd, g_rem_ptr->msg.len,
g_rem_ptr->msg.flags, g_rem_ptr->msg.status);
#ifdef RWL_WIFI
/* send the response to remote if it is findserver cmd, this is specific to wifi */
if (g_rem_ptr->msg.flags & REMOTE_FINDSERVER_IOCTL) {
remote_wifi_response(wl);
continue;
}
#endif /* RWL_WIFI */
/*
* Allocate buffer only if there is a response message expected.
* Some commands such as up/down do not output anything.
*/
if (g_rem_ptr->msg.len) {
if ((buf_ptr = malloc(g_rem_ptr->msg.len)) == NULL) {
DPRINT_ERR(ERR, "malloc of %d bytes failed\n", g_rem_ptr->msg.len);
continue;
}
}
/* Receive the data */
if ((err = remote_rx_data(buf_ptr)) == BCME_ERROR) {
if (buf_ptr)
free(buf_ptr);
continue;
}
/* Process RWL negotiate commands */
if (g_rem_ptr->msg.flags & REMOTE_NEGOTIATE_CMD) {
if (g_rem_ptr->msg.cmd == NEGOTIATE_GET_OS) {
if (g_rem_ptr->msg.len >= sizeof(int)) {
#if defined(LINUX)
*(int*)buf_ptr = LINUX_OS;
#elif defined(VISTA_SERVER)
*(int*)buf_ptr = WINVISTA_OS;
#elif defined(WIN32)
*(int*)buf_ptr = WIN32_OS;
#elif defined(MACOSX)
*(int*)buf_ptr = MAC_OSX;
#else
*(int*)buf_ptr = UNKNOWN_OS;
#endif
g_rem_ptr->msg.len = sizeof(int);
DPRINT_INFO(OUTPUT, "RESP : os type %d\n", *(int*)buf_ptr);
if (remote_tx_response(wl, buf_ptr, 0) < 0)
DPRINT_ERR(ERR, "\nReturn results failed\n");
}
}
#ifdef RWL_SOCKET
close_sock_handle(g_rwl_hndle);
#endif /* RWL_SOCKET */
if (buf_ptr)
free(buf_ptr);
continue;
}
/* Process command */
if (g_rem_ptr->msg.flags & REMOTE_SHELL_CMD) {
/* Get the response length first and get the response buffer in case of
* synchronous shell commands and the buf_ptr will have the response file
* name. In case of asynchronous shell commands, buf_ptr
* will be get updated by the remote_shell_execute function.
*/
need_speedy_response = 1;
#ifndef WIN32
if (buf_ptr) {
async_cmd_flag = strstr((char*)buf_ptr, "%");
}
if ((err = remote_shell_execute((char*)buf_ptr, wl)) > 0) {
if (async_cmd_flag)
g_rem_ptr->msg.len = err;
}
/* Sync shell command: No need to send response from here */
else {
#ifdef RWL_SOCKET
/* Transmitted to client. Then close the handle &
* get the new handle for next transmission & reception.
*/
close_sock_handle(g_rwl_hndle);
#endif /* RWL_SOCKET */
continue;
}
#else
if ((err = remote_shell_execute((char*)buf_ptr, wl)) != SUCCESS) {
DPRINT_ERR(ERR, "Error in executing shell command\n");
if (buf_ptr)
free(buf_ptr);
#ifdef RWL_SOCKET
/* Transmitted to client. Then close the handle &
* get the new handle for next transmission & reception.
*/
close_sock_handle(g_rwl_hndle);
#endif /* RWL_SOCKET */
continue;
}
/* Get the response from the temporary file */
if ((err = remote_shell_get_resp(shell_fname, wl)) != SUCCESS) {
DPRINT_ERR(ERR, "Error in executing shell command\n");
}
if (buf_ptr)
free(buf_ptr);
#ifdef RWL_SOCKET
/* Transmitted to client. Then close the handle &
* get the new handle for next transmission & reception.
*/
close_sock_handle(g_rwl_hndle);
#endif /* RWL_SOCKET */
continue;
#endif /* WIN32 */
} /* REMOTE_SHELL_CMD */
#ifdef RWLASD
if (g_rem_ptr->msg.flags & REMOTE_ASD_CMD) {
if ((err = remote_asd_exec(buf_ptr, (int *)&g_rem_ptr->msg.len)) < 0) {
DPRINT_ERR(ERR, "Error in executing asd command\n");
}
} /* REMOTE_ASD_CMD */
#endif
/*
* added to take care of OID base problem for cross OS RWL cleint server
* In case of LX Server and WIN32 client OID base need to be removed
* In case of WIN32 server and LX client OID base need to be added
*/
if (!(g_rem_ptr->msg.flags & REMOTE_ASD_CMD)) {
#if defined(LINUX) || defined(vxworks)
if (g_rem_ptr->msg.cmd > MAX_IOVAR)
g_rem_ptr->msg.cmd -= WL_OID_BASE;
#endif
#if defined(WIN32)
if (g_rem_ptr->msg.cmd < MAX_IOVAR)
g_rem_ptr->msg.cmd += WL_OID_BASE;
#endif
}
#ifdef VISTA_SERVER
if (g_rem_ptr->msg.flags & REMOTE_VISTA_CMD) {
vista_buf[0] = strtok(buf_ptr, " \t\n");
for (index = 1; (vista_buf[index] = strtok(NULL, " \t\n")) != NULL;
index++);
if ((err = remote_vista_exec(wl, vista_buf)) < 0) {
DPRINT_ERR(ERR, "Error in executing vista command\n");
}
memcpy(buf_ptr, vista_buf[0], strlen(vista_buf[0]));
g_rem_ptr->msg.len = strlen(vista_buf[0]);
} /* REMOTE_VISTA_CMD */
#endif /* VISTA_SERVER */
#ifndef OLYMPIC_RWL
#if defined(LINUX) || defined(vxworks)
#ifndef RWL_DONGLE
if (g_rem_ptr->msg.flags & REMOTE_GET_IOCTL ||
g_rem_ptr->msg.flags & REMOTE_SET_IOCTL) {
if (strlen(g_rem_ptr->intf_name) != 0) {
#if defined(LINUX)
struct ifreq ifr;
/* validate the interface */
memset(&ifr, 0, sizeof(ifr));
if (g_rem_ptr->intf_name)
strncpy(ifr.ifr_name, g_rem_ptr->intf_name, IFNAMSIZ);
if (wl_check((void *)&ifr)) {
DPRINT_ERR(ERR, "%s: wl driver adapter not found\n",
g_rem_ptr->intf_name);
/* Signal end of command output */
g_rem_ptr->msg.len = 0;
remote_tx_response(wl, NULL, BCME_NODEVICE);
if (buf_ptr)
free(buf_ptr);
#ifdef RWL_SOCKET
close_sock_handle(g_rwl_hndle);
#endif
continue;
}
#endif /* LINUX */
#if defined(vxworks)
if (wl_check((void *)g_rem_ptr->intf_name)) {
DPRINT_ERR(ERR, "%s: wl driver adapter not found\n",
g_rem_ptr->intf_name);
/* Signal end of command output */
g_rem_ptr->msg.len = 0;
remote_tx_response(wl, NULL, BCME_NODEVICE);
if (buf_ptr)
free(buf_ptr);
#ifdef RWL_SOCKET
close_sock_handle(g_rwl_hndle);
#endif
continue;
}
#endif /* vxworks */
if (set_interface(wl, g_rem_ptr->intf_name) == BCME_OK)
DPRINT_DBG(OUTPUT, "\n %s Interface will be used \n",
(char *)wl);
}
}
#endif /* ifndef RWL_DONGLE */
#endif /* defined(LINUX) || defined(vxworks) */
#endif /* ifndef OLYMPIC_RWL */
if (g_rem_ptr->msg.flags & REMOTE_SET_IOCTL ||
g_rem_ptr->msg.flags & RDHD_SET_IOCTL) {
#ifdef WIN32
#if defined (RWL_DONGLE) || defined (RWL_WIFI)
/* For commands with msg length as zero initialize the buffer to null */
if (g_rem_ptr->msg.len == 0)
buf_ptr = NULL;
#endif
#else
if (g_rem_ptr->msg.len == 0)
buf_ptr = NULL;
#endif /* WIN32 */
#if defined(LINUX) || defined(TARGETOS_symbian) || defined(TARGETOS_nucleus) || defined(MACOSX) || defined(TARGET_wiced)
#ifdef OLYMPIC_RWL
set_interface(wl, old_intf_name);
#endif
#if defined( TARGET_wiced )
set_interface(wl, g_rem_ptr->intf_name);
#endif
if (g_rem_ptr->msg.flags & REMOTE_SET_IOCTL) {
if (g_rem_ptr->msg.cmd == WLC_SET_VAR && buf_ptr &&
!strncmp((const char *)buf_ptr,
"init", g_rem_ptr->msg.len)) {
DPRINT_INFO(OUTPUT, "REC : init command\n");
err = 0;
} else if (g_rem_ptr->msg.cmd == WLC_SET_VAR && buf_ptr &&
!strncmp((const char *)buf_ptr,
"download", g_rem_ptr->msg.len)) {
DPRINT_INFO(OUTPUT, "REC : download command\n");
download_flag = download_flag? 0: 1;
if (download_flag) {
DPRINT_INFO(OUTPUT, "download started\n");
start_download( );
} else {
DPRINT_INFO(OUTPUT, "download completed\n");
finish_download( );
}
err = 0;
} else if (g_rem_ptr->msg.cmd == WLC_SET_VAR && buf_ptr &&
!strncmp((const char *)buf_ptr,
"membytes", g_rem_ptr->msg.len)) {
DPRINT_INFO(OUTPUT, "REC : membytes command\n");
skip = strlen("membytes ");
uint32_t address = *((uint32_t*)(buf_ptr + skip));
skip += sizeof(uint32_t);
uint32_t len = *((uint32_t*)(buf_ptr + skip));
skip += sizeof(uint32_t);
if ( len != g_rem_ptr->msg.len - skip )
{
DPRINT_ERR(ERR, "Length does not match\n");
}
membytes_write( address, buf_ptr + skip, g_rem_ptr->msg.len - skip );
err = 0;
} else {
err = wl_ioctl(wl, g_rem_ptr->msg.cmd,
(void *)buf_ptr, g_rem_ptr->msg.len, TRUE);
DPRINT_INFO(OUTPUT, "SEND : cmd %d\t msg len %d\n",
g_rem_ptr->msg.cmd, g_rem_ptr->msg.len);
DPRINT_INFO(OUTPUT, "error code: %d\n", err);
}
}
#if defined(LINUX)
if (err == IOCTL_ERROR) {
DPRINT_ERR(ERR, "Error in executing wl_ioctl\n");
DPRINT_ERR(ERR, "Setting Default Interface1 \n");
set_interface(wl, old_intf_name);
}
if (g_rem_ptr->msg.flags & RDHD_SET_IOCTL) {
err = dhd_ioctl(wl, g_rem_ptr->msg.cmd,
(void *)buf_ptr, g_rem_ptr->msg.len, TRUE);
}
#endif /* LINUX */
#elif vxworks
if ((err = wl_ioctl_vx(wl, g_rem_ptr->msg.cmd, (void *)buf_ptr,
g_rem_ptr->msg.len)) != 0) {
DPRINT_ERR(ERR, "Error in executing wl_ioctl_vx\n");
}
if (err == IOCTL_ERROR) {
DPRINT_ERR(ERR, "Error in executing wl_ioctl\n");
DPRINT_ERR(ERR, "Setting Default Interface \n");
set_interface(wl, old_intf_name);
}
#elif WIN32
dwlen = g_rem_ptr->msg.len;
if (g_rem_ptr->msg.flags & RDHD_SET_IOCTL) {
g_rem_ptr->msg.cmd = g_rem_ptr->msg.cmd
- WL_OID_BASE + OID_DHD_IOCTLS;
}
err = (int)ir_setinformation(wl, g_rem_ptr->msg.cmd,
buf_ptr, &dwlen);
#endif /* LINUX TARGETOS_symbian MACOSX */
g_rem_ptr->msg.flags = REMOTE_SET_IOCTL;
} /* RDHD/REMOTE_SET_IOCTL */
if (g_rem_ptr->msg.flags & REMOTE_GET_IOCTL ||
g_rem_ptr->msg.flags & RDHD_GET_IOCTL) {
#if defined(LINUX) || defined(TARGETOS_symbian) || defined(TARGETOS_nucleus) || defined(MACOSX) || defined(TARGET_wiced)
if (g_rem_ptr->msg.cmd == WLC_GET_VAR && buf_ptr &&
strncmp((const char *)buf_ptr, "exit", g_rem_ptr->msg.len) == 0) {
/* exit command from remote client terminates server */
free(buf_ptr);
break;
}
#if defined( TARGET_wiced )
set_interface(wl, g_rem_ptr->intf_name);
#endif
if (g_rem_ptr->msg.flags & REMOTE_GET_IOCTL)
err = wl_ioctl(wl, g_rem_ptr->msg.cmd, (void *)buf_ptr,
g_rem_ptr->msg.len, FALSE);
#if defined (LINUX)
if (err == IOCTL_ERROR) {
DPRINT_ERR(ERR, "REMOTE_GET_IOCTL::Error in executing wl_ioctl\n");
DPRINT_ERR(ERR, "Setting Default Interface \n");
set_interface(wl, old_intf_name);
}
if (g_rem_ptr->msg.flags & RDHD_GET_IOCTL)
err = dhd_ioctl(wl, g_rem_ptr->msg.cmd, (void *)buf_ptr,
g_rem_ptr->msg.len, FALSE);
#endif /* LINUX */
#elif vxworks
if ((err = wl_ioctl_vx(wl, g_rem_ptr->msg.cmd, (void *)buf_ptr,
g_rem_ptr->msg.len)) != 0) {
DPRINT_ERR(ERR, "Error in executing wl_ioctl_vx\n");
}
if (err == IOCTL_ERROR) {
DPRINT_ERR(ERR, "Error in executing wl_ioctl_vx\n");
DPRINT_ERR(ERR, "Setting Default Interface \n");
set_interface(wl, old_intf_name);
}
#elif WIN32
if (g_rem_ptr->msg.cmd == (WL_OID_BASE + WLC_GET_VAR) &&
strncmp(buf_ptr, "exit", g_rem_ptr->msg.len) == 0) {
/* exit command from remote client terminates server */
if (buf_ptr) {
free(buf_ptr);
}
break;
}
dwlen = g_rem_ptr->msg.len;
if (g_rem_ptr->msg.flags & RDHD_GET_IOCTL) {
g_rem_ptr->msg.cmd = g_rem_ptr->msg.cmd -
WL_OID_BASE + OID_DHD_IOCTLS;
}
err = (int)ir_queryinformation(wl,
g_rem_ptr->msg.cmd, buf_ptr, &dwlen);
#endif /* defined(LINUX) || defined(TARGETOS_symbian) || defined(TARGETOS_nucleus) || defined(TARGET_wiced) */
g_rem_ptr->msg.flags = REMOTE_GET_IOCTL;
} /* REMOTE_GET_IOCTL */
DPRINT_INFO(OUTPUT, "RESP : cmd %d\t msg len %d\n",
g_rem_ptr->msg.cmd, g_rem_ptr->msg.len);
#if defined(LINUX) || defined(vxworks)
/* setting back default interface */
set_interface(wl, old_intf_name);
#endif /* defined(LINUX) || defined(vxworks) */
/* Transmit the response results */
if (remote_tx_response(wl, buf_ptr, err) < 0) {
DPRINT_ERR(ERR, "\nReturn results failed\n");
}
#ifdef RWL_SOCKET
if (g_rem_ptr->msg.flags != REMOTE_SHELL_CMD)
/* Transmitted to client. Then close the handle & get the new handle
* for next transmission & reception. In case of shell commands this
* should be closed in respective shellproc files.
*/
close_sock_handle(g_rwl_hndle);
#endif /* RWL_SOCKET */
if (buf_ptr) {
free(buf_ptr);
}
} /* end of while */
#if defined (RWL_SOCKET)
/* Close the main handle for socket */
close_sock_handle(transport_descriptor);
#elif defined(RWL_SERIAL)
/* Close the main handle for serial pipe */
rwl_close_pipe(remote_type, (void*)&transport_descriptor);
#endif
#ifdef RWLASD
wfa_dut_deinit();
#endif
return err;
}
// main(argc, argv)
// The main loop!
int main(int argc, char *argv[])
{
task_t *tracker_task, *listen_task, *t;
struct in_addr tracker_addr;
int tracker_port;
char *s;
const char *myalias;
struct passwd *pwent;
// Default tracker is read.cs.ucla.edu
osp2p_sscanf("131.179.80.139:11111", "%I:%d",
&tracker_addr, &tracker_port);
if ((pwent = getpwuid(getuid()))) {
myalias = (const char *) malloc(strlen(pwent->pw_name) + 20);
sprintf((char *) myalias, "%s%d", pwent->pw_name,
(int) time(NULL));
} else {
myalias = (const char *) malloc(40);
sprintf((char *) myalias, "osppeer%d", (int) getpid());
}
// Ignore broken-pipe signals: if a connection dies, server should not
signal(SIGPIPE, SIG_IGN);
// Process arguments
argprocess:
if (argc >= 3 && strcmp(argv[1], "-t") == 0
&& (osp2p_sscanf(argv[2], "%I:%d", &tracker_addr, &tracker_port) >= 0
|| osp2p_sscanf(argv[2], "%d", &tracker_port) >= 0
|| osp2p_sscanf(argv[2], "%I", &tracker_addr) >= 0)
&& tracker_port > 0 && tracker_port <= 65535) {
argc -= 2, argv += 2;
goto argprocess;
} else if (argc >= 2 && argv[1][0] == '-' && argv[1][1] == 't'
&& (osp2p_sscanf(argv[1], "-t%I:%d", &tracker_addr, &tracker_port) >= 0
|| osp2p_sscanf(argv[1], "-t%d", &tracker_port) >= 0
|| osp2p_sscanf(argv[1], "-t%I", &tracker_addr) >= 0)
&& tracker_port > 0 && tracker_port <= 65535) {
--argc, ++argv;
goto argprocess;
} else if (argc >= 3 && strcmp(argv[1], "-d") == 0) {
if (chdir(argv[2]) == -1)
die("chdir");
argc -= 2, argv += 2;
goto argprocess;
} else if (argc >= 2 && argv[1][0] == '-' && argv[1][1] == 'd') {
if (chdir(argv[1]+2) == -1)
die("chdir");
--argc, ++argv;
goto argprocess;
} else if (argc >= 3 && strcmp(argv[1], "-b") == 0
&& osp2p_sscanf(argv[2], "%d", &evil_mode) >= 0) {
argc -= 2, argv += 2;
goto argprocess;
} else if (argc >= 2 && argv[1][0] == '-' && argv[1][1] == 'b'
&& osp2p_sscanf(argv[1], "-b%d", &evil_mode) >= 0) {
--argc, ++argv;
goto argprocess;
} else if (argc >= 2 && strcmp(argv[1], "-b") == 0) {
evil_mode = 1;
--argc, ++argv;
goto argprocess;
} else if (argc >= 2 && (strcmp(argv[1], "--help") == 0
|| strcmp(argv[1], "-h") == 0)) {
printf("Usage: osppeer [-tADDR:PORT | -tPORT] [-dDIR] [-b]\n"
"Options: -tADDR:PORT Set tracker address and/or port.\n"
" -dDIR Upload and download files from directory DIR.\n"
" -b[MODE] Evil mode!!!!!!!!\n");
exit(0);
}
// Connect to the tracker and register our files.
tracker_task = start_tracker(tracker_addr, tracker_port);
listen_task = start_listen();
register_files(tracker_task, myalias);
// First, download files named on command line.
pid_t pid;
for (; argc > 1; argc--, argv++)
if ((t = start_download(tracker_task, argv[1]))){
pid = fork();
if(pid == 0){
task_download(t, tracker_task);
_exit(0);
}
else if(pid > 0){}
else if(pid == -1)
error("Fork failed for download\n");
}
int n_tasks = 0;
// Then accept connections from other peers and upload files to them!
while ((t = task_listen(listen_task))){
message("Beginning of while, n_tasks = %d\n", n_tasks);
if(n_tasks >= MAX_NUM_TASKS){
if(waitpid(-1, 0, 0) > 0){
message("Waiting for other proc, n_tasks = %d\n", n_tasks);
n_tasks--;
}
}
else if(n_tasks < MAX_NUM_TASKS){
pid = fork();
if(pid == 0){
//message("New Task STARTED for upload, n_tasks = %d\n", n_tasks);
task_upload(t);
//message("New Task ENDED for upload, n_tasks = %d\n", n_tasks);
_exit(0);
}
else if(pid > 0){
n_tasks++;
}
else if(pid == -1)
error("Fork failed for upload\n");
}
}
return 0;
}
