boinc/client/acct_mgr.C

// Berkeley Open Infrastructure for Network Computing
// http://boinc.berkeley.edu
// Copyright (C) 2005 University of California
//
// This 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 2.1 of the License, or (at your option) any later version.
//
// This software 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.
//
// To view the GNU Lesser General Public License visit
// http://www.gnu.org/copyleft/lesser.html
// or write to the Free Software Foundation, Inc.,
// 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

#ifdef _WIN32
#include "boinc_win.h"
#endif
#include "parse.h"
#include "error_numbers.h"
#include "client_msgs.h"
#include "util.h"
#include "file_names.h"
#include "filesys.h"
#include "client_state.h"
#include "gui_http.h"

#include "acct_mgr.h"


int ACCT_MGR_OP::do_rpc(std::string url, std::string name, std::string password) {
    int retval;
    char buf[256];

    strcpy(buf, url.c_str());

    if (!strlen(buf) && strlen(gstate.acct_mgr_info.acct_mgr_url)) {
        msg_printf(NULL, MSG_ALERT_INFO, "Removing account manager info");
        gstate.acct_mgr_info.clear();
        boinc_delete_file(ACCT_MGR_URL_FILENAME);
        boinc_delete_file(ACCT_MGR_LOGIN_FILENAME);
        return 0;
    }

    canonicalize_master_url(buf);
    if (!valid_master_url(buf)) {
        msg_printf(NULL, MSG_ALERT_ERROR, "Can't contact account manager:\n'%s' is not a valid URL", url.c_str());
        return 0;
    }
    strcpy(ami.acct_mgr_url, url.c_str());
    strcpy(ami.acct_mgr_name, "");
    strcpy(ami.login_name, name.c_str());
    strcpy(ami.password, password.c_str());

    sprintf(buf, "%s?name=%s&password=%s", url.c_str(), name.c_str(), password.c_str());
    retval = gstate.gui_http.do_rpc(this, buf, ACCT_MGR_REPLY_FILENAME);
    if (retval) {
        msg_printf(NULL, MSG_ALERT_ERROR,
            "Can't contact account manager at '%s'.\nPlease check the URL and try again",
            url.c_str()
        );
        return retval;
    }
    msg_printf(NULL, MSG_INFO, "Doing account manager RPC to %s", url.c_str());

    return 0;
}



int ACCOUNT::parse(MIOFILE& in) {
    char buf[256];
    while (in.fgets(buf, sizeof(buf))) {
        if (match_tag(buf, "</account>")) {
            if (url.length() && authenticator.length()) return 0;
            return ERR_XML_PARSE;
        }
        if (parse_str(buf, "<url>", url)) continue;
        if (parse_str(buf, "<authenticator>", authenticator)) continue;
    }
    return ERR_XML_PARSE;
}

int ACCT_MGR_OP::parse(MIOFILE& in) {
    char buf[256];
    int retval;

    accounts.clear();
    error_str = "";
    while (in.fgets(buf, sizeof(buf))) {
        if (match_tag(buf, "</acct_mgr_reply>")) return 0;
        if (parse_str(buf, "<name>", ami.acct_mgr_name, 256)) continue;
        if (parse_str(buf, "<error>", error_str)) continue;
        if (match_tag(buf, "<account>")) {
            ACCOUNT account;
            retval = account.parse(in);
            if (!retval) accounts.push_back(account);
        }
    }
    return ERR_XML_PARSE;
}

void ACCT_MGR_OP::handle_reply(int http_op_retval) {
    unsigned int i;
    int retval;

    if (http_op_retval == 0) {
        FILE* f = fopen(ACCT_MGR_REPLY_FILENAME, "r");
        if (f) {
            MIOFILE mf;
            mf.init_file(f);
            retval = parse(mf);
            fclose(f);
        } else {
            retval = ERR_FOPEN;
        }
    } else {
        retval = http_op_retval;
    }
    if (retval) {
        msg_printf(NULL, MSG_ALERT_ERROR,
            "Account manager update failed:\n%s", boincerror(retval)
        );
        return;
    }

    msg_printf(NULL, MSG_INFO, "Handling account manager RPC reply");
    if (error_str.size()) {
        msg_printf(NULL, MSG_ALERT_ERROR, "Account manager update failed:\n%s", error_str.c_str());
        return;
    }
    msg_printf(NULL, MSG_ALERT_INFO, "Account manager update succeeded.\nSee Messages for more info.");

    gstate.acct_mgr_info = ami;
    gstate.acct_mgr_info.write_info();

    for (i=0; i<accounts.size(); i++) {
        ACCOUNT& acct = accounts[i];
        PROJECT* pp = gstate.lookup_project(acct.url.c_str());
        if (pp) {
            if (strcmp(pp->authenticator, acct.authenticator.c_str())) {
                msg_printf(pp, MSG_ERROR,
                    "You're attached to this project with a different account"
                );
            } else {
                msg_printf(pp, MSG_INFO, "Already attached");
            }
        } else {
            msg_printf(NULL, MSG_INFO, "Attaching to %s", acct.url.c_str());
            gstate.add_project(acct.url.c_str(), acct.authenticator.c_str());
        }
    }
    if (accounts.size() == 0) {
        msg_printf(NULL, MSG_ERROR, "No accounts reported by account manager");
    }
}

int ACCT_MGR_INFO::write_info() {
    FILE* p;
    if (strlen(acct_mgr_url)) {
        p = fopen(ACCT_MGR_URL_FILENAME, "w");
        if (p) {
            fprintf(
                p, 
                "<acct_mgr>\n"
                "    <name>%s</name>\n"
                "    <url>%s</url>\n"
                "</acct_mgr>\n",
                acct_mgr_name,
                acct_mgr_url
            );
            fclose(p);
        }
    }

    if (strlen(login_name)) {
        p = fopen(ACCT_MGR_LOGIN_FILENAME, "w");
        if (p) {
            fprintf(
                p, 
                "<acct_mgr_login>\n"
                "    <login>%s</login>\n"
                "    <password>%s</password>\n"
                "</acct_mgr_login>\n",
                login_name,
                password
            );
            fclose(p);
        }
    }
    return 0;
}

void ACCT_MGR_INFO::clear() {
    strcpy(acct_mgr_name, "");
    strcpy(acct_mgr_url, "");
    strcpy(login_name, "");
    strcpy(password, "");
}

ACCT_MGR_INFO::ACCT_MGR_INFO() {
    clear();
}

int ACCT_MGR_INFO::init() {
    char    buf[256];
    MIOFILE mf;
    FILE*   p;

    clear();
    p = fopen(ACCT_MGR_URL_FILENAME, "r");
    if (p) {
        mf.init_file(p);
        while(mf.fgets(buf, sizeof(buf))) {
            if (match_tag(buf, "</acct_mgr>")) break;
            else if (parse_str(buf, "<name>", acct_mgr_name, 256)) continue;
            else if (parse_str(buf, "<url>", acct_mgr_url, 256)) continue;
        }
        fclose(p);
    } else {
        return 0;
    }

    p = fopen(ACCT_MGR_LOGIN_FILENAME, "r");
    if (p) {
        mf.init_file(p);
        while(mf.fgets(buf, sizeof(buf))) {
            if (match_tag(buf, "</acct_mgr_login>")) break;
            else if (parse_str(buf, "<login>", login_name, 256)) continue;
            else if (parse_str(buf, "<password>", password, 256)) continue;
        }
        fclose(p);
    }
    return 0;
}
const char *BOINC_RCSID_8fd9e873bf="$Id$";