git.meshlink.io Git - meshlink/blob - src/utils.c

   1 /*
   2     utils.c -- gathering of some stupid small functions
   3     Copyright (C) 1999-2005 Ivo Timmermans
   4                   2000-2009 Guus Sliepen <guus@tinc-vpn.org>
   5
   6     This program is free software; you can redistribute it and/or modify
   7     it under the terms of the GNU General Public License as published by
   8     the Free Software Foundation; either version 2 of the License, or
   9     (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 General Public License for more details.
  15
  16     You should have received a copy of the GNU General Public License along
  17     with this program; if not, write to the Free Software Foundation, Inc.,
  18     51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  19 */
  20
  21 #include "system.h"
  22
  23 #include "../src/logger.h"
  24 #include "utils.h"
  25
  26 static const char hexadecimals[] = "0123456789ABCDEF";
  27 static const char base64imals[] = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
  28
  29 static int charhex2bin(char c) {
  30         if(isdigit(c))
  31                 return c - '0';
  32         else
  33                 return toupper(c) - 'A' + 10;
  34 }
  35
  36 static int charb64decode(char c) {
  37         if(c >= 'a')
  38                 return c - 'a' + 26;
  39         else if(c >= 'A')
  40                 return c - 'A';
  41         else if(c >= '0')
  42                 return c - '0' + 52;
  43         else if(c == '+')
  44                 return 62;
  45         else
  46                 return 63;
  47 }
  48
  49 int hex2bin(const char *src, char *dst, int length) {
  50         int i;
  51         for(i = 0; i < length && isxdigit(src[i * 2]) && isxdigit(src[i * 2 + 1]); i++)
  52                 dst[i] = charhex2bin(src[i * 2]) * 16 + charhex2bin(src[i * 2 + 1]);
  53         return i;
  54 }
  55
  56 int bin2hex(const char *src, char *dst, int length) {
  57         for(int i = length - 1; i >= 0; i--) {
  58                 dst[i * 2 + 1] = hexadecimals[(unsigned char) src[i] & 15];
  59                 dst[i * 2] = hexadecimals[(unsigned char) src[i] >> 4];
  60         }
  61         dst[length * 2] = 0;
  62         return length * 2;
  63 }
  64
  65 int b64decode(const char *src, char *dst, int length) {
  66         int i;
  67         uint32_t triplet = 0;
  68         unsigned char *udst = (unsigned char *)dst;
  69
  70         for(i = 0; i < length / 3 * 4 && src[i]; i++) {
  71                 triplet |= charb64decode(src[i]) << (6 * (i & 3));
  72                 if((i & 3) == 3) {
  73                         udst[0] = triplet & 0xff; triplet >>= 8;
  74                         udst[1] = triplet & 0xff; triplet >>= 8;
  75                         udst[2] = triplet;
  76                         triplet = 0;
  77                         udst += 3;
  78                 }
  79         }
  80         if((i & 3) == 3) {
  81                 udst[0] = triplet & 0xff; triplet >>= 8;
  82                 udst[1] = triplet & 0xff;
  83                 return i / 4 * 3 + 2;
  84         } else if((i & 3) == 2) {
  85                 udst[0] = triplet & 0xff;
  86                 return i / 4 * 3 + 1;
  87         } else {
  88                 return i / 4 * 3;
  89         }
  90 }
  91
  92 int b64encode(const char *src, char *dst, int length) {
  93         uint32_t triplet;
  94         const unsigned char *usrc = (unsigned char *)src;
  95         int si = length / 3 * 3;
  96         int di = length / 3 * 4;
  97
  98         switch(length % 3) {
  99                 case 2:
 100                         triplet = usrc[si] | usrc[si + 1] << 8;
 101                         dst[di] = base64imals[triplet & 63]; triplet >>= 6;
 102                         dst[di + 1] = base64imals[triplet & 63]; triplet >>= 6;
 103                         dst[di + 2] = base64imals[triplet];
 104                         dst[di + 3] = 0;
 105                         length = di + 2;
 106                         break;
 107                 case 1:
 108                         triplet = usrc[si];
 109                         dst[di] = base64imals[triplet & 63]; triplet >>= 6;
 110                         dst[di + 1] = base64imals[triplet];
 111                         dst[di + 2] = 0;
 112                         length = di + 1;
 113                         break;
 114                 default:
 115                         dst[di] = 0;
 116                         length = di;
 117                         break;
 118         }
 119
 120         while(si > 0) {
 121                 di -= 4;
 122                 si -= 3;
 123                 triplet = usrc[si] | usrc[si + 1] << 8 | usrc[si + 2] << 16;
 124                 dst[di] = base64imals[triplet & 63]; triplet >>= 6;
 125                 dst[di + 1] = base64imals[triplet & 63]; triplet >>= 6;
 126                 dst[di + 2] = base64imals[triplet & 63]; triplet >>= 6;
 127                 dst[di + 3] = base64imals[triplet];
 128         }
 129
 130         return length;
 131 }
 132
 133 #if defined(HAVE_MINGW) || defined(HAVE_CYGWIN)
 134 #ifdef HAVE_CYGWIN
 135 #include <w32api/windows.h>
 136 #endif
 137
 138 const char *winerror(int err) {
 139         static char buf[1024], *ptr;
 140
 141         ptr = buf + sprintf(buf, "(%d) ", err);
 142
 143         if (!FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS,
 144                 NULL, err, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), ptr, sizeof(buf) - (ptr - buf), NULL)) {
 145                 strncpy(buf, "(unable to format errormessage)", sizeof(buf));
 146         };
 147
 148         if((ptr = strchr(buf, '\r')))
 149                 *ptr = '\0';
 150
 151         return buf;
 152 }
 153 #endif
 154
 155 unsigned int bitfield_to_int(const void *bitfield, size_t size) {
 156         unsigned int value = 0;
 157         if(size > sizeof value)
 158                 size = sizeof value;
 159         memcpy(&value, bitfield, size);
 160         return value;
 161 }