ftp.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 <stdlib.h>
#include <stdio.h>
#include <string.h>                                  
#include <sys/types.h>
#include <unistd.h>
#include <errno.h>

#include <curl/curl.h>
#include <curl/types.h>
#include <curl/easy.h>
#include <glib.h>

#include "ftp.h"
#include "util.h"
#include "libuau.h"
#include "error.h"

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

static AProgressCallback downloadCallback_global = NULL;

static size_t callback(void *ptr, size_t size, size_t nmemb, void *data);

gboolean
lutil_ftp_downloadToFile(const char* url, const char* downloadTo) {
        CURL *curl_handle;
        CURLcode ret;
        FILE *tempFile;
        char errorBuffer[CURL_ERROR_SIZE];
        
        tempFile = fopen(downloadTo, "w");
        if (tempFile == NULL) {
                ERROR("Couldn't download file to temporary location %s: %s", downloadTo, strerror(errno));
                return FALSE;
        }
        
        DBUGOUT("Retrieving url %s to %s", url, downloadTo);
        
        curl_handle = curl_easy_init();

        curl_easy_setopt(curl_handle, CURLOPT_URL, url);
        curl_easy_setopt(curl_handle, CURLOPT_FILE, (void *)tempFile);
        curl_easy_setopt(curl_handle, CURLOPT_ERRORBUFFER, errorBuffer);
        curl_easy_setopt(curl_handle, CURLOPT_FAILONERROR, 1);
        if (downloadCallback_global != NULL) {
                curl_easy_setopt(curl_handle, CURLOPT_NOPROGRESS, FALSE);
                curl_easy_setopt(curl_handle, CURLOPT_PROGRESSFUNCTION, downloadCallback_global);
        }
        ret = curl_easy_perform(curl_handle);

        curl_easy_cleanup(curl_handle);
        
        fclose(tempFile);
        
        if (ret != 0) {
                unlink(downloadTo);
                ERROR("Couldn't download %s: %s", url, errorBuffer);
                return FALSE;
        } else {
                return TRUE;
        }
}


GString *
lutil_ftp_getURL(const char* url) {
        AData chunk;
        GString *data;
        CURL *curl_handle;
        CURLcode ret;
        char errorBuffer[CURL_ERROR_SIZE];

        chunk.data = NULL;
        chunk.size = 0;
        
        DBUGOUT("Retrieving url: %s", url);
        
        curl_handle = curl_easy_init();

        curl_easy_setopt(curl_handle, CURLOPT_URL, url);
        curl_easy_setopt(curl_handle, CURLOPT_WRITEFUNCTION, callback);
        curl_easy_setopt(curl_handle, CURLOPT_FILE, (void *)&chunk);
        curl_easy_setopt(curl_handle, CURLOPT_ERRORBUFFER, errorBuffer);
        curl_easy_setopt(curl_handle, CURLOPT_FAILONERROR, 1);
        
        /*if (downloadCallback_global != NULL) {
                curl_easy_setopt(curl_handle, CURLOPT_NOPROGRESS, FALSE);
                curl_easy_setopt(curl_handle, CURLOPT_PROGRESSFUNCTION, downloadCallback_global);
        }*/
        
        ret = curl_easy_perform(curl_handle);

        curl_easy_cleanup(curl_handle);

        if (ret != 0) {
                g_free(chunk.data);
                ERROR("Couldn't download %s: %s", url, errorBuffer);
                return NULL;
        } else {
                data = g_string_new_len(chunk.data, chunk.size);
                g_free(chunk.data);
                return data;
        }
}

void
lutil_ftp_setCallbackFunc(AProgressCallback callback) {
        downloadCallback_global = callback;
}

void
lutil_ftp_resetCallbackFunc(void) {
        downloadCallback_global = NULL;
}

/* Non-Interface Methods */

static size_t
callback(void *ptr, size_t size, size_t nmemb, void *data) {
        int actualSize = size * nmemb;
        AData *file = (AData *)data;
        
        file->data = (char *) g_realloc(file->data, file->size + actualSize + 1);
        if (file->data != NULL) {
                memcpy(&(file->data[file->size]), ptr, actualSize);
                file->size += actualSize;
                file->data[file->size] = '\0';
        }
        return actualSize;
}

---