Line data Source code
1 : /* FreeTDS - Library of routines accessing Sybase and Microsoft databases
2 : * Copyright (C) 1998, 1999, 2000, 2001, 2002, 2003, 2004 Brian Bruns
3 : * Copyright (C) 2005 Frediano Ziglio
4 : *
5 : * This library is free software; you can redistribute it and/or
6 : * modify it under the terms of the GNU Library General Public
7 : * License as published by the Free Software Foundation; either
8 : * version 2 of the License, or (at your option) any later version.
9 : *
10 : * This library is distributed in the hope that it will be useful,
11 : * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 : * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 : * Library General Public License for more details.
14 : *
15 : * You should have received a copy of the GNU Library General Public
16 : * License along with this library; if not, write to the
17 : * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
18 : * Boston, MA 02111-1307, USA.
19 : */
20 :
21 : #include <config.h>
22 :
23 : #include <stdarg.h>
24 : #include <stdio.h>
25 : #include <assert.h>
26 :
27 : #if HAVE_ERRNO_H
28 : #include <errno.h>
29 : #endif /* HAVE_ERRNO_H */
30 :
31 : #if HAVE_STDLIB_H
32 : #include <stdlib.h>
33 : #endif /* HAVE_STDLIB_H */
34 :
35 : #if HAVE_STRING_H
36 : #include <string.h>
37 : #endif /* HAVE_STRING_H */
38 :
39 : #if HAVE_UNISTD_H
40 : #include <unistd.h>
41 : #endif /* HAVE_UNISTD_H */
42 :
43 : #include <freetds/tds.h>
44 : #include <freetds/iconv.h>
45 : #include <freetds/bytes.h>
46 : #include <freetds/stream.h>
47 : #include <freetds/checks.h>
48 :
49 : #if TDS_ADDITIONAL_SPACE < 8
50 : #error Not supported
51 : #endif
52 :
53 : /**
54 : * \addtogroup network
55 : * @{
56 : */
57 :
58 : /*
59 : * CRE 01262002 making buf a void * means we can put any type without casting
60 : * much like read() and memcpy()
61 : */
62 : int
63 213850 : tds_put_n(TDSSOCKET * tds, const void *buf, size_t n)
64 : {
65 : size_t left;
66 213850 : const unsigned char *bufp = (const unsigned char *) buf;
67 :
68 644795 : for (; n;) {
69 217095 : if (tds->out_pos >= tds->out_buf_max) {
70 2775 : tds_write_packet(tds, 0x0);
71 2775 : continue;
72 : }
73 214320 : left = tds->out_buf_max - tds->out_pos;
74 214320 : if (left > n)
75 211560 : left = n;
76 214320 : if (bufp) {
77 129114 : memcpy(tds->out_buf + tds->out_pos, bufp, left);
78 129114 : bufp += left;
79 : } else {
80 85206 : memset(tds->out_buf + tds->out_pos, 0, left);
81 : }
82 214320 : tds->out_pos += (unsigned int)left;
83 214320 : n -= left;
84 : }
85 213850 : return 0;
86 : }
87 :
88 : /**
89 : * Output a string to wire
90 : * automatic translate string to unicode if needed
91 : * \return bytes written to wire
92 : * \param tds state information for the socket and the TDS protocol
93 : * \param s string to write
94 : * \param len length of string in characters, or -1 for null terminated
95 : */
96 : int
97 96079 : tds_put_string(TDSSOCKET * tds, const char *s, int len)
98 : {
99 : int res;
100 : TDSSTATICINSTREAM r;
101 : TDSDATAOUTSTREAM w;
102 : enum TDS_ICONV_ENTRY iconv_entry;
103 :
104 96079 : if (len < 0) {
105 : TDS_ENCODING *client;
106 24898 : client = &tds->conn->char_convs[client2ucs2]->from.charset;
107 :
108 24898 : if (client->min_bytes_per_char == 1) { /* ascii or UTF-8 */
109 24898 : len = (int)strlen(s);
110 0 : } else if (client->min_bytes_per_char == 2) { /* UCS-2 or variant */
111 : const char *p = s;
112 :
113 0 : while (p[0] || p[1])
114 0 : p += 2;
115 0 : len = (int)(p - s);
116 :
117 0 : } else if (client->min_bytes_per_char == 4) { /* UCS-4 or variant */
118 : const char *p = s;
119 :
120 0 : while (p[0] || p[1] || p[2] || p[3])
121 0 : p += 4;
122 0 : len = (int)(p - s);
123 : } else {
124 0 : assert(client->min_bytes_per_char < 3); /* FIXME */
125 : }
126 : }
127 :
128 96079 : assert(len >= 0);
129 :
130 : /* valid test only if client and server share a character set. TODO conversions for Sybase */
131 96079 : if (IS_TDS7_PLUS(tds->conn)) {
132 : iconv_entry = client2ucs2;
133 15446 : } else if (IS_TDS50(tds->conn)) {
134 : iconv_entry = client2server_chardata;
135 : } else {
136 0 : tds_put_n(tds, s, len);
137 0 : return len;
138 : }
139 :
140 96079 : tds_staticin_stream_init(&r, s, len);
141 96079 : tds_dataout_stream_init(&w, tds);
142 :
143 96079 : res = tds_convert_stream(tds, tds->conn->char_convs[iconv_entry], to_server, &r.stream, &w.stream);
144 96079 : return (int) w.written;
145 : }
146 :
147 : int
148 9378 : tds_put_int8(TDSSOCKET * tds, TDS_INT8 i)
149 : {
150 : TDS_UCHAR *p;
151 :
152 9378 : if (tds->out_pos >= tds->out_buf_max)
153 10 : tds_write_packet(tds, 0x0);
154 :
155 9378 : p = &tds->out_buf[tds->out_pos];
156 9378 : TDS_PUT_UA4LE(p, (TDS_UINT) i);
157 9378 : TDS_PUT_UA4LE(p+4, (TDS_UINT) (i >> 32));
158 9378 : tds->out_pos += 8;
159 9378 : return 0;
160 : }
161 :
162 : int
163 125517 : tds_put_int(TDSSOCKET * tds, TDS_INT i)
164 : {
165 : TDS_UCHAR *p;
166 :
167 125517 : if (tds->out_pos >= tds->out_buf_max)
168 0 : tds_write_packet(tds, 0x0);
169 :
170 125517 : p = &tds->out_buf[tds->out_pos];
171 125517 : TDS_PUT_UA4LE(p, i);
172 125517 : tds->out_pos += 4;
173 125517 : return 0;
174 : }
175 :
176 : int
177 153964 : tds_put_smallint(TDSSOCKET * tds, TDS_SMALLINT si)
178 : {
179 : TDS_UCHAR *p;
180 :
181 153964 : if (tds->out_pos >= tds->out_buf_max)
182 0 : tds_write_packet(tds, 0x0);
183 :
184 153964 : p = &tds->out_buf[tds->out_pos];
185 153964 : TDS_PUT_UA2LE(p, si);
186 153964 : tds->out_pos += 2;
187 153964 : return 0;
188 : }
189 :
190 : int
191 251999 : tds_put_byte(TDSSOCKET * tds, unsigned char c)
192 : {
193 251999 : if (tds->out_pos >= tds->out_buf_max)
194 0 : tds_write_packet(tds, 0x0);
195 251999 : tds->out_buf[tds->out_pos++] = c;
196 251999 : return 0;
197 : }
198 :
199 : int
200 5595 : tds_init_write_buf(TDSSOCKET * tds)
201 : {
202 5595 : TDS_MARK_UNDEFINED(tds->out_buf, tds->out_buf_max);
203 5595 : tds->out_pos = 8;
204 5595 : return 0;
205 : }
206 :
207 : /**
208 : * Flush packet to server
209 : * @return TDS_FAIL or TDS_SUCCESS
210 : */
211 : TDSRET
212 89256 : tds_flush_packet(TDSSOCKET * tds)
213 : {
214 89256 : TDSRET result = TDS_FAIL;
215 :
216 : /* GW added check for tds->s */
217 89256 : if (!IS_TDSDEAD(tds)) {
218 89215 : if (tds->out_pos > tds->out_buf_max)
219 21 : TDS_PROPAGATE(tds_write_packet(tds, 0x00));
220 89215 : result = tds_write_packet(tds, 0x01);
221 : }
222 : return result;
223 : }
224 :
225 : /** @} */
|
