boinc/clientgui/MainDocument.h

419 lines
14 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 _MAINDOCUMENT_H_
#define _MAINDOCUMENT_H_
#if defined(__GNUG__) && !defined(__APPLE__)
#pragma interface "MainDocument.cpp"
#endif
#include <vector>
#include "common_defs.h"
#include "gui_rpc_client.h"
#include "AsyncRPC.h"
typedef struct {
int slot;
std::string project_url;
std::string name;
#ifdef _WIN32
HANDLE pid;
#else
int pid;
#endif
} RUNNING_GFX_APP;
///
/// Bitmask values for GetCurrentViewPage()
/// Used by CMainDocument::RunPeriodicRPCs() and Mac Accessibility
///
#define VW_PROJ 1
#define VW_TASK 2
#define VW_XFER 4
#define VW_STAT 8
#define VW_DISK 16
#define VW_NOTIF 128
#define VW_SGUI 256
#define VW_SMSG 2048
#define VW_SNOTIF 4096
extern bool g_use_sandbox;
class CMainDocument;
class CBOINCClientManager;
class CNetworkConnection : public wxObject {
public:
CNetworkConnection(CMainDocument* pDocument);
~CNetworkConnection();
void Poll();
void FireReconnectEvent() { m_bConnectEvent = true; };
void ForceDisconnect() { m_bForceReconnect = false; m_bReconnectOnError = false; m_bConnectEvent = false; SetStateDisconnected(); };
void ForceReconnect() { m_bForceReconnect = true; SetStateDisconnected(); };
int FrameShutdownDetected();
int GetConnectedComputerName(wxString& strMachine);
int GetConnectedComputerVersion(wxString& strVersion);
int GetConnectingComputerName(wxString& strMachine);
bool IsComputerNameLocal(const wxString& strMachine);
int GetLocalPassword(wxString& strPassword);
int SetComputer(
const wxChar* szComputer, const int iPort, const wxChar* szPassword,
const bool bUseDefaultPassword
);
void SetStateError();
void SetStateErrorAuthentication();
void SetStateReconnecting();
void SetStateSuccess(wxString& strComputer, wxString& strComputerPassword);
void SetStateDisconnected();
bool IsConnectEventSignaled() { return m_bConnectEvent; };
bool IsConnected() { return m_bConnected; };
bool IsReconnecting() { return m_bReconnecting; };
private:
CMainDocument* m_pDocument;
bool m_bFrameShutdownDetected;
bool m_bConnectEvent;
bool m_bForceReconnect;
bool m_bReconnectOnError;
bool m_bConnected;
bool m_bReconnecting;
bool m_bUseDefaultPassword;
bool m_bUsedDefaultPassword;
int m_iReadGUIRPCAuthFailure;
bool m_bNewConnection;
wxString m_strNewComputerName;
wxString m_strNewComputerPassword;
wxString m_strConnectedComputerName;
wxString m_strConnectedComputerPassword;
wxString m_strConnectedComputerVersion;
int m_iPort;
};
class CMainDocument : public wxObject {
DECLARE_DYNAMIC_CLASS(CMainDocument)
public:
CMainDocument();
~CMainDocument();
//
// Global
//
private:
wxDateTime m_dtCachedCCStatusTimestamp;
bool m_bClientStartCheckCompleted;
public:
int OnInit();
int OnExit();
int OnPoll();
int OnRefreshState();
int CachedStateUpdate();
int ResetState();
int Connect(
const wxChar* szComputer,
const int iPort,
const wxChar* szComputerPassword = wxEmptyString,
const bool bDisconnect = FALSE,
const bool bUseDefaultPassword = FALSE
);
int Reconnect();
int CachedStateLock();
int CachedStateUnlock();
void ForceDisconnect();
int FrameShutdownDetected();
int CoreClientQuit();
int GetConnectedComputerName(wxString& strMachine);
int GetConnectedComputerVersion(wxString& strVersion);
int GetConnectingComputerName(wxString& strMachine);
bool IsComputerNameLocal(const wxString strMachine);
bool IsConnected();
bool IsReconnecting();
int GetCoreClientStatus(CC_STATUS&, bool bForce = false);
int SetActivityRunMode(int iMode, int iTimeout);
int SetGPURunMode(int iMode, int iTimeout);
int SetNetworkRunMode(int iMode, int iTimeout);
void RefreshRPCs(bool fullReset = false);
void RunPeriodicRPCs(int frameRefreshRate);
int ForceCacheUpdate(bool immediate = true);
int RunBenchmarks();
bool IsUserAuthorized();
CNetworkConnection* m_pNetworkConnection;
CBOINCClientManager* m_pClientManager;
AsyncRPC rpc;
RPC_CLIENT rpcClient;
PROJECTS async_projects_update_buf;
CC_STATE state;
CC_STATE async_state_buf;
int m_iGet_state_rpc_result;
CC_STATUS status;
CC_STATUS async_status_buf;
int m_iGet_status_rpc_result;
wxDateTime m_dtCachedStateTimestamp;
//
// Async RPC support
//
public:
int RequestRPC(ASYNC_RPC_REQUEST& request, bool hasPriority = false);
void OnRPCComplete(CRPCFinishedEvent& event);
ASYNC_RPC_REQUEST* GetCurrentRPCRequest() { return &current_rpc_request; }
bool WaitingForRPC() { return m_bWaitingForRPC; }
wxDialog* GetRPCWaitDialog() { return m_RPCWaitDlg; }
// void TestAsyncRPC(); // For testing Async RPCs
RPCThread* m_RPCThread;
bool m_bRPCThreadIsReady;
bool m_bShutDownRPCThread;
private:
void HandleCompletedRPC();
void KillRPCThread();
int CopyProjectsToStateBuffer(PROJECTS& p, CC_STATE& state);
ASYNC_RPC_REQUEST current_rpc_request;
AsyncRPCDlg* m_RPCWaitDlg;
std::vector<ASYNC_RPC_REQUEST> RPC_requests;
bool m_bWaitingForRPC;
bool m_bNeedRefresh;
bool m_bNeedTaskBarRefresh;
BOINC_Mutex* m_pRPC_Thread_Mutex;
BOINC_Condition* m_pRPC_Thread_Condition;
BOINC_Mutex* m_pRPC_Request_Mutex;
BOINC_Condition* m_pRPC_Request_Condition;
wxDateTime m_dtLasAsyncRPCDlgTime;
wxDateTime m_dtLastFrameViewRefreshRPCTime;
//
// Projects Tab
//
private:
int m_iGet_project_status1_rpc_result;
wxDateTime m_dtProjectsStatusTimestamp;
public:
int CachedProjectStatusUpdate(bool bForce = false);
PROJECT* project(unsigned int);
PROJECT* project(char* url);
float m_fProjectTotalResourceShare;
int GetProjectCount();
int ProjectNoMoreWork(int iIndex);
int ProjectAllowMoreWork(int iIndex);
int ProjectAttach(const wxString& strURL, const wxString& strAccountKey);
int ProjectDetach(int iIndex);
int ProjectUpdate(int iIndex);
int ProjectReset(int iIndex);
int ProjectSuspend(int iIndex);
int ProjectResume(int iIndex);
//
// Work Tab
//
private:
int CachedResultsStatusUpdate();
wxDateTime m_dtResultsTimestamp;
double m_fResultsRPCExecutionTime;
wxDateTime m_dtKillInactiveGfxTimestamp;
std::vector<RUNNING_GFX_APP> m_running_gfx_apps;
RUNNING_GFX_APP* GetRunningGraphicsApp(RESULT* result, int slot);
void KillAllRunningGraphicsApps();
void KillInactiveGraphicsApps();
#ifdef _WIN32
void KillGraphicsApp(HANDLE pid);
#else
void KillGraphicsApp(int tpid);
#endif
public:
RESULTS results;
RESULTS async_results_buf;
int m_iGet_results_rpc_result;
bool m_ActiveTasksOnly;
RESULT* result(unsigned int);
RESULT* result(const wxString& name, const wxString& project_url);
int GetWorkCount();
int WorkSuspend(char* url, char* name);
int WorkResume(char* url, char* name);
int WorkShowGraphics(RESULT* result);
int WorkShowVMConsole(RESULT* result);
int WorkAbort(char* url, char* name);
CC_STATE* GetState() { return &state; };
//
// Notices Tab
//
private:
wxDateTime m_dtNoticesTimeStamp;
int m_iNoticeSequenceNumber;
int m_iLastReadNoticeSequenceNumber;
double m_dLastReadNoticeArrivalTime;
bool m_bWaitingForGetNoticesRPC;
public:
NOTICES notices;
int m_iGet_notices_rpc_result;
NOTICE* notice(unsigned int);
int CachedNoticeUpdate();
int GetNoticeCount();
int GetUnreadNoticeCount();
void SaveUnreadNoticeInfo();
void RestoreUnreadNoticeInfo();
void UpdateUnreadNoticeState();
int ResetNoticeState();
bool LocalizeNoticeText(wxString& strMessage, bool bSanitize = false, bool bClean = false);
//
// Messages Tab
//
private:
public:
MESSAGES messages;
int m_iGet_messages_rpc_result;
MESSAGE* message(unsigned int);
int CachedMessageUpdate();
int GetMessageCount();
int ResetMessageState();
int m_iMessageSequenceNumber;
//
// Transfers Tab
//
private:
int CachedFileTransfersUpdate();
wxDateTime m_dtFileTransfersTimestamp;
public:
FILE_TRANSFERS ft;
FILE_TRANSFERS async_ft_buf;
int m_iGet_file_transfers_rpc_result;
FILE_TRANSFER* file_transfer(unsigned int);
FILE_TRANSFER* file_transfer(const wxString& fileName, const wxString& project_url);
int GetTransferCount();
int TransferRetryNow(int iIndex);
int TransferRetryNow(const wxString& fileName, const wxString& project_url);
int TransferAbort(int iIndex);
int TransferAbort(const wxString& fileName, const wxString& project_url);
//
// Disk Tab
//
private:
wxDateTime m_dtDiskUsageTimestamp;
public:
DISK_USAGE disk_usage;
DISK_USAGE async_disk_usage_buf;
int m_iGet_dsk_usage_rpc_result;
PROJECT* DiskUsageProject(unsigned int);
int CachedDiskUsageUpdate();
//
// Statistics Tab
//
private:
int CachedStatisticsStatusUpdate();
wxDateTime m_dtStatisticsStatusTimestamp;
public:
PROJECTS statistics_status;
PROJECTS async_statistics_status_buf;
PROJECT* statistic(unsigned int);
int m_iGet_statistics_rpc_result;
int GetStatisticsCount();
//
// Proxy Configuration
//
private:
public:
GR_PROXY_INFO proxy_info;
int GetProxyConfiguration();
int SetProxyConfiguration();
//
// Simple GUI Updates
//
int m_iGet_simple_gui2_rpc_result;
int m_iAcct_mgr_info_rpc_result;
int CachedSimpleGUIUpdate(bool bForce = false);
private:
wxDateTime m_dtCachedSimpleGUITimestamp;
wxDateTime m_dtCachedAcctMgrInfoTimestamp;
public:
ACCT_MGR_INFO ami;
ACCT_MGR_INFO async_ami_buf;
int GetSimpleProjectCount();
int GetSimpleGUIWorkCount();
};
extern wxString suspend_reason_wxstring(int reason);
extern wxString result_description(RESULT*, bool show_resources=true);
extern wxString process_client_message(const char*);
#ifdef SANDBOX
#define BOINC_MASTER_GROUP_NAME "boinc_master"
#endif
#endif