summaryrefslogtreecommitdiffstats
path: root/src/lib/stdio.C
blob: d61b48cf9387211471a3e7de5c73acef83ebb9ab (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
/* IBM_PROLOG_BEGIN_TAG                                                   */
/* This is an automatically generated prolog.                             */
/*                                                                        */
/* $Source: src/lib/stdio.C $                                             */
/*                                                                        */
/* OpenPOWER HostBoot Project                                             */
/*                                                                        */
/* COPYRIGHT International Business Machines Corp. 2011,2014              */
/*                                                                        */
/* Licensed under the Apache License, Version 2.0 (the "License");        */
/* you may not use this file except in compliance with the License.       */
/* You may obtain a copy of the License at                                */
/*                                                                        */
/*     http://www.apache.org/licenses/LICENSE-2.0                         */
/*                                                                        */
/* Unless required by applicable law or agreed to in writing, software    */
/* distributed under the License is distributed on an "AS IS" BASIS,      */
/* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or        */
/* implied. See the License for the specific language governing           */
/* permissions and limitations under the License.                         */
/*                                                                        */
/* IBM_PROLOG_END_TAG                                                     */
#include <stdint.h>
#include <stdio.h>
#include <util/sprintf.H>

class SprintfBuffer : public Util::ConsoleBufferInterface
{
    public:
        int putc(int c)
        {
            if ('\b' == c)
            {
                iv_pos--;
            }
            else if (iv_pos < iv_size)
            {
                iv_buffer[iv_pos++] = c;
            }
            else
            {
                iv_pos++;
            }
            return c;
        }

        explicit SprintfBuffer(char* buf, size_t size = UINT64_MAX) :
            iv_pos(0), iv_size(size), iv_buffer(buf) {};

        size_t operator()(int c) { return putc(c); }

    private:
        size_t iv_pos;
        size_t iv_size;
        char * iv_buffer;
};

int sprintf(char *str, const char * format, ...)
{
    using Util::vasprintf;

    va_list args;
    va_start(args, format);

    SprintfBuffer console(str);
    size_t count = vasprintf(console, format, args);

    va_end(args);
    console.putc('\0');
    return count;
}

int snprintf(char *str, size_t size, const char * format, ...)
{
    using Util::vasprintf;

    va_list args;
    va_start(args, format);

    SprintfBuffer console(str, size);
    size_t count = vasprintf(console, format, args);

    va_end(args);
    console.putc('\0');
    return count;
}

int vsprintf(char *str, const char * format, va_list args)
{
    using Util::vasprintf;

    SprintfBuffer console(str);
    size_t count = vasprintf(console, format, args);

    console.putc('\0');
    return count;
}

int vsnprintf(char *str, size_t size, const char * format, va_list args)
{
    SprintfBuffer console(str, size);
    size_t count = vasprintf(console, format, args);

    console.putc('\0');
    return count;
}
OpenPOWER on IntegriCloud