tests/lib/fetch.c

   1 /* libquvi-scripts
   2  * Copyright (C) 2012  Toni Gundogdu <legatvs@gmail.com>
   3  *
   4  * This file is part of libquvi-scripts <http://quvi.sourceforge.net>.
   5  *
   6  * This program is free software: you can redistribute it and/or
   7  * modify it under the terms of the GNU Affero General Public
   8  * License as published by the Free Software Foundation, either
   9  * version 3 of the License, or (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU Affero General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Affero General
  17  * Public License along with this program.  If not, see
  18  * <http://www.gnu.org/licenses/>.
  19  */
  20
  21 #include "config.h"
  22
  23 #include <string.h>
  24 #include <glib.h>
  25 #include <quvi.h>
  26 #include <curl/curl.h>
  27
  28 #include "tests.h"
  29
  30 struct temp_s
  31 {
  32   gsize size;
  33   gchar *p;
  34 };
  35 typedef struct temp_s *temp_t;
  36
  37 static gpointer temp_new()
  38 {
  39   return (g_new0(struct temp_s, 1));
  40 }
  41
  42 static void temp_free(temp_t t)
  43 {
  44   if (t == NULL)
  45     return;
  46
  47   g_free(t->p);
  48   t->p = NULL;
  49
  50   g_free(t);
  51   t = NULL;
  52 }
  53
  54 /* cURL write callback. */
  55 static gsize temp_wrcb(gpointer p, gsize sz, gsize nmemb, gpointer d)
  56 {
  57   const gsize rsize = sz*nmemb;
  58   gpointer *np;
  59   temp_t t;
  60
  61   t = (temp_t) d;
  62   np = g_realloc(t->p, t->size+rsize+1);
  63
  64   if (np != NULL)
  65     {
  66       t->p = (gchar*) np;
  67       memcpy(&(t->p[t->size]), p, rsize);
  68       t->size += rsize;
  69       t->p[t->size] = '\0';
  70     }
  71   return (rsize);
  72 }
  73
  74 static void set_opts(CURL *c, temp_t t, const gchar *url)
  75 {
  76   typedef curl_write_callback cwc;
  77
  78   curl_easy_setopt(c, CURLOPT_USERAGENT, "Mozilla/5.0");
  79   curl_easy_setopt(c, CURLOPT_FOLLOWLOCATION, 1L);
  80   curl_easy_setopt(c, CURLOPT_MAXREDIRS, 5L); /* http://is.gd/kFsvE4 */
  81   curl_easy_setopt(c, CURLOPT_NOBODY, 0L);
  82
  83   curl_easy_setopt(c, CURLOPT_WRITEFUNCTION, (cwc) temp_wrcb);
  84   curl_easy_setopt(c, CURLOPT_URL, url);
  85   curl_easy_setopt(c, CURLOPT_WRITEDATA, t);
  86   /* CURLOPT_ENCODING -> CURLOPT_ACCEPT_ENCODING 7.21.6+ */
  87   curl_easy_setopt(c, CURLOPT_ENCODING, "");
  88
  89   if (chk_env("TEST_VERBOSE", NULL) == TRUE)
  90     curl_easy_setopt(c, CURLOPT_VERBOSE, 1L);
  91 }
  92
  93 static void reset_opts(CURL *c)
  94 {
  95   curl_easy_setopt(c, CURLOPT_WRITEFUNCTION, NULL);
  96   curl_easy_setopt(c, CURLOPT_WRITEDATA, NULL);
  97 }
  98
  99 static gint _fetch(CURL *c)
 100 {
 101   CURLcode curlcode;
 102   glong respcode;
 103   gint rc;
 104
 105   curlcode = curl_easy_perform(c);
 106   curl_easy_getinfo(c, CURLINFO_RESPONSE_CODE, &respcode);
 107
 108   rc = 0;
 109
 110   if (curlcode == CURLE_OK && respcode == 200)
 111     ;
 112   else
 113     {
 114       if (curlcode == CURLE_OK)
 115         {
 116 #define _EOK "server responded with code %03ld"
 117           g_test_message(_EOK, respcode);
 118 #undef _EOK
 119         }
 120       else
 121         {
 122           const gchar *s = curl_easy_strerror(curlcode);
 123           const glong c = respcode;
 124           const gint cc = curlcode;
 125 #define _ENO "%s (HTTP/%03ld, cURL=0x%03x)"
 126           g_test_message(_ENO, s, c, cc);
 127 #undef _ENO
 128         }
 129       rc = 1;
 130     }
 131   return (rc);
 132 }
 133
 134 gchar *fetch(const gchar *url)
 135 {
 136   gchar *r;
 137   temp_t t;
 138   CURL *c;
 139   gint rc;
 140
 141   curl_global_init(CURL_GLOBAL_ALL);
 142
 143   c = curl_easy_init();
 144   if (c == NULL)
 145     {
 146       g_message("curl_easy_init returned NULL");
 147       return (NULL);
 148     }
 149
 150   t = temp_new();
 151   r = NULL;
 152
 153   set_opts(c, t, url);
 154   rc = _fetch(c);
 155   reset_opts(c);
 156
 157   if (rc == 0)
 158     r = g_strdup(t->p);
 159
 160   temp_free(t);
 161   t = NULL;
 162
 163   curl_easy_cleanup(c);
 164   c = NULL;
 165
 166   curl_global_cleanup();
 167
 168   return (r);
 169 }
 170
 171 /* vim: set ts=2 sw=2 tw=72 expandtab: */