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_client.C

528 lines
17 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 <sys/types.h>
#include <sys/socket.h>
#include <sys/un.h>
#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
#include <netinet/in.h>
#include <netinet/tcp.h>
#include <arpa/inet.h>
#endif
#include "parse.h"
#include "error_numbers.h"
#include "miofile.h"
#include "gui_rpc_client.h"
using std::string;
using std::vector;
int RPC_CLIENT::init(char* path) {
int retval;
sockaddr_in addr;
addr.sin_family = AF_INET;
addr.sin_port = htons(GUI_RPC_PORT);
#ifdef _WIN32
addr.sin_addr.s_addr = inet_addr("127.0.0.1");
#else
addr.sin_addr.s_addr = htonl(INADDR_ANY);
#endif
sock = socket(AF_INET, SOCK_STREAM, 0);
if (sock <= 0) {
perror("socket");
exit(1);
}
retval = connect(sock, (const sockaddr*)(&addr), sizeof(addr));
if (retval) {
#ifdef _WIN32
printf( "Windows Socket Error '%d'\n", WSAGetLastError() );
#endif
perror("connect");
exit(1);
}
return 0;
}
RPC_CLIENT::~RPC_CLIENT() {
}
int RPC_CLIENT::send_request(char* p) {
send(sock, p, strlen(p), 0);
return 0;
}
// get reply from server. Caller must free buf
//
int RPC_CLIENT::get_reply(char*& mbuf) {
char buf[1025];
MFILE mf;
int n;
while (1) {
n = recv(sock, buf, 1024, 0);
if (n <= 0) break;
buf[n]=0;
mf.puts(buf);
if (strchr(buf, '\003')) break;
}
mf.get_buf(mbuf, n);
return 0;
}
int RPC_CLIENT::get_state() {
char buf[256];
PROJECT* project;
char* mbuf;
send_request("<get_state/>\n");
get_reply(mbuf);
MIOFILE fin;
fin.init_buf(mbuf);
while (fin.fgets(buf, 256)) {
if (match_tag(buf, "</client_state>")) break;
else if (match_tag(buf, "<project>")) {
project = new PROJECT;
project->parse(fin);
projects.push_back(project);
continue;
}
else if (match_tag(buf, "<app>")) {
APP* app = new APP;
app->parse(fin);
app->project = project;
apps.push_back(app);
continue;
}
else if (match_tag(buf, "<app_version>")) {
APP_VERSION* app_version = new APP_VERSION;
app_version->parse(fin);
app_version->project = project;
app_version->app = lookup_app(app_version->app_name);
app_versions.push_back(app_version);
continue;
}
else if (match_tag(buf, "<workunit>")) {
WORKUNIT* wu = new WORKUNIT;
wu->parse(fin);
wu->project = project;
wu->app = lookup_app(wu->app_name);
wu->avp = lookup_app_version(wu->app_name, wu->version_num);
wus.push_back(wu);
continue;
}
else if (match_tag(buf, "<result>")) {
RESULT* result = new RESULT;
result->parse(fin);
result->project = project;
result->wup = lookup_wu(result->wu_name);
result->app = result->wup->app;
results.push_back(result);
continue;
}
else if (match_tag(buf, "<file_info>")) {
FILE_INFO* fip = new FILE_INFO;
fip->parse(fin);
fip->project = project;
file_infos.push_back(fip);
continue;
}
else if (match_tag(buf, "<active_task>")) {
ACTIVE_TASK* atp = new ACTIVE_TASK;
atp->parse(fin);
atp->result = lookup_result(atp->result_name);
active_tasks.push_back(atp);
continue;
}
}
if (mbuf) free(mbuf);
return 0;
}
int RPC_CLIENT::result_show_graphics(RESULT& result) {
return 0;
}
int RPC_CLIENT::project_reset(PROJECT& project) {
return 0;
}
int RPC_CLIENT::project_attach(char* url, char* auth) {
return 0;
}
int RPC_CLIENT::project_detach(PROJECT&) {
return 0;
}
int RPC_CLIENT::project_update(PROJECT&) {
return 0;
}
int RPC_CLIENT::set_run_mode(int mode) {
return 0;
}
int RPC_CLIENT::run_benchmarks() {
return 0;
}
int RPC_CLIENT::set_proxy_settings(PROXY_INFO& pi) {
return 0;
}
int RPC_CLIENT::get_messages(
int nmessages, int seqno, vector<MESSAGE_DESC>& msgs
) {
char buf[256];
char* mbuf;
sprintf(buf,
"<get_messages>\n"
" <nmessages>%d</nmessages>\n"
" <seqno>%d</seqno>\n"
"</get_messages>\n",
nmessages, seqno
);
send_request(buf);
get_reply(mbuf);
MIOFILE fin;
fin.init_buf(mbuf);
while (fin.fgets(buf, 256)) {
puts(buf);
if (match_tag(buf, "<msgs>")) continue;
if (match_tag(buf, "</msgs>")) break;
if (match_tag(buf, "<msg>")) {
MESSAGE_DESC md;
while (fin.fgets(buf, 256)) {
puts(buf);
if (match_tag(buf, "</msg>")) break;
if (parse_str(buf, "<project>", md.project)) continue;
if (match_tag(buf, "<body>" )) {
copy_element_contents(fin, "</body>", md.body);
continue;
}
if (parse_int(buf, "<pri>", md.priority)) continue;
if (parse_int(buf, "<time>", md.timestamp)) continue;
}
msgs.push_back(md);
}
}
if (mbuf) free(mbuf);
return 0;
}
void RPC_CLIENT::print() {
unsigned int i;
printf("======== Projects ========\n");
for (i=0; i<projects.size(); i++) {
printf("%d) -----------\n", i+1);
projects[i]->print();
}
printf("\n======== Applications ========\n");
for (i=0; i<apps.size(); i++) {
printf("%d) -----------\n", i+1);
apps[i]->print();
}
printf("\n======== Application versions ========\n");
for (i=0; i<app_versions.size(); i++) {
printf("%d) -----------\n", i+1);
app_versions[i]->print();
}
printf("\n======== Files ========\n");
for (i=0; i<file_infos.size(); i++) {
printf("%d) -----------\n", i+1);
file_infos[i]->print();
}
printf("\n======== Workunits ========\n");
for (i=0; i<wus.size(); i++) {
printf("%d) -----------\n", i+1);
wus[i]->print();
}
printf("\n======== Results ========\n");
for (i=0; i<results.size(); i++) {
printf("%d) -----------\n", i+1);
results[i]->print();
}
printf("\n======== Active tasks ========\n");
for (i=0; i<active_tasks.size(); i++) {
printf("%d) -----------\n", i+1);
active_tasks[i]->print();
}
}
int FILE_INFO::parse(MIOFILE& in) {
char buf[256];
while (in.fgets(buf, 256)) {
if (match_tag(buf, "</file_info>")) return 0;
else if (parse_str(buf, "<name>", name)) continue;
else if (match_tag(buf, "<generated_locally/>")) {
generated_locally = true;
continue;
}
else if (match_tag(buf, "<uploaded/>")) {
uploaded = true;
continue;
}
else if (match_tag(buf, "<upload_when_present/>")) {
upload_when_present = true;
continue;
}
else if (match_tag(buf, "<sticky/>")) {
sticky = true;
continue;
}
else if (match_tag(buf, "<persistent_file_xfer>")) {
pers_xfer_active = true;
continue;
}
else if (match_tag(buf, "<file_xfer>")) {
xfer_active = true;
continue;
}
else if (parse_int(buf, "<num_retries>", num_retries)) continue;
else if (parse_double(buf, "<bytes_xferred>", bytes_xferred)) continue;
else if (parse_double(buf, "<file_offset>", file_offset)) continue;
else if (parse_double(buf, "<xfer_speed>", xfer_speed)) continue;
else if (parse_str(buf, "<hostname>", hostname)) continue;
}
return ERR_XML_PARSE;
}
void FILE_INFO::print() {
printf(" name: %s\n", name.c_str());
printf(" generated locally: %s\n", generated_locally?"yes":"no");
printf(" uploaded: %s\n", uploaded?"yes":"no");
printf(" upload when present: %s\n", upload_when_present?"yes":"no");
printf(" sticky: %s\n", sticky?"yes":"no");
printf(" generated locally: %s\n", generated_locally?"yes":"no");
}
int PROJECT::parse(MIOFILE& in) {
char buf[256];
while (in.fgets(buf, 256)) {
if (match_tag(buf, "</project>")) return 0;
else if (parse_str(buf, "<master_url", master_url)) continue;
else if (parse_double(buf, "<resource_share", resource_share)) continue;
else if (parse_str(buf, "<project_name>", project_name)) continue;
else if (parse_str(buf, "<user_name>", user_name)) continue;
else if (parse_str(buf, "<team_name>", team_name)) continue;
else if (parse_double(buf, "<user_total_credit>", user_total_credit)) continue;
else if (parse_double(buf, "<user_expavg_credit>", user_expavg_credit)) continue;
else if (parse_double(buf, "<host_total_credit>", host_total_credit)) continue;
else if (parse_double(buf, "<host_expavg_credit>", host_expavg_credit)) continue;
else if (parse_int(buf, "<nrpc_failures>", nrpc_failures)) continue;
else if (parse_int(buf, "<master_fetch_failures>", master_fetch_failures)) continue;
else if (parse_int(buf, "<min_rpc_time>", min_rpc_time)) continue;
else if (match_tag(buf, "<master_url_fetch_pending>")) {
master_url_fetch_pending = true;
continue;
}
else if (match_tag(buf, "<sched_rpc_pending>")) {
sched_rpc_pending = true;
continue;
}
else if (match_tag(buf, "<tentative>")) {
tentative = true;
continue;
}
}
return ERR_XML_PARSE;
}
void PROJECT::print() {
printf(" name: %s\n", project_name.c_str());
printf(" master URL: %s\n", master_url.c_str());
printf(" user_name: %s\n", user_name.c_str());
printf(" team_name: %s\n", team_name.c_str());
printf(" resource share: %f\n", resource_share);
printf(" user_total_credit: %f\n", user_total_credit);
printf(" user_expavg_credit: %f\n", user_expavg_credit);
printf(" host_total_credit: %f\n", host_total_credit);
printf(" host_expavg_credit: %f\n", host_expavg_credit);
printf(" nrpc_failures: %d\n", nrpc_failures);
printf(" master_fetch_failures: %d\n", master_fetch_failures);
printf(" master fetch pending: %s\n", master_url_fetch_pending?"yes":"no");
printf(" scheduler RPC pending: %s\n", sched_rpc_pending?"yes":"no");
printf(" tentative: %s\n", tentative?"yes":"no");
}
int APP::parse(MIOFILE& in) {
char buf[256];
while (in.fgets(buf, 256)) {
if (match_tag(buf, "</app>")) return 0;
else if (parse_str(buf, "<name>", name)) continue;
}
return ERR_XML_PARSE;
}
void APP::print() {
printf(" name: %s\n", name.c_str());
printf(" Project: %s\n", project->project_name.c_str());
}
int APP_VERSION::parse(MIOFILE& in) {
char buf[256];
while (in.fgets(buf, 256)) {
if (match_tag(buf, "</app_version>")) return 0;
else if (parse_str(buf, "<app_name>", app_name)) continue;
else if (parse_int(buf, "<version_num>", version_num)) continue;
}
return ERR_XML_PARSE;
}
void APP_VERSION::print() {
printf(" application: %s\n", app->name.c_str());
printf(" version: %.2f\n", version_num/100.0);
printf(" project: %s\n", project->project_name.c_str());
}
int WORKUNIT::parse(MIOFILE& in) {
char buf[256];
while (in.fgets(buf, 256)) {
if (match_tag(buf, "</workunit>")) return 0;
else if (parse_str(buf, "<name>", name)) continue;
else if (parse_str(buf, "<app_name>", app_name)) continue;
else if (parse_int(buf, "<version_num>", version_num)) continue;
else if (parse_double(buf, "<rsc_fpops_est>", rsc_fpops_est)) continue;
else if (parse_double(buf, "<rsc_fpops_bound>", rsc_fpops_bound)) continue;
else if (parse_double(buf, "<rsc_memory_bound>", rsc_memory_bound)) continue;
else if (parse_double(buf, "<rsc_disk_bound>", rsc_disk_bound)) continue;
}
return ERR_XML_PARSE;
}
void WORKUNIT::print() {
printf(" name: %s\n", name.c_str());
printf(" FP estimate: %f\n", rsc_fpops_est);
printf(" FP bound: %f\n", rsc_fpops_bound);
printf(" memory bound: %f\n", rsc_memory_bound);
printf(" disk bound: %f\n", rsc_disk_bound);
}
int RESULT::parse(MIOFILE& in) {
char buf[256];
while (in.fgets(buf, 256)) {
if (match_tag(buf, "</result>")) return 0;
else if (parse_str(buf, "<name>", name)) continue;
else if (parse_str(buf, "<wu_name>", wu_name)) continue;
else if (parse_int(buf, "<report_deadline>", report_deadline)) continue;
else if (match_tag(buf, "<ready_to_report>")) {
ready_to_report = true;
continue;
}
else if (match_tag(buf, "<got_server_ack>")) {
got_server_ack = true;
continue;
}
else if (parse_double(buf, "<final_cpu_time>", final_cpu_time)) continue;
else if (parse_int(buf, "<state>", state)) continue;
else if (parse_int(buf, "<exit_status>", exit_status)) continue;
else if (parse_int(buf, "<signal>", signal)) continue;
else if (parse_int(buf, "<active_task_state>", active_task_state)) continue;
else if (match_tag(buf, "<stderr_out>")) {
copy_element_contents(in, "</stderr_out>", stderr_out);
}
}
return ERR_XML_PARSE;
}
void RESULT::print() {
printf(" name: %s\n", name.c_str());
printf(" WU name: %s\n", wup->name.c_str());
printf(" ready to report: %s\n", ready_to_report?"yes":"no");
printf(" got server ack: %s\n", got_server_ack?"yes":"no");
printf(" final CPU time: %f\n", final_cpu_time);
printf(" state: %d\n", state);
printf(" exit_status: %d\n", exit_status);
printf(" signal: %d\n", signal);
printf(" active_task_state: %d\n", active_task_state);
printf(" stderr_out: %s\n", stderr_out.c_str());
}
int ACTIVE_TASK::parse(MIOFILE& in) {
char buf[256];
while (in.fgets(buf, 256)) {
if (match_tag(buf, "</active_task>")) return 0;
else if (parse_str(buf, "<result_name>", result_name)) continue;
else if (parse_int(buf, "<app_version_num>", app_version_num)) continue;
else if (parse_double(buf, "<checkpoint_cpu_time>", checkpoint_cpu_time)) continue;
else if (parse_double(buf, "<current_cpu_time>", current_cpu_time)) continue;
else if (parse_double(buf, "<fraction_done>", fraction_done)) continue;
}
return ERR_XML_PARSE;
}
void ACTIVE_TASK::print() {
printf(" result name: %s\n", result_name.c_str());
printf(" app version num: %d\n", app_version_num);
printf(" checkpoint CPU time: %f\n", checkpoint_cpu_time);
printf(" current CPU time: %f\n", current_cpu_time);
printf(" fraction done: %f\n", fraction_done);
}
APP* RPC_CLIENT::lookup_app(string& str) {
unsigned int i;
for (i=0; i<apps.size(); i++) {
if (apps[i]->name == str) return apps[i];
}
printf("CAN'T FIND APP %s\n", str.c_str());
return 0;
}
WORKUNIT* RPC_CLIENT::lookup_wu(string& str) {
unsigned int i;
for (i=0; i<wus.size(); i++) {
if (wus[i]->name == str) return wus[i];
}
printf("CAN'T FIND WU %s\n", str.c_str());
return 0;
}
RESULT* RPC_CLIENT::lookup_result(string& str) {
unsigned int i;
for (i=0; i<results.size(); i++) {
if (results[i]->name == str) return results[i];
}
printf("CAN'T FIND RESULT %s\n", str.c_str());
return 0;
}
APP_VERSION* RPC_CLIENT::lookup_app_version(string& str, int version_num) {
unsigned int i;
for (i=0; i<app_versions.size(); i++) {
if (app_versions[i]->app_name == str && app_versions[i]->version_num == version_num) return app_versions[i];
}
return 0;
}
Reference in New Issue View Git Blame Copy Permalink