UEFITool/common/ustring.cpp

/* ustring.cpp
 
 Copyright (c) 2016, Nikolaj Schlej. All rights reserved.
 This program and the accompanying materials
 are licensed and made available under the terms and conditions of the BSD License
 which accompanies this distribution.  The full text of the license may be found at
 http://opensource.org/licenses/bsd-license.php
 
 THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
 WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
 */

#include "ustring.h"
#include <cstddef>
#include <cstdint>
#include <stdarg.h>
#include "printf/printf.c"

static bool cStyleHexEnabled = false;

void putchar_(char c) {}

void setCStyleHexView(const bool enable)
{
    cStyleHexEnabled = enable;
}

#if defined(QT_CORE_LIB)

typedef struct {
    char flag_cstyle_Xh;
    UString* string;
} qstring_opt_t;

void qstring_out(char c, void* extra_arg)
{
    ((qstring_opt_t*)extra_arg)->string->append(c);
}

UString usprintf(const char* fmt, ...)
{
    UString msg;
    qstring_opt_t opt = { cStyleHexEnabled ? '\1' : '\0', &msg };

    va_list vl;
    va_start(vl, fmt);

    int n = vofctprintf(qstring_out, &opt, fmt, vl);

    va_end(vl);
    return msg;
}

UString urepeated(char c, int len)
{
    return UString(len, c);
}
#else
/* Give WATCOM C/C++, MSVC some latitude for their non-support of vsnprintf */
#if defined(__WATCOMC__) || defined(_MSC_VER)
#define exvsnprintf(r,b,n,f,a) {r = _vsnprintf (b,n,f,a);}
#else
#ifdef BSTRLIB_NOVSNP
/* This is just a hack.  If you are using a system without a vsnprintf, it is
 not recommended that bformat be used at all. */
#define exvsnprintf(r,b,n,f,a) {vsprintf (b,f,a); r = -1;}
#define START_VSNBUFF (256)
#else

#if defined (__GNUC__) && !defined (__PPC__) && !defined(__WIN32__)
/* Something is making gcc complain about this prototype not being here, so
 I've just gone ahead and put it in. */
extern "C" {
extern int vsnprintf(char *buf, size_t count, const char *format, va_list arg);
}
#endif

#define exvsnprintf(r,b,n,f,a) {r = vsnprintf (b,n,f,a);}
#endif
#endif

#ifndef START_VSNBUFF
#define START_VSNBUFF (16)
#endif

UString usprintf(const char* fmt, ...)
{
    UString msg;
    bstring b;
    va_list arglist;
    int r, n;
    
    if (fmt == NULL) {
        msg = "<NULL>";
    }
    else {
        
        if ((b = bfromcstr("")) == NULL) {
            msg = "<NULL>";
        }
        else {
            if ((n = (int)(2 * (strlen)(fmt))) < START_VSNBUFF) n = START_VSNBUFF;
            for (;;) {
                if (BSTR_OK != balloc(b, n + 2)) {
                    b = bformat("<NULL>");
                    break;
                }
                
                va_start(arglist, fmt);
                *(char*)b->data = cStyleHexEnabled ? '\1' : '\0';
                r = vosnprintf_((char*)b->data, n + 1, fmt, arglist);
                va_end(arglist);
                
                b->data[n] = '\0';
                b->slen = (int)(strlen)((char *)b->data);
                
                if (b->slen < n) break;
                if (r > n) n = r; else n += n;
            }
            msg = *b;
            bdestroy(b);
        }
    }
    
    return msg;
}

UString urepeated(char c, int len)
{
    return UString(c, len);
}
#endif

UString uFromUcs2(const char* str, size_t max_len) 
{
    // Naive implementation assuming that only ASCII LE part of UCS2 is used, str may not be aligned.
    UString msg;
    const char *str8 = str;
    size_t rest = (max_len == 0) ? SIZE_MAX : max_len;
    while (str8[0] && rest) {
        msg += str8[0];
        str8 += 2;
        rest--;
    }
    return msg;
}