boinc/sched/time_stats_log.cpp

84 lines
2.4 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/>.
#include <sys/types.h>
#include <sys/stat.h>
#include "filesys.h"
#include "parse.h"
#include "sched_msgs.h"
#include "time_stats_log.h"
static char* stats_buf = 0;
// Got a <time_stats_log> flag in scheduler request.
// Copy the contents to a malloced buffer;
// don't write them to disk yet, since we haven't authenticated the host
//
void handle_time_stats_log(FILE* fin) {
dup_element_contents(fin, "</time_stats_log>", &stats_buf);
}
// The host has been authenticated, so write the stats.
// Use a directory hierarchy since there may be many hosts
//
void write_time_stats_log(SCHEDULER_REPLY& reply) {
char dirname[256], filename[256];
int hostid = reply.host.id;
int dirnum = hostid % 1000;
sprintf(dirname, "../time_stats_log/%d", dirnum);
if (!is_dir(dirname)) {
int retval = boinc_mkdir(dirname);
if (retval) {
log_messages.printf(MSG_CRITICAL,
"Can't make time stats log dir %s: %d\n", dirname, retval
);
perror("mkdir");
return;
}
}
sprintf(filename, "../time_stats_log/%d/%d", dirnum, hostid);
#ifndef _USING_FCGI_
FILE* f = fopen(filename, "w");
#else
FCGI_FILE *f = FCGI::fopen(filename, "w");
#endif
if (!f) {
log_messages.printf(MSG_CRITICAL,
"Can't create time stats file %s\n", filename
);
return;
}
fputs(stats_buf, f);
fclose(f);
free(stats_buf);
stats_buf = 0;
}
bool have_time_stats_log(SCHEDULER_REPLY& reply) {
char filename[256];
int hostid = reply.host.id;
int dirnum = hostid % 1000;
sprintf(filename, "../time_stats_log/%d/%d", dirnum, hostid);
return is_file(filename);
}