git.mutantstargoat.com Git - raydungeon/blob - libs/goat3d/src/util.c

   1 /*
   2 goat3d - 3D scene, and animation file format library.
   3 Copyright (C) 2013-2023  John Tsiombikas <nuclear@member.fsf.org>
   4
   5 This program is free software: you can redistribute it and/or modify
   6 it under the terms of the GNU Lesser General Public License as published by
   7 the Free Software Foundation, either version 3 of the License, or
   8 (at your option) any later version.
   9
  10 This program is distributed in the hope that it will be useful,
  11 but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 GNU Lesser General Public License for more details.
  14
  15 You should have received a copy of the GNU Lesser General Public License
  16 along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17 */
  18 #include <stdlib.h>
  19 #include <string.h>
  20 #include "util.h"
  21
  22 static int b64bits(int c);
  23
  24 int calc_b64_size(const char *s)
  25 {
  26         int len = strlen(s);
  27         const char *end = s + len;
  28         while(end > s && *--end == '=') len--;
  29         return len * 3 / 4;
  30 }
  31
  32
  33 GOAT3DAPI void *goat3d_b64decode(const char *str, void *buf, int *bufsz)
  34 {
  35         unsigned char *dest, *end;
  36         unsigned char acc;
  37         int bits, sz;
  38         unsigned int gidx;
  39
  40         if(buf) {
  41                 sz = *bufsz;
  42         } else {
  43                 sz = calc_b64_size(str);
  44                 if(!(buf = malloc(sz))) {
  45                         return 0;
  46                 }
  47                 if(bufsz) *bufsz = sz;
  48         }
  49         dest = buf;
  50         end = (unsigned char*)buf + sz;
  51
  52         sz = 0;
  53         gidx = 0;
  54         acc = 0;
  55         while(*str) {
  56                 if((bits = b64bits(*str++)) == -1) {
  57                         continue;
  58                 }
  59
  60                 switch(gidx++ & 3) {
  61                 case 0:
  62                         acc = bits << 2;
  63                         break;
  64                 case 1:
  65                         if(dest < end) *dest = acc | (bits >> 4);
  66                         dest++;
  67                         acc = bits << 4;
  68                         break;
  69                 case 2:
  70                         if(dest < end) *dest = acc | (bits >> 2);
  71                         dest++;
  72                         acc = bits << 6;
  73                         break;
  74                 case 3:
  75                         if(dest < end) *dest = acc | bits;
  76                         dest++;
  77                 default:
  78                         break;
  79                 }
  80         }
  81
  82         if(gidx & 3) {
  83                 if(dest < end) *dest = acc;
  84                 dest++;
  85         }
  86
  87         if(bufsz) *bufsz = dest - (unsigned char*)buf;
  88         return buf;
  89 }
  90
  91 static int b64bits(int c)
  92 {
  93         if(c >= 'A' && c <= 'Z') {
  94                 return c - 'A';
  95         }
  96         if(c >= 'a' && c <= 'z') {
  97                 return c - 'a' + 26;
  98         }
  99         if(c >= '0' && c <= '9') {
 100                 return c - '0' + 52;
 101         }
 102         if(c == '+') return 62;
 103         if(c == '/') return 63;
 104
 105         return -1;
 106 }
 107
 108 GOAT3DAPI void goat3d_bswap32(void *buf, int count)
 109 {
 110         int i;
 111         register uint32_t x;
 112         uint32_t *ptr = buf;
 113
 114         for(i=0; i<count; i++) {
 115                 x = *ptr;
 116                 *ptr++ = (x >> 24) | (x << 24) | ((x >> 8) & 0xff00) | ((x << 8) & 0xff0000);
 117         }
 118 }