ucx: src/printf.c@26500fc24058

     1 /*

     2  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.

     3  *

     4  * Copyright 2021 Mike Becker, Olaf Wintermann All rights reserved.

     5  *

     6  * Redistribution and use in source and binary forms, with or without

     7  * modification, are permitted provided that the following conditions are met:

     8  *

     9  *   1. Redistributions of source code must retain the above copyright

    10  *      notice, this list of conditions and the following disclaimer.

    11  *

    12  *   2. Redistributions in binary form must reproduce the above copyright

    13  *      notice, this list of conditions and the following disclaimer in the

    14  *      documentation and/or other materials provided with the distribution.

    15  *

    16  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"

    17  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE

    18  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE

    19  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE

    20  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR

    21  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF

    22  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS

    23  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN

    24  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)

    25  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE

    26  * POSSIBILITY OF SUCH DAMAGE.

    27  */

    29 #include "cx/printf.h"

    31 #include <stdio.h>

    32 #include <string.h>

    34 #ifndef CX_PRINTF_SBO_SIZE

    35 #define CX_PRINTF_SBO_SIZE 512

    36 #endif

    37 unsigned const cx_printf_sbo_size = CX_PRINTF_SBO_SIZE;

    39 int cx_fprintf(

    40         void *stream,

    41         cx_write_func wfc,

    42         char const *fmt,

    43         ...

    44 ) {

    45     int ret;

    46     va_list ap;

    47     va_start(ap, fmt);

    48     ret = cx_vfprintf(stream, wfc, fmt, ap);

    49     va_end(ap);

    50     return ret;

    51 }

    53 int cx_vfprintf(

    54         void *stream,

    55         cx_write_func wfc,

    56         char const *fmt,

    57         va_list ap

    58 ) {

    59     char buf[CX_PRINTF_SBO_SIZE];

    60     va_list ap2;

    61     va_copy(ap2, ap);

    62     int ret = vsnprintf(buf, CX_PRINTF_SBO_SIZE, fmt, ap);

    63     if (ret < 0) {

    64         return ret;

    65     } else if (ret < CX_PRINTF_SBO_SIZE) {

    66         return (int) wfc(buf, 1, ret, stream);

    67     } else {

    68         int len = ret + 1;

    69         char *newbuf = malloc(len);

    70         if (!newbuf) {

    71             return -1;

    72         }

    74         ret = vsnprintf(newbuf, len, fmt, ap2);

    75         if (ret > 0) {

    76             ret = (int) wfc(newbuf, 1, ret, stream);

    77         }

    78         free(newbuf);

    79     }

    80     return ret;

    81 }

    83 cxmutstr cx_asprintf_a(

    84         CxAllocator const *allocator,

    85         char const *fmt,

    86         ...

    87 ) {

    88     va_list ap;

    89     va_start(ap, fmt);

    90     cxmutstr ret = cx_vasprintf_a(allocator, fmt, ap);

    91     va_end(ap);

    92     return ret;

    93 }

    95 cxmutstr cx_vasprintf_a(

    96         CxAllocator const *a,

    97         char const *fmt,

    98         va_list ap

    99 ) {

   100     cxmutstr s;

   101     s.ptr = NULL;

   102     s.length = 0;

   103     char buf[CX_PRINTF_SBO_SIZE];

   104     va_list ap2;

   105     va_copy(ap2, ap);

   106     int ret = vsnprintf(buf, CX_PRINTF_SBO_SIZE, fmt, ap);

   107     if (ret > 0 && ret < CX_PRINTF_SBO_SIZE) {

   108         s.ptr = cxMalloc(a, ret + 1);

   109         if (s.ptr) {

   110             s.length = (size_t) ret;

   111             memcpy(s.ptr, buf, ret);

   112             s.ptr[s.length] = '\0';

   113         }

   114     } else {

   115         int len = ret + 1;

   116         s.ptr = cxMalloc(a, len);

   117         if (s.ptr) {

   118             ret = vsnprintf(s.ptr, len, fmt, ap2);

   119             if (ret < 0) {

   120                 free(s.ptr);

   121                 s.ptr = NULL;

   122             } else {

   123                 s.length = (size_t) ret;

   124             }

   125         }

   126     }

   127     return s;

   128 }

Mercurial > hg > ucx / file revision