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
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
/*
* The Sleuth Kit
*
* Brian Carrier [carrier <at> sleuthkit [dot] org]
* Copyright (c) 2007 Brian Carrier. All Rights reserved
*
* This software is distributed under the Common Public License 1.0
*/
/** \file tsk_printf.c
* These are printf wrappers that are needed so that we can
* easily print in both Unix and Windows. For Unix, the
* internal UTF-8 representation is kept and a normal printf
* is performed. For Windows, the UTF-8 representation is first
* converted to UTF-16 and then printed
*/
#include "tsk_base_i.h"
#include <stdarg.h>
/** \internal
* Convert the UTF-8 printf arguments to UTF-16 and fill in the
* printf types (%s, %d, etc.)
*
* @param wbuf wide char string to write result to
* @param wlen number of characters in wbuf
* @param msg printf message string
* @param args Arguments to use when filling in message string
* @returns 1 on error and 0 on success
*/
#ifdef TSK_WIN32
static int
tsk_printf_conv(WCHAR * wbuf, int wlen, const char *msg, va_list * args)
{
char *cbuf;
UTF8 *ptr8;
UTF16 *ptr16;
int retVal;
size_t len, clen;
wbuf[0] = '\0';
/* Allocate a UTF-8 buffer and process the printf args */
clen = wlen * 3;
if (NULL == (cbuf = (char *) tsk_malloc(clen))) {
return 1;
}
memset(cbuf, 0, clen);
#ifdef _MSC_VER
vsnprintf_s(cbuf, clen - 1, _TRUNCATE, msg, *args);
#else
vsnprintf(cbuf, clen - 1, msg, *args);
#endif
len = strlen(cbuf);
//Convert to UTF-16
ptr8 = (UTF8 *) cbuf;
ptr16 = (UTF16 *) wbuf;
retVal =
tsk_UTF8toUTF16((const UTF8 **) &ptr8, &ptr8[len + 1], &ptr16,
&ptr16[wlen], TSKlenientConversion);
if (retVal != TSKconversionOK) {
*ptr16 = '\0';
if (tsk_verbose)
tsk_fprintf(stderr,
"tsk_printf_conv: error converting string to UTF-16\n");
}
free(cbuf);
return 0;
}
#endif
/**
* \ingroup baselib
* fprintf wrapper function that takes UTF-8 strings as input
* (on all platforms) and does what is necessary to output
* strings in the correct encoding (UTF-8 on Unix and
* UTF-16 on Windows).
*
* @param fd File to print to
* @param msg printf message
*/
void
tsk_fprintf(FILE * fd, const char *msg, ...)
{
va_list args;
va_start(args, msg);
#ifdef TSK_WIN32
{
WCHAR wbuf[2048];
tsk_printf_conv(wbuf, 2048, msg, &args);
fwprintf(fd, _TSK_T("%s"), wbuf);
}
#else
vfprintf(fd, msg, args);
#endif
va_end(args);
}
/**
* \ingroup baselib
* printf wrapper function that takes UTF-8 strings as input
* (on all platforms) and does what is necessary to output
* strings in the correct encoding (UTF-8 on Unix and
* UTF-16 on Windows).
*
* @param msg printf message
*/
void
tsk_printf(const char *msg, ...)
{
va_list args;
va_start(args, msg);
#ifdef TSK_WIN32
{
WCHAR wbuf[2048];
tsk_printf_conv(wbuf, 2048, msg, &args);
wprintf(_TSK_T("%s"), wbuf);
}
#else
vprintf(msg, args);
#endif
va_end(args);
}
|
