2003-06-11 23:36:48 +00:00
|
|
|
// The contents of this file are subject to the BOINC Public License
|
2002-04-30 22:22:54 +00:00
|
|
|
// 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
|
2003-06-11 23:36:48 +00:00
|
|
|
// http://boinc.berkeley.edu/license_1.0.txt
|
2003-07-02 02:02:18 +00:00
|
|
|
//
|
2002-04-30 22:22:54 +00:00
|
|
|
// 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
|
2003-07-02 02:02:18 +00:00
|
|
|
// under the License.
|
|
|
|
//
|
|
|
|
// The Original Code is the Berkeley Open Infrastructure for Network Computing.
|
|
|
|
//
|
2002-04-30 22:22:54 +00:00
|
|
|
// The Initial Developer of the Original Code is the SETI@home project.
|
2003-07-02 20:57:59 +00:00
|
|
|
// Portions created by the SETI@home project are Copyright (C) 2002
|
2003-07-02 02:02:18 +00:00
|
|
|
// University of California at Berkeley. All Rights Reserved.
|
|
|
|
//
|
2002-04-30 22:22:54 +00:00
|
|
|
// Contributor(s):
|
|
|
|
//
|
|
|
|
|
|
|
|
// functions relating to file transfer
|
|
|
|
//
|
|
|
|
|
2002-06-06 18:42:01 +00:00
|
|
|
#include "windows_cpp.h"
|
|
|
|
|
|
|
|
#ifdef _WIN32
|
|
|
|
#include <io.h>
|
|
|
|
#endif
|
|
|
|
|
2003-07-23 22:24:28 +00:00
|
|
|
#include <cassert>
|
2002-04-30 22:22:54 +00:00
|
|
|
#include <sys/stat.h>
|
2002-06-06 18:42:01 +00:00
|
|
|
#include <sys/types.h>
|
2002-04-30 22:22:54 +00:00
|
|
|
|
|
|
|
#include "md5_file.h"
|
2002-07-07 20:39:24 +00:00
|
|
|
#include "crypt.h"
|
|
|
|
|
2002-04-30 22:22:54 +00:00
|
|
|
#include "file_xfer.h"
|
|
|
|
#include "file_names.h"
|
2002-06-01 20:26:21 +00:00
|
|
|
#include "client_types.h"
|
2002-04-30 22:22:54 +00:00
|
|
|
#include "client_state.h"
|
2003-05-09 22:44:35 +00:00
|
|
|
#include "filesys.h"
|
2002-07-11 01:09:53 +00:00
|
|
|
#include "error_numbers.h"
|
2002-04-30 22:22:54 +00:00
|
|
|
|
2003-03-02 19:24:09 +00:00
|
|
|
// Decide whether to consider starting a new file transfer
|
|
|
|
// TODO: limit the number of file xfers in some way
|
2002-08-30 22:11:36 +00:00
|
|
|
//
|
|
|
|
bool CLIENT_STATE::start_new_file_xfer() {
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2003-05-09 22:44:35 +00:00
|
|
|
void CLIENT_STATE::trunc_stderr_stdout() {
|
|
|
|
double f_size;
|
|
|
|
|
2003-05-14 18:18:33 +00:00
|
|
|
fflush(stdout);
|
|
|
|
fflush(stderr);
|
|
|
|
|
2003-05-09 22:44:35 +00:00
|
|
|
// If the stderr.txt or stdout.txt files are too big, reset them
|
|
|
|
// TODO: should we tell the user we're resetting these?
|
|
|
|
file_size(STDERR_FILE_NAME, f_size);
|
|
|
|
if (f_size > MAX_STDERR_FILE_SIZE) {
|
|
|
|
freopen(STDERR_FILE_NAME, "w", stderr);
|
|
|
|
}
|
|
|
|
file_size(STDOUT_FILE_NAME, f_size);
|
|
|
|
if (f_size > MAX_STDOUT_FILE_SIZE) {
|
|
|
|
freopen(STDOUT_FILE_NAME, "w", stderr);
|
|
|
|
}
|
|
|
|
}
|
2002-08-30 22:11:36 +00:00
|
|
|
|
2003-03-02 19:24:09 +00:00
|
|
|
// Make a directory for each of the projects in the client state
|
2002-08-22 21:29:58 +00:00
|
|
|
//
|
|
|
|
int CLIENT_STATE::make_project_dirs() {
|
|
|
|
unsigned int i;
|
2003-03-08 23:48:05 +00:00
|
|
|
int retval;
|
2002-08-22 21:29:58 +00:00
|
|
|
for (i=0; i<projects.size(); i++) {
|
2003-03-08 23:48:05 +00:00
|
|
|
retval = make_project_dir(*projects[i]);
|
|
|
|
if (retval) return retval;
|
2002-08-22 21:29:58 +00:00
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2002-07-15 23:21:20 +00:00
|
|
|
// Verify the validity of a downloaded file, through MD5 checksum
|
|
|
|
// or an RSA signature
|
|
|
|
//
|
2002-07-07 20:39:24 +00:00
|
|
|
int verify_downloaded_file(char* pathname, FILE_INFO& file_info) {
|
|
|
|
char cksum[64];
|
2003-07-03 05:01:29 +00:00
|
|
|
PROJECT* project = file_info.project;
|
2002-07-07 20:39:24 +00:00
|
|
|
bool verified;
|
|
|
|
int retval;
|
|
|
|
|
|
|
|
if (file_info.signature_required) {
|
2002-08-30 20:56:02 +00:00
|
|
|
if (!file_info.file_signature) {
|
|
|
|
fprintf(stdout, "ERROR: file %s missing signature\n", file_info.name);
|
|
|
|
return ERR_NO_SIGNATURE;
|
|
|
|
}
|
2002-07-07 20:39:24 +00:00
|
|
|
retval = verify_file2(
|
|
|
|
pathname, file_info.file_signature, project->code_sign_key, verified
|
|
|
|
);
|
2002-08-15 21:52:04 +00:00
|
|
|
if (retval) {
|
2003-07-03 05:01:29 +00:00
|
|
|
msg_printf(project, MSG_ERROR, "verify_downloaded_file(): internal error\n");
|
2002-08-15 21:52:04 +00:00
|
|
|
return ERR_RSA_FAILED;
|
2002-07-11 01:09:53 +00:00
|
|
|
}
|
2002-08-15 21:52:04 +00:00
|
|
|
if (!verified) {
|
2003-07-03 05:01:29 +00:00
|
|
|
msg_printf(project, MSG_ERROR, "verify_downloaded_file(): file not verified\n");
|
2002-08-13 22:36:54 +00:00
|
|
|
return ERR_RSA_FAILED;
|
2002-07-11 01:09:53 +00:00
|
|
|
}
|
2003-05-20 00:03:39 +00:00
|
|
|
} else if (strlen(file_info.md5_cksum)) {
|
2003-03-08 23:48:05 +00:00
|
|
|
retval = md5_file(pathname, cksum, file_info.nbytes);
|
|
|
|
if (strcmp(cksum, file_info.md5_cksum) || retval) {
|
2003-07-03 05:01:29 +00:00
|
|
|
msg_printf(project, MSG_ERROR, "verify_downloaded_file(): MD5 check failed\n");
|
2002-08-15 21:52:04 +00:00
|
|
|
return ERR_MD5_FAILED;
|
|
|
|
}
|
2002-07-07 20:39:24 +00:00
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2002-08-21 19:12:42 +00:00
|
|
|
// scan all FILE_INFOs and PERS_FILE_XFERs.
|
|
|
|
// start and finish downloads and uploads as needed.
|
2002-07-15 23:21:20 +00:00
|
|
|
//
|
2002-08-21 19:12:42 +00:00
|
|
|
bool CLIENT_STATE::handle_pers_file_xfers() {
|
2002-04-30 22:22:54 +00:00
|
|
|
unsigned int i;
|
|
|
|
FILE_INFO* fip;
|
2002-08-07 22:52:10 +00:00
|
|
|
PERS_FILE_XFER *pfx;
|
2002-04-30 22:22:54 +00:00
|
|
|
bool action = false;
|
|
|
|
|
2003-02-20 00:06:07 +00:00
|
|
|
// Look for FILE_INFOs for which we should start a transfer,
|
|
|
|
// and make PERS_FILE_XFERs for them
|
|
|
|
//
|
2002-04-30 22:22:54 +00:00
|
|
|
for (i=0; i<file_infos.size(); i++) {
|
|
|
|
fip = file_infos[i];
|
2002-08-07 22:52:10 +00:00
|
|
|
pfx = fip->pers_file_xfer;
|
2002-08-28 21:50:51 +00:00
|
|
|
if (pfx) continue;
|
2002-08-23 00:53:00 +00:00
|
|
|
if (!fip->generated_locally && fip->status == FILE_NOT_PRESENT) {
|
2002-08-07 22:52:10 +00:00
|
|
|
pfx = new PERS_FILE_XFER;
|
2002-08-15 21:52:04 +00:00
|
|
|
pfx->init(fip, false);
|
2002-08-07 22:52:10 +00:00
|
|
|
fip->pers_file_xfer = pfx;
|
2003-03-02 19:24:09 +00:00
|
|
|
pers_xfers->insert(fip->pers_file_xfer);
|
2002-04-30 22:22:54 +00:00
|
|
|
action = true;
|
2002-08-23 00:53:00 +00:00
|
|
|
} else if (fip->upload_when_present && fip->status == FILE_PRESENT && !fip->uploaded) {
|
|
|
|
|
2002-08-07 22:52:10 +00:00
|
|
|
pfx = new PERS_FILE_XFER;
|
2002-08-23 00:53:00 +00:00
|
|
|
pfx->init(fip, true);
|
2002-08-07 22:52:10 +00:00
|
|
|
fip->pers_file_xfer = pfx;
|
2003-03-02 19:24:09 +00:00
|
|
|
pers_xfers->insert(fip->pers_file_xfer);
|
2002-04-30 22:22:54 +00:00
|
|
|
action = true;
|
|
|
|
}
|
|
|
|
}
|
2002-08-21 19:12:42 +00:00
|
|
|
|
2003-02-20 00:06:07 +00:00
|
|
|
// Scan existing PERS_FILE_XFERs, looking for those that are done,
|
|
|
|
// and deleting them
|
|
|
|
//
|
|
|
|
vector<PERS_FILE_XFER*>::iterator iter;
|
|
|
|
iter = pers_xfers->pers_file_xfers.begin();
|
|
|
|
while (iter != pers_xfers->pers_file_xfers.end()) {
|
|
|
|
pfx = *iter;
|
2002-08-23 00:53:00 +00:00
|
|
|
|
2002-08-21 19:12:42 +00:00
|
|
|
// If the transfer finished, remove the PERS_FILE_XFER object
|
|
|
|
// from the set and delete it
|
2002-08-28 21:50:51 +00:00
|
|
|
//
|
2002-08-21 19:12:42 +00:00
|
|
|
if (pfx->xfer_done) {
|
2003-07-23 22:24:28 +00:00
|
|
|
// pfx->fip->pers_file_xfer = NULL;
|
|
|
|
FILE_INFO* DEBUG_fip = pfx->fip;
|
2003-02-20 00:06:07 +00:00
|
|
|
iter = pers_xfers->pers_file_xfers.erase(iter);
|
2002-08-21 19:12:42 +00:00
|
|
|
delete pfx;
|
|
|
|
action = true;
|
2003-07-23 22:24:28 +00:00
|
|
|
// `delete pfx' should have set pfx->fip->pfx to NULL
|
|
|
|
assert (DEBUG_fip == NULL || DEBUG_fip->pers_file_xfer == NULL);
|
2003-02-20 00:58:55 +00:00
|
|
|
} else {
|
|
|
|
iter++;
|
2002-08-21 19:12:42 +00:00
|
|
|
}
|
|
|
|
}
|
2003-07-02 02:02:18 +00:00
|
|
|
|
2002-04-30 22:22:54 +00:00
|
|
|
return action;
|
|
|
|
}
|