ucx: src/printf.c@19211b88cc0f

     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 #define UCX_PRINTF_BUFSIZE 256

    36 int cx_fprintf(void *stream, cx_write_func wfc, char const *fmt, ...) {

    37     int ret;

    38     va_list ap;

    39     va_start(ap, fmt);

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

    41     va_end(ap);

    42     return ret;

    43 }

    45 int cx_vfprintf(void *stream, cx_write_func wfc, char const *fmt, va_list ap) {

    46     char buf[UCX_PRINTF_BUFSIZE];

    47     va_list ap2;

    48     va_copy(ap2, ap);

    49     int ret = vsnprintf(buf, UCX_PRINTF_BUFSIZE, fmt, ap);

    50     if (ret < 0) {

    51         return ret;

    52     } else if (ret < UCX_PRINTF_BUFSIZE) {

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

    54     } else {

    55         int len = ret + 1;

    56         char *newbuf = malloc(len);

    57         if (!newbuf) {

    58             return -1;

    59         }

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

    62         if (ret > 0) {

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

    64         }

    65         free(newbuf);

    66     }

    67     return ret;

    68 }

    70 cxmutstr cx_asprintf_a(CxAllocator *allocator, char const *fmt, ...) {

    71     va_list ap;

    72     cxmutstr ret;

    73     va_start(ap, fmt);

    74     ret = cx_vasprintf_a(allocator, fmt, ap);

    75     va_end(ap);

    76     return ret;

    77 }

    79 cxmutstr cx_vasprintf_a(CxAllocator *a, char const *fmt, va_list ap) {

    80     cxmutstr s;

    81     s.ptr = NULL;

    82     s.length = 0;

    83     char buf[UCX_PRINTF_BUFSIZE];

    84     va_list ap2;

    85     va_copy(ap2, ap);

    86     int ret = vsnprintf(buf, UCX_PRINTF_BUFSIZE, fmt, ap);

    87     if (ret > 0 && ret < UCX_PRINTF_BUFSIZE) {

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

    89         if (s.ptr) {

    90             s.length = (size_t) ret;

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

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

    93         }

    94     } else {

    95         int len = ret + 1;

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

    97         if (s.ptr) {

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

    99             if (ret < 0) {

   100                 free(s.ptr);

   101                 s.ptr = NULL;

   102             } else {

   103                 s.length = (size_t) ret;

   104             }

   105         }

   106     }

   107     return s;

   108 }

Mercurial > hg > ucx / file revision