* New upstream release:
[debian/iodine.git] / src / encoding.c
1 /*
2  * Copyright (c) 2006-2009 Bjorn Andersson <flex@kryo.se>, Erik Ekman <yarrick@kryo.se>
3  *
4  * Permission to use, copy, modify, and distribute this software for any
5  * purpose with or without fee is hereby granted, provided that the above
6  * copyright notice and this permission notice appear in all copies.
7  *
8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15  */
16
17 #include <string.h>
18 #include "common.h"
19 #include "encoding.h"
20
21 int
22 build_hostname(char *buf, size_t buflen, 
23                 const char *data, const size_t datalen, 
24                 const char *topdomain, struct encoder *encoder, int maxlen)
25 {
26         int encsize;
27         size_t space;
28         char *b;
29
30         space = MIN(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         encsize = 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 }