*
*/
+#ifdef HAVE_CONFIG_H
+#include "config.h"
+#endif
+
+#include <stdio.h>
+#include <errno.h>
+#include <ctype.h>
+#include <string.h>
+#include <strings.h>
+#include <gtk/gtk.h>
+#include <sys/stat.h>
+#include <sys/types.h>
+
#include "download.h"
+
+#ifdef HAVE_LIBCURL
+#include "curl_download.h"
+#else
#include "http.h"
+#endif
-/* success = 0, -1 = couldn't connect, -2 HTTP error, -3 file exists, -4 couldn't write to file... */
-/* uri: like "/uri.html?whatever" */
-/* only reason for the "wrapper" is so we can do redirects. */
-int a_http_download_get_url ( const char *hostname, const char *uri, const char *fn )
+#ifdef WINDOWS
+
+#include <io.h>
+#define access(a,b) _access(a,b)
+#define close(a) closesocket(a)
+
+char *dirname ( char * dir )
+{
+ char *tmp = dir + strlen(dir) - 1;
+ while ( tmp != dir && *tmp != '\\' )
+ tmp--;
+ *tmp = '\0';
+ return dir;
+}
+
+#else
+
+#include <unistd.h>
+#include <sys/types.h>
+
+/* dirname */
+#include <libgen.h>
+
+#endif
+
+static int check_map_file(FILE* f)
{
- return http_download_get_url ( hostname, uri, fn, 0, 1 );
+ char **s;
+ char *bp;
+ int res = 0; /* good */
+ fpos_t pos;
+ char buf[33];
+ size_t nr;
+ char * html_str[] = {
+ "<html",
+ "<!DOCTYPE html",
+ "<head",
+ "<title",
+ NULL
+ };
+
+
+ bzero(buf, sizeof(buf));
+ fgetpos(f, &pos);
+ rewind(f);
+ nr = fread(buf, 1, sizeof(buf) - 1, f);
+ fsetpos(f, &pos);
+ for (bp = buf; (bp < (buf + sizeof(buf) - 1)) && (nr > (bp - buf)); bp++) {
+ if (!(isspace(*bp)))
+ break;
+ }
+ if ((bp >= (buf + sizeof(buf) -1)) || ((bp - buf) >= nr))
+ return(res);
+ for (s = html_str; *s; s++) {
+ if (strncmp(*s, bp, strlen(*s)) == 0)
+ return(-1);
+ }
+ return(res);
+}
+
+static int download( const char *hostname, const char *uri, const char *fn, DownloadOptions *options)
+{
+ FILE *f;
+ int ret;
+ char *tmpfilename;
+
+ /* Check file */
+ if ( access ( fn, F_OK ) == 0 )
+ {
+ /* File exists: return */
+ return -3;
+ } else {
+ if ( errno == ENOENT)
+ {
+ char *tmp = g_strdup ( fn );
+#ifdef WINDOWS
+ mkdir( dirname ( dirname ( tmp ) ) );
+ g_free ( tmp ); tmp = g_strdup ( fn );
+ mkdir( dirname ( tmp ) );
+#else
+ mkdir( dirname ( dirname ( tmp ) ), 0777 );
+ g_free ( tmp ); tmp = g_strdup ( fn );
+ mkdir( dirname ( tmp ), 0777 );
+#endif
+ g_free ( tmp );
+ }
+ /* create placeholder file */
+ if ( ! (f = fopen ( fn, "w+b" )) ) /* immediately open file so other threads won't -- prevents race condition */
+ return -4;
+ fclose ( f );
+ }
+
+ tmpfilename = g_strdup_printf("%s.tmp", fn);
+ f = fopen ( tmpfilename, "w+b" );
+ if ( ! f ) {
+ g_free ( tmpfilename );
+ remove ( fn ); /* couldn't create temporary. delete 0-byte file. */
+ return -4;
+ }
+
+ /* Call the backend function */
+#ifdef HAVE_LIBCURL
+ ret = curl_download_get_url ( hostname, uri, f, options );
+#else
+ ret = http_download_get_url ( hostname, uri, f, 0, options );
+#endif
+
+ if (ret == -1 || ret == 1 || ret == -2 || check_map_file(f))
+ {
+ fprintf(stderr, "Download error: %s\n", fn);
+ fclose ( f );
+ remove ( tmpfilename );
+ g_free ( tmpfilename );
+ remove ( fn ); /* couldn't create temporary. delete 0-byte file. */
+ return -1;
+ }
+
+ fclose ( f );
+ rename ( tmpfilename, fn ); /* move completely-downloaded file to permanent location */
+ g_free ( tmpfilename );
+ return ret;
}
-int a_http_download_get_url_nohostname ( const char *hostname, const char *uri, const char *fn )
+/* success = 0, -1 = couldn't connect, -2 HTTP error, -3 file exists, -4 couldn't write to file... */
+/* uri: like "/uri.html?whatever" */
+/* only reason for the "wrapper" is so we can do redirects. */
+int a_http_download_get_url ( const char *hostname, const char *uri, const char *fn, DownloadOptions *opt )
{
- return http_download_get_url ( hostname, uri, fn, 0, 0 );
+ return download ( hostname, uri, fn, opt );
}
projects / andy / viking.git / blobdiff