src/libical/libicals_w32_vsnprintf_replacement.c

   1 #ifdef WIN32
   2 #include "config.h"
   3 #ifndef HAVE_SNPRINTF
   4 #error DO NOT USE libical version of vsnprintf.
   5 #endif
   6 /*
   7  * Revision 12: http://theos.com/~deraadt/snprintf.c
   8  *
   9  * Copyright (c) 1997 Theo de Raadt
  10  *
  11  * Redistribution and use in source and binary forms, with or without
  12  * modification, are permitted provided that the following conditions
  13  * are met:
  14  * 1. Redistributions of source code must retain the above copyright
  15  *    notice, this list of conditions and the following disclaimer.
  16  * 2. Redistributions in binary form must reproduce the above copyright
  17  *    notice, this list of conditions and the following disclaimer in the
  18  *    documentation and/or other materials provided with the distribution.
  19  *
  20  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
  21  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  22  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  23  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
  24  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  25  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  26  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  27  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  28  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  29  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  30  */
  31
  32 #ifndef WIN32
  33 #include <sys/param.h>
  34 #include <sys/mman.h>
  35 #include <unistd.h>
  36 #endif
  37 #include <sys/types.h>
  38 #include <signal.h>
  39 #include <stdio.h>
  40 #include <string.h>
  41 #if __STDC__
  42 #include <stdarg.h>
  43 #include <stdlib.h>
  44 #else
  45 #include <varargs.h>
  46 #endif
  47 #include <setjmp.h>
  48
  49 #ifndef roundup
  50 #define roundup(x, y) ((((x)+((y)-1))/(y))*(y))
  51 #endif
  52
  53 static int pgsize;
  54 static char *curobj;
  55 static sigjmp_buf bail;
  56
  57 #define EXTRABYTES      2       /* XXX: why 2? you don't want to know */
  58
  59 static char *
  60 msetup(str, n)
  61         char *str;
  62         size_t n;
  63 {
  64         char *e;
  65
  66         if (n == 0)
  67                 return NULL;
  68         if (pgsize == 0)
  69                 pgsize = getpagesize();
  70         curobj = (char *)malloc(n + EXTRABYTES + pgsize * 2);
  71         if (curobj == NULL)
  72                 return NULL;
  73         e = curobj + n + EXTRABYTES;
  74         e = (char *)roundup((unsigned long)e, pgsize);
  75         if (mprotect(e, pgsize, PROT_NONE) == -1) {
  76                 free(curobj);
  77                 curobj = NULL;
  78                 return NULL;
  79         }
  80         e = e - n - EXTRABYTES;
  81         *e = '\0';
  82         return (e);
  83 }
  84
  85 static void
  86 mcatch(int i)
  87 {
  88         siglongjmp(bail, 1);
  89 }
  90
  91 static void
  92 mcleanup(str, n, p)
  93         char *str;
  94         size_t n;
  95         char *p;
  96 {
  97         strncpy(str, p, n-1);
  98         str[n-1] = '\0';
  99         if (mprotect((caddr_t)(p + n + EXTRABYTES), pgsize,
 100             PROT_READ|PROT_WRITE|PROT_EXEC) == -1)
 101                 mprotect((caddr_t)(p + n + EXTRABYTES), pgsize,
 102                     PROT_READ|PROT_WRITE);
 103         free(curobj);
 104 }
 105
 106 int
 107 #if __STDC__
 108 vsnprintf(char *str, size_t n, char const *fmt, va_list ap)
 109 #else
 110 vsnprintf(str, n, fmt, ap)
 111         char *str;
 112         size_t n;
 113         char *fmt;
 114         char *ap;
 115 #endif
 116 {
 117         struct sigaction osa, nsa;
 118         char *p;
 119         int ret = n + 1;        /* if we bail, indicated we overflowed */
 120
 121         memset(&nsa, 0, sizeof nsa);
 122         nsa.sa_handler = mcatch;
 123         sigemptyset(&nsa.sa_mask);
 124
 125         p = msetup(str, n);
 126         if (p == NULL) {
 127                 *str = '\0';
 128                 return 0;
 129         }
 130         if (sigsetjmp(bail, 1) == 0) {
 131                 if (sigaction(SIGSEGV, &nsa, &osa) == -1) {
 132                         mcleanup(str, n, p);
 133                         return (0);
 134                 }
 135                 ret = vsprintf(p, fmt, ap);
 136         }
 137         mcleanup(str, n, p);
 138         (void) sigaction(SIGSEGV, &osa, NULL);
 139         return (ret);
 140 }
 141
 142 int
 143 #if __STDC__
 144 snprintf(char *str, size_t n, char const *fmt, ...)
 145 #else
 146 snprintf(str, n, fmt, va_alist)
 147         char *str;
 148         size_t n;
 149         char *fmt;
 150         va_dcl
 151 #endif
 152 {
 153         va_list ap;
 154 #if __STDC__
 155         va_start(ap, fmt);
 156 #else
 157         va_start(ap);
 158 #endif
 159
 160         return (vsnprintf(str, n, fmt, ap));
 161         va_end(ap);
 162 }
 163
 164
 165 #else
 166
 167 /* ANSI C forbids an empty source file... */
 168
 169 static void vsnprintf_dummy_func() {
 170    vsnprintf_dummy_func();
 171 }
 172
 173 #endif