mirror of https://github.com/BOINC/boinc.git
151 lines
4.8 KiB
C++
151 lines
4.8 KiB
C++
// The contents of this file are subject to the Mozilla 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://www.mozilla.org/MPL/
|
|
//
|
|
// 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):
|
|
//
|
|
|
|
#ifndef _TASK_
|
|
#define _TASK_
|
|
|
|
#include "windows_cpp.h"
|
|
#ifdef _WIN32
|
|
#include <afxwin.h>
|
|
#endif
|
|
|
|
#include <stdio.h>
|
|
#include <vector>
|
|
#include "client_types.h"
|
|
#include "app_ipc.h"
|
|
|
|
class CLIENT_STATE;
|
|
typedef int PROCESS_ID;
|
|
|
|
// Possible states of a process in an ACTIVE_TASK
|
|
#define PROCESS_UNINITIALIZED 0
|
|
#define PROCESS_RUNNING 1
|
|
#define PROCESS_EXITED 2
|
|
#define PROCESS_WAS_SIGNALED 3
|
|
#define PROCESS_EXIT_UNKNOWN 4
|
|
#define PROCESS_ABORT_PENDING 5
|
|
// process exceeded limits; killed it, waiting to exit
|
|
#define PROCESS_ABORTED 6
|
|
// process has exited
|
|
#define PROCESS_COULDNT_START 7
|
|
|
|
|
|
// Represents a task in progress.
|
|
// The execution of a task may be divided into many "episodes"
|
|
// (if the host is turned off/on, e.g.)
|
|
// A task may checkpoint now and then.
|
|
// Each episode begins with the state of the last checkpoint.
|
|
//
|
|
class ACTIVE_TASK {
|
|
public:
|
|
#ifdef _WIN32
|
|
HANDLE pid_handle, thread_handle, quitRequestEvent, shm_handle;
|
|
#else
|
|
key_t shm_key;
|
|
#endif
|
|
RESULT* result;
|
|
WORKUNIT* wup;
|
|
APP_VERSION* app_version;
|
|
PROCESS_ID pid;
|
|
int slot; // which slot (determines directory)
|
|
int state;
|
|
int exit_status;
|
|
int signal;
|
|
double fraction_done;
|
|
// App's estimate of how much of the work unit is done.
|
|
// Passed from the application via an API call;
|
|
// will be zero if the app doesn't use this call
|
|
double starting_cpu_time;
|
|
// total CPU time at the start of current episode
|
|
double checkpoint_cpu_time;
|
|
// total CPU at the last checkpoint
|
|
double current_cpu_time;
|
|
// most recent total CPU time reported by app
|
|
int current_disk_usage(double&);
|
|
// disk used by output files and temp files of this task
|
|
char slot_dir[256]; // directory where process runs
|
|
double max_cpu_time; // abort if total CPU exceeds this
|
|
double max_disk_usage; // abort if disk usage (in+out+temp) exceeds this
|
|
|
|
APP_CLIENT_SHM app_client_shm; // core/app shared mem
|
|
time_t last_status_msg_time;
|
|
|
|
// info related to app's graphics mode (win, screensaver, etc.)
|
|
int graphics_requested_mode; // our last request to this app
|
|
int graphics_request_time; // when we sent it
|
|
int graphics_acked_mode; // most recent mode reported by app
|
|
int graphics_mode_before_ss; // mode before last screensaver request
|
|
void request_graphics_mode(int);
|
|
void check_graphics_mode_ack();
|
|
|
|
ACTIVE_TASK();
|
|
int init(RESULT*);
|
|
|
|
int start(bool first_time); // start the task running
|
|
int request_exit(); // Send a SIGQUIT signal or equivalent
|
|
int kill_task(); // send a SIGKILL signal or equivalent
|
|
int suspend(); // send a SIGSTOP signal or equivalent
|
|
int unsuspend(); // send a SIGCONT signal or equivalent
|
|
int abort(); // flag as abort pending and send kill signal
|
|
bool task_exited(); // return true if this task has exited
|
|
|
|
int get_cpu_time_via_shmem(time_t);
|
|
int get_cpu_time_via_os();
|
|
double est_time_to_completion();
|
|
bool read_stderr_file();
|
|
|
|
int write(FILE*);
|
|
int parse(FILE*, CLIENT_STATE*);
|
|
};
|
|
|
|
class ACTIVE_TASK_SET {
|
|
public:
|
|
vector<ACTIVE_TASK*> active_tasks;
|
|
int insert(ACTIVE_TASK*);
|
|
int remove(ACTIVE_TASK*);
|
|
int wait_for_exit(double);
|
|
ACTIVE_TASK* lookup_pid(int);
|
|
bool poll();
|
|
void suspend_all();
|
|
void unsuspend_all();
|
|
int restart_tasks();
|
|
void request_tasks_exit();
|
|
int exit_tasks();
|
|
void kill_tasks();
|
|
void get_cpu_times();
|
|
int check_max_disk_usage();
|
|
bool check_app_exited();
|
|
bool check_max_cpu_exceeded();
|
|
bool check_max_disk_exceeded();
|
|
int get_free_slot(int total_slots);
|
|
|
|
// screensaver-related functions
|
|
ACTIVE_TASK* get_graphics_capable_app();
|
|
ACTIVE_TASK* get_app_requested(int req_mode);
|
|
void save_app_modes();
|
|
void hide_apps();
|
|
void restore_apps();
|
|
void check_graphics_mode_ack();
|
|
|
|
int write(FILE*);
|
|
int parse(FILE*, CLIENT_STATE*);
|
|
};
|
|
|
|
#endif
|