Rooba
/
boinc
mirror of https://github.com/BOINC/boinc.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
boinc/client/gui_rpc_server.C

402 lines
11 KiB
C
Raw Blame History

// The contents of this file are subject to the BOINC Public License
// Version 1.0 (the "License"); you may not use this file except in
// compliance with the License. You may obtain a copy of the License at
// http://boinc.berkeley.edu/license_1.0.txt
//
// Software distributed under the License is distributed on an "AS IS"
// basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See the
// License for the specific language governing rights and limitations
// under the License.
//
// The Original Code is the Berkeley Open Infrastructure for Network Computing.
//
// The Initial Developer of the Original Code is the SETI@home project.
// Portions created by the SETI@home project are Copyright (C) 2002
// University of California at Berkeley. All Rights Reserved.
//
// Contributor(s):
//
#ifdef _WIN32
#include "boinc_win.h"
#endif
#ifndef _WIN32
#include <stdio.h>
#include <unistd.h>
#include <sys/socket.h>
#include <sys/un.h>
#include <vector>
#include <string.h>
#include <netinet/in.h>
#include <netinet/tcp.h>
#endif
#include "util.h"
#include "error_numbers.h"
#include "parse.h"
#include "client_msgs.h"
#include "client_state.h"
using std::string;
using std::list;
using std::vector;
GUI_RPC_CONN::GUI_RPC_CONN(int s) {
sock = s;
}
GUI_RPC_CONN::~GUI_RPC_CONN() {
#ifdef _WIN32
closesocket(sock);
#else
close(sock);
#endif
}
static PROJECT* get_project(char* buf, MIOFILE& fout) {
string url;
if (!parse_str(buf, "<project_url>", url)) {
fout.printf("<error>Missing project URL</error>\n");
return 0;
}
PROJECT* p = gstate.lookup_project(url.c_str());
if (!p) {
fout.printf("<error>No such project</error>\n");
return 0 ;
}
return p;
}
static void handle_result_show_graphics(char* buf, MIOFILE& fout) {
string result_name;
PROJECT* p = get_project(buf, fout);
if (!p) return;
if (!parse_str(buf, "<result_name>", result_name)) {
fout.printf("<error>Missing result name</error>\n");
return;
}
RESULT* rp = gstate.lookup_result(p, result_name.c_str());
if (!rp) {
fout.printf("<error>No such result</error>\n");
return;
}
ACTIVE_TASK* atp = gstate.lookup_active_task_by_result(rp);
if (!atp) {
fout.printf("<error>Result not active</error>\n");
return;
}
atp->request_graphics_mode(MODE_WINDOW);
fout.printf("<success/>\n");
}
static void handle_project_reset(char* buf, MIOFILE& fout) {
PROJECT* p = get_project(buf, fout);
if (p) {
gstate.reset_project(p);
fout.printf("<success/>\n");
}
}
static void handle_project_attach(char* buf, MIOFILE& fout) {
string url, authenticator;
if (!parse_str(buf, "<url>", url)) {
fout.printf("<error>Missing URL</error>\n");
return;
}
if (!parse_str(buf, "<authenticator>", authenticator)) {
fout.printf("<error>Missing authenticator</error>\n");
return;
}
gstate.add_project(url.c_str(), authenticator.c_str());
fout.printf("<success/>\n");
}
static void handle_project_detach(char* buf, MIOFILE& fout) {
PROJECT* p = get_project(buf, fout);
if (p) {
gstate.detach_project(p);
fout.printf("<success/>\n");
}
}
static void handle_project_update(char* buf, MIOFILE& fout) {
PROJECT* p = get_project(buf, fout);
if (p) {
p->sched_rpc_pending = true;
p->min_rpc_time = 0;
fout.printf("<success/>\n");
}
}
static void handle_set_run_mode(char* buf, MIOFILE& fout) {
if (match_tag(buf, "<always>")) {
gstate.user_run_request = USER_RUN_REQUEST_ALWAYS;
} else if (match_tag(buf, "<never>")) {
gstate.user_run_request = USER_RUN_REQUEST_NEVER;
} else if (match_tag(buf, "<auto>")) {
gstate.user_run_request = USER_RUN_REQUEST_AUTO;
} else {
fout.printf("<error>Missing mode</error>\n");
return;
}
fout.printf("<success/>\n");
}
static void handle_run_benchmarks(char* buf, MIOFILE& fout) {
// TODO: suspend activities; make sure run at right priority
//
gstate.start_cpu_benchmarks();
fout.printf("<success/>\n");
}
static void handle_set_proxy_settings(char* buf, MIOFILE& fout) {
string socks_proxy_server_name,http_proxy_server_name;
int socks_proxy_server_port,http_proxy_server_port;
if (!parse_str(buf, "<socks_proxy_server_name>", socks_proxy_server_name)) {
fout.printf("<error>SOCKS proxy server name missing</error>\n");
return;
}
if (!parse_int(buf, "<socks_proxy_server_port>", socks_proxy_server_port)) {
fout.printf("<error>SOCKS proxy server port missing</error>\n");
return;
}
if (!parse_str(buf, "<http_proxy_server_name>", http_proxy_server_name)) {
fout.printf("<error>HTTP proxy server name missing</error>\n");
return;
}
if (!parse_int(buf, "<http_proxy_server_port>", http_proxy_server_port)) {
fout.printf("<error>HTTP proxy server port missing</error>\n");
return;
}
safe_strcpy(gstate.pi.socks_server_name, socks_proxy_server_name.c_str());
gstate.pi.socks_server_port = socks_proxy_server_port;
safe_strcpy(gstate.pi.http_server_name, http_proxy_server_name.c_str());
gstate.pi.http_server_port = http_proxy_server_port;
fout.printf("<success/>\n");
}
// params:
// <nmessages>x</nmessages>
// return at most this many messages
// <offset>n</offset>
// start at message n.
// if no offset is given, return last n messages
//
void handle_get_messages(char* buf, MIOFILE& fout) {
int nmessages=-1, seqno=-1, j;
parse_int(buf, "<nmessages>", nmessages);
parse_int(buf, "<seqno>", seqno);
if (nmessages < 0) {
fout.printf("<error>No nmessages given</error>\n");
return;
}
fout.printf("<msgs>\n");
list<MESSAGE_DESC*>::const_iterator iter;
for (iter=message_descs.begin(), j=0;
iter != message_descs.end() && j<nmessages;
iter++, j++
) {
MESSAGE_DESC* mdp = *iter;
if (mdp->seqno <= seqno) break;
fout.printf(
"<msg>\n"
" <pri>%d</pri>\n"
" <seqno>%d</seqno>\n"
" <body>\n%s\n</body>\n"
" <time>%d</time>\n",
mdp->priority,
mdp->seqno,
mdp->message.c_str(),
mdp->timestamp
);
if (mdp->project) {
fout.printf(
" <project>%s</project>\n",
mdp->project->get_project_name()
);
}
fout.printf("</msg>\n");
}
fout.printf("</msgs>\n");
}
int GUI_RPC_CONN::handle_rpc() {
char request_msg[1024];
int n;
MIOFILE mf;
MFILE m;
char* p;
mf.init_mfile(&m);
SCOPE_MSG_LOG scope_messages(log_messages, CLIENT_MSG_LOG::DEBUG_GUIRPC);
// read the request message in one read()
// so that the core client won't hang because
// of malformed request msgs
//
#ifdef _WIN32
n = recv(sock, request_msg, 1024, 0);
#else
n = read(sock, request_msg, 1024);
#endif
if (n <= 0) return -1;
request_msg[n] = 0;
scope_messages.printf("GUI RPC Command = '%s'\n", request_msg);
if (match_tag(request_msg, "<get_state")) {
gstate.write_state(mf);
} else if (match_tag(request_msg, "<result_show_graphics>")) {
handle_result_show_graphics(request_msg, mf);
} else if (match_tag(request_msg, "<project_reset>")) {
handle_project_reset(request_msg, mf);
} else if (match_tag(request_msg, "<project_attach>")) {
handle_project_attach(request_msg, mf);
} else if (match_tag(request_msg, "<project_detach>")) {
handle_project_detach(request_msg, mf);
} else if (match_tag(request_msg, "<project_update>")) {
handle_project_update(request_msg, mf);
} else if (match_tag(request_msg, "<set_run_mode>")) {
handle_set_run_mode(request_msg, mf);
} else if (match_tag(request_msg, "<run_benchmarks")) {
handle_run_benchmarks(request_msg, mf);
} else if (match_tag(request_msg, "<set_proxy_settings>")) {
handle_set_proxy_settings(request_msg, mf);
} else if (match_tag(request_msg, "<get_messages>")) {
handle_get_messages(request_msg, mf);
} else {
mf.printf("<unrecognized/>\n");
}
mf.printf("\003");
m.get_buf(p, n);
send(sock, p, n, 0);
if (p) free(p);
return 0;
}
int GUI_RPC_CONN_SET::insert(GUI_RPC_CONN* p) {
gui_rpcs.push_back(p);
return 0;
}
int GUI_RPC_CONN_SET::init() {
sockaddr_in addr;
int retval;
lsock = socket(AF_INET, SOCK_STREAM, 0);
if (lsock < 0) {
msg_printf(NULL, MSG_ERROR,
"GUI RPC failed to initialize socket (retval = '%d')\n",
lsock
);
return ERR_SOCKET;
}
addr.sin_family = AF_INET;
addr.sin_port = htons(GUI_RPC_PORT);
addr.sin_addr.s_addr = htonl(INADDR_ANY);
int one = 1;
setsockopt(lsock, SOL_SOCKET, SO_REUSEADDR, (char*)&one, 4);
retval = bind(lsock, (const sockaddr*)(&addr), sizeof(addr));
if (retval) {
msg_printf(NULL, MSG_ERROR,
"GUI RPC bind failed (retval = '%d')\n", retval
);
return ERR_BIND;
}
retval = listen(lsock, 999);
if (retval) {
msg_printf(NULL, MSG_ERROR,
"GUI RPC listen failed (retval = '%d')\n",
retval
);
return ERR_LISTEN;
}
return 0;
}
bool GUI_RPC_CONN_SET::poll() {
int n = 0;
if (lsock >= 0) {
unsigned int i;
fd_set read_fds, error_fds;
int sock, retval;
vector<GUI_RPC_CONN*>::iterator iter;
GUI_RPC_CONN* gr;
struct timeval tv;
FD_ZERO(&read_fds);
FD_ZERO(&error_fds);
FD_SET(lsock, &read_fds);
for (i=0; i<gui_rpcs.size(); i++) {
gr = gui_rpcs[i];
FD_SET(gr->sock, &read_fds);
FD_SET(gr->sock, &error_fds);
}
memset(&tv, 0, sizeof(tv));
n = select(FD_SETSIZE, &read_fds, 0, &error_fds, &tv);
if (FD_ISSET(lsock, &read_fds)) {
struct sockaddr_in addr;
#ifdef _WIN32
int addr_len = sizeof(addr);
sock = accept(lsock, (struct sockaddr*)&addr, &addr_len);
#else
socklen_t addr_len = sizeof(addr);
sock = accept(lsock, (struct sockaddr*)&addr, &addr_len);
#endif
int peer_ip = (int) ntohl(addr.sin_addr.s_addr);
if ((!gstate.allow_remote_gui_rpc) && (peer_ip != 0x7f000001)) {
msg_printf(
NULL, MSG_ERROR,
"GUI RPC request from non-local address 0x%x\n",
peer_ip
);
#ifdef _WIN32
closesocket(sock);
#else
close(sock);
#endif
} else {
GUI_RPC_CONN* gr = new GUI_RPC_CONN(sock);
insert(gr);
}
}
iter = gui_rpcs.begin();
while (iter != gui_rpcs.end()) {
gr = *iter;
if (FD_ISSET(gr->sock, &error_fds)) {
delete gr;
gui_rpcs.erase(iter);
} else {
iter++;
}
}
iter = gui_rpcs.begin();
while (iter != gui_rpcs.end()) {
gr = *iter;
if (FD_ISSET(gr->sock, &read_fds)) {
retval = gr->handle_rpc();
if (retval) {
delete gr;
gui_rpcs.erase(iter);
continue;
}
}
iter++;
}
}
return (n != 0);
}
Reference in New Issue View Git Blame Copy Permalink