workbench/devs/diskimage/support/aros/asprintf.c

   1 /* Copyright 2007-2012 Fredrik Wikstrom. All rights reserved.
   2 **
   3 ** Redistribution and use in source and binary forms, with or without
   4 ** modification, are permitted provided that the following conditions
   5 ** are met:
   6 **
   7 ** 1. Redistributions of source code must retain the above copyright
   8 **    notice, this list of conditions and the following disclaimer.
   9 **
  10 ** 2. Redistributions in binary form must reproduce the above copyright
  11 **    notice, this list of conditions and the following disclaimer in the
  12 **    documentation and/or other materials provided with the distribution.
  13 **
  14 ** THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS `AS IS'
  15 ** AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  16 ** IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  17 ** ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
  18 ** LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  19 ** CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  20 ** SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  21 ** INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  22 ** CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  23 ** ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  24 ** POSSIBILITY OF SUCH DAMAGE.
  25 */
  26
  27 #include "support.h"
  28 #include <exec/rawfmt.h>
  29 #include <proto/exec.h>
  30 #include <SDI_stdarg.h>
  31
  32 typedef struct {
  33         STRPTR Target;
  34         ULONG TargetSize;
  35 } SNPrintfStream;
  36
  37 SNPrintfStream *SNPrintfPutCh(SNPrintfStream * s, UBYTE c)
  38 {
  39         if (s->TargetSize > 0) {
  40                 if (s->TargetSize > 1) {
  41                         *s->Target++ = c;
  42                 } else {
  43                         *s->Target++ = 0;
  44                 }
  45                 s->TargetSize--;
  46         }
  47
  48         return s;
  49 }
  50
  51 VARARGS68K void SNPrintf (STRPTR buf, LONG len, CONST_STRPTR fmt, ...) {
  52         VA_LIST args;
  53         VA_START(args, fmt);
  54         VSNPrintf(buf, len, fmt, args);
  55         VA_END(args);
  56 }
  57
  58 VARARGS68K STRPTR ASPrintf (CONST_STRPTR fmt, ...) {
  59         VA_LIST args;
  60         STRPTR res;
  61         VA_START(args, fmt);
  62         res = VASPrintf(fmt, args);
  63         VA_END(args);
  64         return res;
  65 }
  66
  67 void VSNPrintf (STRPTR buf, LONG len, CONST_STRPTR fmt, VA_LIST args) {
  68         SNPrintfStream s = { buf, len };
  69         VNewRawDoFmt(fmt, (VOID_FUNC)SNPrintfPutCh, &s, args);
  70 }
  71
  72 STRPTR VASPrintf (CONST_STRPTR fmt, VA_LIST args) {
  73         STRPTR buf;
  74         ULONG len = 0;
  75         VA_LIST copy;
  76
  77         /* We must copy the VA_LIST, as it be altered in place
  78          */
  79         VA_COPY(copy, args);
  80         VNewRawDoFmt(fmt, RAWFMTFUNC_COUNT, &len, args);
  81         VA_END(copy);
  82
  83         buf = AllocVec(len, MEMF_ANY);
  84         if (buf) {
  85                 SNPrintfStream s = { buf, len };
  86                 VNewRawDoFmt(fmt, (VOID_FUNC)SNPrintfPutCh, &s, args);
  87         }
  88         return buf;
  89 }