c79989b51701888cf7a2bb61921c9e78d2c4349e
[samba.git] / lib / util / dprintf.c
1 /* 
2    Unix SMB/CIFS implementation.
3    display print functions
4    Copyright (C) Andrew Tridgell 2001
5    Copyright (C) Jelmer Vernooij 2007
6    
7    This program is free software; you can redistribute it and/or modify
8    it under the terms of the GNU General Public License as published by
9    the Free Software Foundation; either version 3 of the License, or
10    (at your option) any later version.
11    
12    This program is distributed in the hope that it will be useful,
13    but WITHOUT ANY WARRANTY; without even the implied warranty of
14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15    GNU General Public License for more details.
16    
17    You should have received a copy of the GNU General Public License
18    along with this program.  If not, see <http://www.gnu.org/licenses/>.
19 */
20
21
22 /*
23   this module provides functions for printing internal strings in the 
24   "display charset".
25   
26   This charset may be quite different from the chosen unix charset.
27
28   Eventually these functions will need to take care of column count constraints
29
30   The d_ prefix on print functions in Samba refers to the display character set
31   conversion
32 */
33
34 #include "includes.h"
35 #include "system/locale.h"
36
37 static smb_iconv_t display_cd = (smb_iconv_t)-1;
38
39 static int d_vfprintf(FILE *f, const char *format, va_list ap) 
40 {
41         char *p, *p2;
42         int ret;
43         size_t clen;
44         bool cret;
45         va_list ap2;
46
47         /* If there's nothing to convert, take a shortcut */
48         if (display_cd == (smb_iconv_t)-1) {
49                 return vfprintf(f, format, ap);
50         }
51
52         va_copy(ap2, ap);
53         ret = vasprintf(&p, format, ap2);
54         va_end(ap2);
55
56         if (ret <= 0) return ret;
57
58         cret = convert_string_talloc(NULL, CH_UTF8, CH_DISPLAY, p, ret, (void **)&p2, &clen);
59         if (!cret) {
60                 /* the string can't be converted - do the best we can,
61                    filling in non-printing chars with '?' */
62                 int i;
63                 for (i=0;i<ret;i++) {
64                         if (isprint(p[i]) || isspace(p[i])) {
65                                 fwrite(p+i, 1, 1, f);
66                         } else {
67                                 fwrite("?", 1, 1, f);
68                         }
69                 }
70                 SAFE_FREE(p);
71                 return ret;
72         }
73
74         /* good, its converted OK */
75         SAFE_FREE(p);
76         ret = fwrite(p2, 1, clen, f);
77         talloc_free(p2);
78
79         return ret;
80 }
81
82
83 _PUBLIC_ int d_fprintf(FILE *f, const char *format, ...) 
84 {
85         int ret;
86         va_list ap;
87
88         va_start(ap, format);
89         ret = d_vfprintf(f, format, ap);
90         va_end(ap);
91
92         return ret;
93 }
94
95 static FILE *outfile;
96
97 _PUBLIC_  int d_printf(const char *format, ...)
98 {
99         int ret;
100        va_list ap;
101        
102        if (!outfile) outfile = stdout;
103        
104        va_start(ap, format);
105        ret = d_vfprintf(outfile, format, ap);
106        va_end(ap);
107        
108        return ret;
109 }
110
111 /* interactive programs need a way of tell d_*() to write to stderr instead
112    of stdout */
113 void display_set_stderr(void)
114 {
115         outfile = stderr;
116 }