src/url.c

   1 #ifdef HAVE_CONFIG_H
   2 #include <config.h>
   3 #endif
   4 #include <stdlib.h>
   5 #include <string.h>
   6 #include <ctype.h>
   7
   8 #include "common.h"
   9 #include "log.h"
  10 #include "url.h"
  11
  12 // ----------------------------------------------------------------------------
  13 // URL encode / decode
  14 // code from: http://www.geekhideout.com/urlcode.shtml
  15
  16 /* Converts a hex character to its integer value */
  17 char from_hex(char ch) {
  18         return (char)(isdigit(ch) ? ch - '0' : tolower(ch) - 'a' + 10);
  19 }
  20
  21 /* Converts an integer value to its hex character*/
  22 char to_hex(char code) {
  23         static char hex[] = "0123456789abcdef";
  24         return hex[code & 15];
  25 }
  26
  27 /* Returns a url-encoded version of str */
  28 /* IMPORTANT: be sure to free() the returned string after use */
  29 char *url_encode(char *str) {
  30         char *buf, *pbuf;
  31
  32         pbuf = buf = malloc(strlen(str) * 3 + 1);
  33
  34         if(!buf)
  35                 fatal("Cannot allocate memory.");
  36
  37         while (*str) {
  38                 if (isalnum(*str) || *str == '-' || *str == '_' || *str == '.' || *str == '~')
  39                         *pbuf++ = *str;
  40
  41                 else if (*str == ' ')
  42                         *pbuf++ = '+';
  43
  44                 else
  45                         *pbuf++ = '%', *pbuf++ = to_hex(*str >> 4), *pbuf++ = to_hex(*str & 15);
  46
  47                 str++;
  48         }
  49         *pbuf = '\0';
  50
  51         // FIX: I think this is prudent. URLs can be as long as 2 KiB or more.
  52         //      We allocated 3 times more space to accomodate %NN encoding of
  53         //      non ASCII chars. If URL has none of these kind of chars we will
  54         //      end up with a big unused buffer.
  55         //
  56         //      Try to shrink the buffer...
  57         if (!!(pbuf = (char *)realloc(buf, strlen(buf)+1)))
  58                 buf = pbuf;
  59
  60         return buf;
  61 }
  62
  63 /* Returns a url-decoded version of str */
  64 /* IMPORTANT: be sure to free() the returned string after use */
  65 char *url_decode(char *str) {
  66         char *pstr = str,
  67                 *buf = malloc(strlen(str) + 1),
  68                 *pbuf = buf;
  69
  70         if(!buf)
  71                 fatal("Cannot allocate memory.");
  72
  73         while (*pstr) {
  74                 if (*pstr == '%') {
  75                         if (pstr[1] && pstr[2]) {
  76                                 *pbuf++ = from_hex(pstr[1]) << 4 | from_hex(pstr[2]);
  77                                 pstr += 2;
  78                         }
  79                 }
  80                 else if (*pstr == '+')
  81                         *pbuf++ = ' ';
  82
  83                 else
  84                         *pbuf++ = *pstr;
  85
  86                 pstr++;
  87         }
  88
  89         *pbuf = '\0';
  90
  91         return buf;
  92 }
  93