|
|
1.1 root 1: /*
2: * Copyright (C) 2011 Michael Brown <[email protected]>.
3: *
4: * This program is free software; you can redistribute it and/or
5: * modify it under the terms of the GNU General Public License as
6: * published by the Free Software Foundation; either version 2 of the
7: * License, or any later version.
8: *
9: * This program is distributed in the hope that it will be useful, but
10: * WITHOUT ANY WARRANTY; without even the implied warranty of
11: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12: * General Public License for more details.
13: *
14: * You should have received a copy of the GNU General Public License
15: * along with this program; if not, write to the Free Software
16: * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
17: */
18:
19: FILE_LICENCE ( GPL2_OR_LATER );
20:
21: #include <stddef.h>
22: #include <stdarg.h>
23: #include <ipxe/vsprintf.h>
24: #include <ipxe/efi/efi_strings.h>
25:
26: /** Context used by efi_vsnprintf() and friends */
27: struct efi_sputc_context {
28: /** printf context */
29: struct printf_context ctx;
30: /** Buffer for formatted string (used by efi_printf_sputc()) */
31: wchar_t *buf;
32: /** Buffer length (used by efi_printf_sputc())
33: *
34: * Note that this is a number of wide characters, not a number
35: * of bytes.
36: */
37: size_t max_wlen;
38: };
39:
40: /**
41: * Write wide character to buffer
42: *
43: * @v ctx Context
44: * @v c Character
45: */
46: static void efi_printf_sputc ( struct printf_context *ctx, unsigned int c ) {
47: struct efi_sputc_context * sctx =
48: container_of ( ctx, struct efi_sputc_context, ctx );
49:
50: if ( ctx->len < sctx->max_wlen )
51: sctx->buf[ctx->len] = c;
52: }
53:
54: /**
55: * Write a formatted string to a wide-character buffer
56: *
57: * @v wbuf Buffer into which to write the string
58: * @v wsize Size of buffer (in wide characters)
59: * @v fmt Format string
60: * @v args Arguments corresponding to the format string
61: * @ret wlen Length of formatted string (in wide characters)
62: *
63: * If the buffer is too small to contain the string, the returned
64: * length is the length that would have been written had enough space
65: * been available.
66: */
67: int efi_vsnprintf ( wchar_t *wbuf, size_t wsize, const char *fmt,
68: va_list args ) {
69: struct efi_sputc_context sctx;
70: size_t wlen;
71: size_t wend;
72:
73: /* Hand off to vcprintf */
74: sctx.ctx.handler = efi_printf_sputc;
75: sctx.buf = wbuf;
76: sctx.max_wlen = wsize;
77: wlen = vcprintf ( &sctx.ctx, fmt, args );
78:
79: /* Add trailing NUL */
80: if ( wsize ) {
81: wend = wsize - 1;
82: if ( wlen < wend )
83: wend = wlen;
84: wbuf[wend] = '\0';
85: }
86:
87: return wlen;
88: }
89:
90: /**
91: * Write a formatted string to a buffer
92: *
93: * @v wbuf Buffer into which to write the string
94: * @v wsize Size of buffer (in wide characters)
95: * @v fmt Format string
96: * @v ... Arguments corresponding to the format string
97: * @ret wlen Length of formatted string (in wide characters)
98: */
99: int efi_snprintf ( wchar_t *wbuf, size_t wsize, const char *fmt, ... ) {
100: va_list args;
101: int i;
102:
103: va_start ( args, fmt );
104: i = efi_vsnprintf ( wbuf, wsize, fmt, args );
105: va_end ( args );
106: return i;
107: }
108:
109: /**
110: * Version of efi_vsnprintf() that accepts a signed buffer size
111: *
112: * @v wbuf Buffer into which to write the string
113: * @v swsize Size of buffer (in wide characters)
114: * @v fmt Format string
115: * @v args Arguments corresponding to the format string
116: * @ret wlen Length of formatted string (in wide characters)
117: */
118: int efi_vssnprintf ( wchar_t *wbuf, ssize_t swsize, const char *fmt,
119: va_list args ) {
120:
121: /* Treat negative buffer size as zero buffer size */
122: if ( swsize < 0 )
123: swsize = 0;
124:
125: /* Hand off to vsnprintf */
126: return efi_vsnprintf ( wbuf, swsize, fmt, args );
127: }
128:
129: /**
130: * Version of efi_vsnprintf() that accepts a signed buffer size
131: *
132: * @v wbuf Buffer into which to write the string
133: * @v swsize Size of buffer (in wide characters)
134: * @v fmt Format string
135: * @v ... Arguments corresponding to the format string
136: * @ret wlen Length of formatted string (in wide characters)
137: */
138: int efi_ssnprintf ( wchar_t *wbuf, ssize_t swsize, const char *fmt, ... ) {
139: va_list args;
140: int len;
141:
142: /* Hand off to vssnprintf */
143: va_start ( args, fmt );
144: len = efi_vssnprintf ( wbuf, swsize, fmt, args );
145: va_end ( args );
146: return len;
147: }
This archive runs on limited infrastructure. Preserving old code on modern bandwidth. Automated agents are requested to crawl responsibly.