network.c

Go to the documentation of this file.

/*
 * luau (Lib Update/Auto-Update): Simple Update Library
 * Copyright (C) 2003  David Eklund
 *
 * - This library 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 library 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 this library; if not, write to the Free Software       -
 * - Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA -
 */

#ifdef HAVE_CONFIG_H
#  include <config.h>
#endif

#include <unistd.h>
#include <string.h>

#include "libuau.h"
#include "network.h"
#include "ftp.h"
#include "util.h"
#include "error.h"
#include "parseupdates.h"
#include "md5.h"

#ifdef WITH_DMALLOC
#  include <dmalloc.h>
#endif

GPtrArray *
luau_net_queryServer(const AProgInfo *info) {
        GPtrArray *updates;
        GString *contents, *temp;
        int len;
        
        if (info->url == NULL) {
                ERROR("Can't check for updates: no URL specified in database");
                return NULL;
        }
        contents = lutil_ftp_getURL(info->url);
        if (contents == NULL) {
                ERROR("Couldn't read/download file %s", info->url);
                return NULL;
        }
        
        len = strlen(info->url);
        if (len > 3 && lutil_streq(info->url+len-3, ".gz")) {
                DBUGOUT("Updates file is compressed: uncompressing");
                temp = lutil_uncompress(contents);
                g_string_free(contents, TRUE);
                if (temp == NULL) {
                        ERROR("Couldn't uncompress gzipped file: aborting");
                        return NULL;
                }
                contents = temp;
        }
        
        updates = luau_parseUpdateFileXML(contents->str);
        
        g_string_free(contents, TRUE);
        
        return updates;
}

gboolean
luau_net_downloadUpdate(const AProgInfo* info, const AUpdate *update, APkgType pkgType, const char* downloadTo) {
        char md5[33], *loc;
        APackage *package;
        gboolean loopAgain, result;
        int choice;
        
        md5[0] = '\0';
        
        package = luau_getUpdatePackage(update, pkgType);
        
        if (package == NULL || package->mirrors == NULL || package->mirrors->len == 0) {
                ERROR("Couldn't get filename for this update!");
                return FALSE;
        }
        
        loc = luau_getPackageURL(package);
        result = lutil_ftp_downloadToFile(loc, downloadTo);
        if (result == FALSE) {
                ERROR("Couldn't retrieve file");
                return FALSE;
        }
        
        if (package->md5sum[0] != '\0') {
                lutil_md5_file(downloadTo, md5);
                if (md5[0] == '\0') {
                        ERROR("Couldn't compute md5sum of file");
                        return FALSE;
                }
                
                result = FALSE;
                if (! lutil_streq(md5, package->md5sum)) {
                        while (1) {
                                loopAgain = FALSE;
                                choice = lutil_error_prompt("MD5 Sum Mismatch", "The computed md5 sum of the downloaded file does not match the expected value", 3, 2, "Re-Download", "Use Anyway", "Cancel");
                                switch (choice) {
                                        case 0: /* Re-Download */
                                                unlink(downloadTo);
                                                result = luau_net_downloadUpdate(info, update, pkgType, downloadTo);
                                                break;
                                        case 1: /* Use Anyway */
                                                result = TRUE;
                                                break;
                                        case 2: /* Cancel */
                                                ERROR("Couldn't download update: md5sum mismatch (found %s, expected %s)", md5, package->md5sum);
                                                result = FALSE;
                                                unlink(downloadTo);
                                                break;
                                        default:
                                                loopAgain = TRUE;
                                }
                                
                                if (!loopAgain)
                                        break;
                        }
                } else {
                        result = TRUE;
                }
        } else {
                result = TRUE;
        }
        
        return result;
}

---