src/encoding.c

   1 /*
   2  * Copyright (c) 2006-2014 Erik Ekman <yarrick@kryo.se>,
   3  * 2006-2009 Bjorn Andersson <flex@kryo.se>
   4  *
   5  * Permission to use, copy, modify, and distribute this software for any
   6  * purpose with or without fee is hereby granted, provided that the above
   7  * copyright notice and this permission notice appear in all copies.
   8  *
   9  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
  10  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  11  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
  12  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  13  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  14  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  15  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  16  */
  17
  18 #include <string.h>
  19 #include "common.h"
  20 #include "encoding.h"
  21
  22 int
  23 build_hostname(char *buf, size_t buflen,
  24                 const char *data, const size_t datalen,
  25                 const char *topdomain, struct encoder *encoder, int maxlen)
  26 {
  27         size_t space;
  28         char *b;
  29
  30         space = MIN((size_t)maxlen, buflen) - strlen(topdomain) - 8;
  31         /* 8 = 5 max header length + 1 dot before topdomain + 2 safety */
  32
  33         if (!encoder->places_dots())
  34                 space -= (space / 57); /* space for dots */
  35
  36         memset(buf, 0, buflen);
  37
  38         encoder->encode(buf, &space, data, datalen);
  39
  40         if (!encoder->places_dots())
  41                 inline_dotify(buf, buflen);
  42
  43         b = buf;
  44         b += strlen(buf);
  45
  46         /* move b back one step to see if the dot is there */
  47         b--;
  48         if (*b != '.')
  49                 *++b = '.';
  50         b++;
  51         /* move b ahead of the string so we can copy to it */
  52
  53         strncpy(b, topdomain, strlen(topdomain)+1);
  54
  55         return space;
  56 }
  57
  58 int
  59 unpack_data(char *buf, size_t buflen, char *data, size_t datalen, struct encoder *enc)
  60 {
  61         if (!enc->eats_dots())
  62                 datalen = inline_undotify(data, datalen);
  63         return enc->decode(buf, &buflen, data, datalen);
  64 }
  65
  66 int
  67 inline_dotify(char *buf, size_t buflen)
  68 {
  69         unsigned dots;
  70         unsigned pos;
  71         unsigned total;
  72         char *reader, *writer;
  73
  74         total = strlen(buf);
  75         dots = total / 57;
  76
  77         writer = buf;
  78         writer += total;
  79         writer += dots;
  80
  81         total += dots;
  82         if (strlen(buf) + dots > buflen) {
  83                 writer = buf;
  84                 writer += buflen;
  85                 total = buflen;
  86         }
  87
  88         reader = writer - dots;
  89         pos = (unsigned) (reader - buf) + 1;
  90
  91         while (dots) {
  92                 *writer-- = *reader--;
  93                 pos--;
  94                 if (pos % 57 == 0) {
  95                         *writer-- = '.';
  96                         dots--;
  97                 }
  98         }
  99
 100         /* return new length of string */
 101         return total;
 102 }
 103
 104 int
 105 inline_undotify(char *buf, size_t len)
 106 {
 107         unsigned pos;
 108         unsigned dots;
 109         char *reader, *writer;
 110
 111         writer = buf;
 112         reader = writer;
 113
 114         pos = 0;
 115         dots = 0;
 116
 117         while (pos < len) {
 118                 if (*reader == '.') {
 119                         reader++;
 120                         pos++;
 121                         dots++;
 122                         continue;
 123                 }
 124                 *writer++ = *reader++;
 125                 pos++;
 126         }
 127
 128         /* return new length of string */
 129         return len - dots;
 130 }