mirror of https://github.com/BOINC/boinc.git
146 lines
4.1 KiB
C
146 lines
4.1 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):
|
|
//
|
|
|
|
#include <string.h>
|
|
#include <stdlib.h>
|
|
#include <assert.h>
|
|
|
|
#include "db.h"
|
|
#include "error_numbers.h"
|
|
#include "parse.h"
|
|
#include "crypt.h"
|
|
|
|
#define OUTFILE_MACRO "<OUTFILE_"
|
|
#define UPLOAD_URL_MACRO "<UPLOAD_URL/>"
|
|
|
|
// compute an XML signature element for some text
|
|
//
|
|
int generate_signature(
|
|
char* signed_xml, char* signature_xml, R_RSA_PRIVATE_KEY& key
|
|
) {
|
|
DATA_BLOCK block, signature;
|
|
unsigned char signature_buf[SIGNATURE_SIZE_BINARY];
|
|
char buf[MAX_BLOB_SIZE];
|
|
int retval;
|
|
|
|
block.data = (unsigned char*)signed_xml;
|
|
block.len = strlen(signed_xml);
|
|
signature.data = signature_buf;
|
|
signature.len = SIGNATURE_SIZE_BINARY;
|
|
retval = sign_block(block, key, signature);
|
|
if (retval) return retval;
|
|
sprint_hex_data(buf, signature);
|
|
#if 0
|
|
printf("signing [\n%s]\n", signed_xml);
|
|
printf("signature: [\n%s]\n", buf);
|
|
#endif
|
|
sprintf(signature_xml,
|
|
"<xml_signature>\n%s</xml_signature>\n", buf
|
|
);
|
|
return 0;
|
|
}
|
|
|
|
// At the end of every <file_info> element,
|
|
// add a signature of its contents up to that point.
|
|
//
|
|
int add_signatures(char* xml, R_RSA_PRIVATE_KEY& key) {
|
|
char* p = xml, *q1, *q2, buf[MAX_BLOB_SIZE], buf2[MAX_BLOB_SIZE];;
|
|
char signature[MAX_BLOB_SIZE];
|
|
int retval, len;
|
|
|
|
while (1) {
|
|
q1 = strstr(p, "<file_info>\n");
|
|
if (!q1) break;
|
|
q2 = strstr(q1, "</file_info>");
|
|
if (!q2) {
|
|
fprintf(stderr, "add_signatures: malformed XML: %s\n", xml);
|
|
return ERR_XML_PARSE;
|
|
}
|
|
q1 += strlen("<file_info>\n");
|
|
len = q2 - q1;
|
|
memcpy(buf, q1, len);
|
|
buf[len] = 0;
|
|
retval = generate_signature(buf, signature, key);
|
|
if (retval) return retval;
|
|
strcpy(buf2, q2);
|
|
strcpy(q1, buf);
|
|
strcat(q1, signature);
|
|
strcat(q1, buf2);
|
|
p = q1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
// remove file upload signatures from a result XML doc
|
|
//
|
|
int remove_signatures(char* xml) {
|
|
char* p, *q;
|
|
while (1) {
|
|
p = strstr(xml, "<xml_signature>");
|
|
if (!p) break;
|
|
q = strstr(p, "</xml_signature>");
|
|
if (!q) {
|
|
fprintf(stderr, "remove_signatures: invalid XML:\n%s", xml);
|
|
return ERR_XML_PARSE;
|
|
}
|
|
q += strlen("</xml_signature>");
|
|
strcpy(p, q);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
// macro-substitute a result template:
|
|
// - replace OUTFILE_x with base_filename_x, etc.
|
|
// - add signatures for file uploads
|
|
//
|
|
int process_result_template(
|
|
char* result_template,
|
|
R_RSA_PRIVATE_KEY& key,
|
|
char* base_filename,
|
|
char* upload_url, char* download_url
|
|
) {
|
|
char* p,*q;
|
|
char temp[MAX_BLOB_SIZE];
|
|
char num;
|
|
int i;
|
|
|
|
while (1) {
|
|
p = strstr(result_template, OUTFILE_MACRO);
|
|
if (p) {
|
|
i = atoi(p+strlen(OUTFILE_MACRO));
|
|
q = p+strlen(OUTFILE_MACRO);
|
|
num = q[0];
|
|
strcpy(temp, p+strlen(OUTFILE_MACRO)+1+2);
|
|
strcpy(p, base_filename);
|
|
strncat(p, &num, 1);
|
|
strcat(p, temp);
|
|
continue;
|
|
}
|
|
p = strstr(result_template, UPLOAD_URL_MACRO);
|
|
if (p) {
|
|
strcpy(temp, p+strlen(UPLOAD_URL_MACRO));
|
|
strcpy(p, upload_url);
|
|
strcat(p, temp);
|
|
continue;
|
|
}
|
|
break;
|
|
}
|
|
return add_signatures(result_template, key);
|
|
}
|