boinc/client/scheduler_op.h

141 lines
4.2 KiB
C++

// This file is part of BOINC.
// http://boinc.berkeley.edu
// Copyright (C) 2008 University of California
//
// BOINC is free software; you can redistribute it and/or modify it
// under the terms of the GNU Lesser General Public License
// as published by the Free Software Foundation,
// either version 3 of the License, or (at your option) any later version.
//
// BOINC is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
// See the GNU Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with BOINC. If not, see <http://www.gnu.org/licenses/>.
#ifndef _SCHEDULER_OP_
#define _SCHEDULER_OP_
#include <vector>
#include "client_types.h"
#include "auto_update.h"
#include "http_curl.h"
#include "prefs.h"
#define SCHEDULER_OP_STATE_IDLE 0
// invariant: in this state, our HTTP_OP is not in the HTTP_OP_SET
#define SCHEDULER_OP_STATE_GET_MASTER 1
#define SCHEDULER_OP_STATE_RPC 2
// defaults related to scheduler RPC policy
// See client_state.h for definitions
#define MASTER_FETCH_PERIOD 10
// fetch and parse master URL if nrpc_failures is a multiple of this
#define RETRY_CAP 10
// cap on nrpc_failures
#define MASTER_FETCH_RETRY_CAP 3
// after this many master-fetch failures,
// move into a state in which we retry master fetch
// at the frequency below
#define MASTER_FETCH_INTERVAL (86400) // 1 day
// See above
// constants used to bound RPC backoff
#define SCHED_RETRY_DELAY_MIN 60 // 1 minute
#define SCHED_RETRY_DELAY_MAX (60*60*4) // 4 hours
/// SCHEDULER_OP encapsulates the mechanism for
/// 1) fetching master files
/// 2) communicating with scheduling servers
/// Only one such operation can be in progress at once.
class SCHEDULER_OP {
private:
int scheduler_op_retval;
HTTP_OP http_op;
HTTP_OP_SET* http_ops;
char scheduler_url[256];
/// index within project's URL list
int url_index;
public:
/// project we're currently contacting
PROJECT* cur_proj;
int state;
int reason;
/// used to randomize order
double url_random;
public:
SCHEDULER_OP(HTTP_OP_SET*);
bool poll();
int init_op_project(PROJECT*, int);
int init_master_fetch(PROJECT*);
bool check_master_fetch_start();
void backoff(PROJECT* p, const char *error_msg);
/// if we're doing an op to this project, abort it
void abort(PROJECT*);
private:
bool update_urls(PROJECT*, std::vector<std::string> &urls);
int start_op(PROJECT*);
int start_rpc(PROJECT*);
void rpc_failed(const char*);
int parse_master_file(PROJECT*, std::vector<std::string>&);
};
struct USER_MESSAGE {
std::string message;
std::string priority;
USER_MESSAGE(char*, char*);
};
struct SCHEDULER_REPLY {
int hostid;
double request_delay;
double next_rpc_delay;
std::vector<USER_MESSAGE> messages;
/// not including <global_preferences> tags;
/// may include <venue> elements
char* global_prefs_xml;
/// not including <project_preferences> tags
/// may include <venue> elements
char* project_prefs_xml;
char master_url[256];
char host_venue[256];
unsigned int user_create_time;
std::vector<APP> apps;
std::vector<FILE_INFO> file_infos;
std::vector<std::string> file_deletes;
std::vector<APP_VERSION> app_versions;
std::vector<WORKUNIT> workunits;
std::vector<RESULT> results;
std::vector<RESULT> result_acks;
std::vector<RESULT> result_abort;
std::vector<RESULT> result_abort_if_not_started;
char* code_sign_key;
char* code_sign_key_signature;
bool message_ack;
bool project_is_down;
bool send_file_list;
int send_time_stats_log;
int send_job_log;
int scheduler_version;
AUTO_UPDATE auto_update;
double cpu_backoff;
double cuda_backoff;
double ati_backoff;
std::vector<RSS_FEED> sr_feeds;
bool got_rss_feeds;
// whether scheduler reply included <rss_feeds>
SCHEDULER_REPLY();
~SCHEDULER_REPLY();
int parse(FILE*, PROJECT*);
};
#endif