← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to mysys/mf_iocache2.c

  • Committer: Monty Taylor
  • Date: 2008-10-09 22:38:27 UTC
  • mto: This revision was merged to the branch mainline in revision 497.
  • Revision ID: monty@inaugust.com-20081009223827-bc9gvpiplsmvpwyq
Moved test() to its own file.
Made a new function to possibly replace int10_to_str.

Show diffs side-by-side

added added

removed removed

Lines of Context:
mysys/mf_iocache2.c
 
1
/* Copyright (C) 2000 MySQL AB
 
2
 
 
3
   This program is free software; you can redistribute it and/or modify
 
4
   it under the terms of the GNU General Public License as published by
 
5
   the Free Software Foundation; version 2 of the License.
 
6
 
 
7
   This program is distributed in the hope that it will be useful,
 
8
   but WITHOUT ANY WARRANTY; without even the implied warranty of
 
9
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
10
   GNU General Public License for more details.
 
11
 
 
12
   You should have received a copy of the GNU General Public License
 
13
   along with this program; if not, write to the Free Software
 
14
   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
 
15
 
 
16
/*
 
17
  More functions to be used with IO_CACHE files
 
18
*/
 
19
 
 
20
#include "mysys_priv.h"
 
21
#include <mystrings/m_string.h>
 
22
#include <stdarg.h>
 
23
#include <mystrings/m_ctype.h>
 
24
#include <mysys/iocache.h>
 
25
 
 
26
/*
 
27
  Copy contents of an IO_CACHE to a file.
 
28
 
 
29
  SYNOPSIS
 
30
    my_b_copy_to_file()
 
31
    cache  IO_CACHE to copy from
 
32
    file   File to copy to
 
33
 
 
34
  DESCRIPTION
 
35
    Copy the contents of the cache to the file. The cache will be
 
36
    re-inited to a read cache and will read from the beginning of the
 
37
    cache.
 
38
 
 
39
    If a failure to write fully occurs, the cache is only copied
 
40
    partially.
 
41
 
 
42
  TODO
 
43
    Make this function solid by handling partial reads from the cache
 
44
    in a correct manner: it should be atomic.
 
45
 
 
46
  RETURN VALUE
 
47
    0  All OK
 
48
    1  An error occured
 
49
*/
 
50
int
 
51
my_b_copy_to_file(IO_CACHE *cache, FILE *file)
 
52
{
 
53
  size_t bytes_in_cache;
 
54
 
 
55
  /* Reinit the cache to read from the beginning of the cache */
 
56
  if (reinit_io_cache(cache, READ_CACHE, 0L, false, false))
 
57
    return(1);
 
58
  bytes_in_cache= my_b_bytes_in_cache(cache);
 
59
  do
 
60
  {
 
61
    if (my_fwrite(file, cache->read_pos, bytes_in_cache,
 
62
                  MYF(MY_WME | MY_NABP)) == (size_t) -1)
 
63
      return(1);
 
64
    cache->read_pos= cache->read_end;
 
65
  } while ((bytes_in_cache= my_b_fill(cache)));
 
66
  return(0);
 
67
}
 
68
 
 
69
 
 
70
my_off_t my_b_append_tell(IO_CACHE* info)
 
71
{
 
72
  /*
 
73
    Prevent optimizer from putting res in a register when debugging
 
74
    we need this to be able to see the value of res when the assert fails
 
75
  */
 
76
  volatile my_off_t res; 
 
77
 
 
78
  /*
 
79
    We need to lock the append buffer mutex to keep flush_io_cache()
 
80
    from messing with the variables that we need in order to provide the
 
81
    answer to the question.
 
82
  */
 
83
  pthread_mutex_lock(&info->append_buffer_lock);
 
84
  res = info->end_of_file + (info->write_pos-info->append_read_pos);
 
85
  pthread_mutex_unlock(&info->append_buffer_lock);
 
86
  return res;
 
87
}
 
88
 
 
89
my_off_t my_b_safe_tell(IO_CACHE *info)
 
90
{
 
91
  if (unlikely(info->type == SEQ_READ_APPEND))
 
92
    return my_b_append_tell(info);
 
93
  return my_b_tell(info);
 
94
}
 
95
 
 
96
/*
 
97
  Make next read happen at the given position
 
98
  For write cache, make next write happen at the given position
 
99
*/
 
100
 
 
101
void my_b_seek(IO_CACHE *info,my_off_t pos)
 
102
{
 
103
  my_off_t offset;
 
104
 
 
105
  /*
 
106
    TODO:
 
107
       Verify that it is OK to do seek in the non-append
 
108
       area in SEQ_READ_APPEND cache
 
109
     a) see if this always works
 
110
     b) see if there is a better way to make it work
 
111
  */
 
112
  if (info->type == SEQ_READ_APPEND)
 
113
    flush_io_cache(info);
 
114
 
 
115
  offset=(pos - info->pos_in_file);
 
116
 
 
117
  if (info->type == READ_CACHE || info->type == SEQ_READ_APPEND)
 
118
  {
 
119
    /* TODO: explain why this works if pos < info->pos_in_file */
 
120
    if ((uint64_t) offset < (uint64_t) (info->read_end - info->buffer))
 
121
    {
 
122
      /* The read is in the current buffer; Reuse it */
 
123
      info->read_pos = info->buffer + offset;
 
124
      return;
 
125
    }
 
126
    else
 
127
    {
 
128
      /* Force a new read on next my_b_read */
 
129
      info->read_pos=info->read_end=info->buffer;
 
130
    }
 
131
  }
 
132
  else if (info->type == WRITE_CACHE)
 
133
  {
 
134
    /* If write is in current buffer, reuse it */
 
135
    if ((uint64_t) offset <
 
136
        (uint64_t) (info->write_end - info->write_buffer))
 
137
    {
 
138
      info->write_pos = info->write_buffer + offset;
 
139
      return;
 
140
    }
 
141
    flush_io_cache(info);
 
142
    /* Correct buffer end so that we write in increments of IO_SIZE */
 
143
    info->write_end=(info->write_buffer+info->buffer_length-
 
144
                     (pos & (IO_SIZE-1)));
 
145
  }
 
146
  info->pos_in_file=pos;
 
147
  info->seek_not_done=1;
 
148
  return;
 
149
}
 
150
 
 
151
 
 
152
/*
 
153
  Fill buffer of the cache.
 
154
 
 
155
  NOTES
 
156
    This assumes that you have already used all characters in the CACHE,
 
157
    independent of the read_pos value!
 
158
 
 
159
  RETURN
 
160
  0  On error or EOF (info->error = -1 on error)
 
161
  #  Number of characters
 
162
*/
 
163
 
 
164
 
 
165
size_t my_b_fill(IO_CACHE *info)
 
166
{
 
167
  my_off_t pos_in_file=(info->pos_in_file+
 
168
                        (size_t) (info->read_end - info->buffer));
 
169
  size_t diff_length, length, max_length;
 
170
 
 
171
  if (info->seek_not_done)
 
172
  {                                     /* File touched, do seek */
 
173
    if (my_seek(info->file,pos_in_file,MY_SEEK_SET,MYF(0)) ==
 
174
        MY_FILEPOS_ERROR)
 
175
    {
 
176
      info->error= 0;
 
177
      return 0;
 
178
    }
 
179
    info->seek_not_done=0;
 
180
  }
 
181
  diff_length=(size_t) (pos_in_file & (IO_SIZE-1));
 
182
  max_length=(info->read_length-diff_length);
 
183
  if (max_length >= (info->end_of_file - pos_in_file))
 
184
    max_length= (size_t) (info->end_of_file - pos_in_file);
 
185
 
 
186
  if (!max_length)
 
187
  {
 
188
    info->error= 0;
 
189
    return 0;                                   /* EOF */
 
190
  }
 
191
  if ((length= my_read(info->file,info->buffer,max_length,
 
192
                       info->myflags)) == (size_t) -1)
 
193
  {
 
194
    info->error= -1;
 
195
    return 0;
 
196
  }
 
197
  info->read_pos=info->buffer;
 
198
  info->read_end=info->buffer+length;
 
199
  info->pos_in_file=pos_in_file;
 
200
  return length;
 
201
}
 
202
 
 
203
 
 
204
/*
 
205
  Read a string ended by '\n' into a buffer of 'max_length' size.
 
206
  Returns number of characters read, 0 on error.
 
207
  last byte is set to '\0'
 
208
  If buffer is full then to[max_length-1] will be set to \0.
 
209
*/
 
210
 
 
211
size_t my_b_gets(IO_CACHE *info, char *to, size_t max_length)
 
212
{
 
213
  char *start = to;
 
214
  size_t length;
 
215
  max_length--;                                 /* Save place for end \0 */
 
216
 
 
217
  /* Calculate number of characters in buffer */
 
218
  if (!(length= my_b_bytes_in_cache(info)) &&
 
219
      !(length= my_b_fill(info)))
 
220
    return 0;
 
221
 
 
222
  for (;;)
 
223
  {
 
224
    unsigned char *pos, *end;
 
225
    if (length > max_length)
 
226
      length=max_length;
 
227
    for (pos=info->read_pos,end=pos+length ; pos < end ;)
 
228
    {
 
229
      if ((*to++ = *pos++) == '\n')
 
230
      {
 
231
        info->read_pos=pos;
 
232
        *to='\0';
 
233
        return (size_t) (to-start);
 
234
      }
 
235
    }
 
236
    if (!(max_length-=length))
 
237
    {
 
238
     /* Found enough charcters;  Return found string */
 
239
      info->read_pos=pos;
 
240
      *to='\0';
 
241
      return (size_t) (to-start);
 
242
    }
 
243
    if (!(length=my_b_fill(info)))
 
244
      return 0;
 
245
  }
 
246
}
 
247
 
 
248
 
 
249
my_off_t my_b_filelength(IO_CACHE *info)
 
250
{
 
251
  if (info->type == WRITE_CACHE)
 
252
    return my_b_tell(info);
 
253
 
 
254
  info->seek_not_done= 1;
 
255
  return my_seek(info->file, 0L, MY_SEEK_END, MYF(0));
 
256
}
 
257
 
 
258
 
 
259
/*
 
260
  Simple printf version.  Supports '%s', '%d', '%u', "%ld" and "%lu"
 
261
  Used for logging in MySQL
 
262
  returns number of written character, or (size_t) -1 on error
 
263
*/
 
264
 
 
265
size_t my_b_printf(IO_CACHE *info, const char* fmt, ...)
 
266
{
 
267
  size_t result;
 
268
  va_list args;
 
269
  va_start(args,fmt);
 
270
  result=my_b_vprintf(info, fmt, args);
 
271
  va_end(args);
 
272
  return result;
 
273
}
 
274
 
 
275
 
 
276
size_t my_b_vprintf(IO_CACHE *info, const char* fmt, va_list args)
 
277
{
 
278
  size_t out_length= 0;
 
279
  uint32_t minimum_width; /* as yet unimplemented */
 
280
  uint32_t minimum_width_sign;
 
281
  uint32_t precision; /* as yet unimplemented for anything but %b */
 
282
  bool is_zero_padded;
 
283
 
 
284
  /*
 
285
    Store the location of the beginning of a format directive, for the
 
286
    case where we learn we shouldn't have been parsing a format string
 
287
    at all, and we don't want to lose the flag/precision/width/size
 
288
    information.
 
289
   */
 
290
  const char* backtrack;
 
291
 
 
292
  for (; *fmt != '\0'; fmt++)
 
293
  {
 
294
    /* Copy everything until '%' or end of string */
 
295
    const char *start=fmt;
 
296
    size_t length;
 
297
    
 
298
    for (; (*fmt != '\0') && (*fmt != '%'); fmt++) ;
 
299
 
 
300
    length= (size_t) (fmt - start);
 
301
    out_length+=length;
 
302
    if (my_b_write(info, (const unsigned char*) start, length))
 
303
      goto err;
 
304
 
 
305
    if (*fmt == '\0')                           /* End of format */
 
306
      return out_length;
 
307
 
 
308
    /* 
 
309
      By this point, *fmt must be a percent;  Keep track of this location and
 
310
      skip over the percent character. 
 
311
    */
 
312
    assert(*fmt == '%');
 
313
    backtrack= fmt;
 
314
    fmt++;
 
315
 
 
316
    is_zero_padded= false;
 
317
    minimum_width_sign= 1;
 
318
    minimum_width= 0;
 
319
    precision= 0;
 
320
    /* Skip if max size is used (to be compatible with printf) */
 
321
 
 
322
process_flags:
 
323
    switch (*fmt)
 
324
    {
 
325
      case '-': 
 
326
        minimum_width_sign= -1; fmt++; goto process_flags;
 
327
      case '0':
 
328
        is_zero_padded= true; fmt++; goto process_flags;
 
329
      case '#':
 
330
        /** @todo Implement "#" conversion flag. */  fmt++; goto process_flags;
 
331
      case ' ':
 
332
        /** @todo Implement " " conversion flag. */  fmt++; goto process_flags;
 
333
      case '+':
 
334
        /** @todo Implement "+" conversion flag. */  fmt++; goto process_flags;
 
335
    }
 
336
 
 
337
    if (*fmt == '*')
 
338
    {
 
339
      precision= (int) va_arg(args, int);
 
340
      fmt++;
 
341
    }
 
342
    else
 
343
    {
 
344
      while (my_isdigit(&my_charset_utf8_general_ci, *fmt)) {
 
345
        minimum_width=(minimum_width * 10) + (*fmt - '0');
 
346
        fmt++;
 
347
      }
 
348
    }
 
349
    minimum_width*= minimum_width_sign;
 
350
 
 
351
    if (*fmt == '.')
 
352
    {
 
353
      fmt++;
 
354
      if (*fmt == '*') {
 
355
        precision= (int) va_arg(args, int);
 
356
        fmt++;
 
357
      }
 
358
      else
 
359
      {
 
360
        while (my_isdigit(&my_charset_utf8_general_ci, *fmt)) {
 
361
          precision=(precision * 10) + (*fmt - '0');
 
362
          fmt++;
 
363
        }
 
364
      }
 
365
    }
 
366
 
 
367
    if (*fmt == 's')                            /* String parameter */
 
368
    {
 
369
      register char *par = va_arg(args, char *);
 
370
      size_t length2 = strlen(par);
 
371
      /* TODO: implement precision */
 
372
      out_length+= length2;
 
373
      if (my_b_write(info, (const unsigned char*) par, length2))
 
374
        goto err;
 
375
    }
 
376
    else if (*fmt == 'b')                       /* Sized buffer parameter, only precision makes sense */
 
377
    {
 
378
      char *par = va_arg(args, char *);
 
379
      out_length+= precision;
 
380
      if (my_b_write(info, (const unsigned char*) par, precision))
 
381
        goto err;
 
382
    }
 
383
    else if (*fmt == 'd' || *fmt == 'u')        /* Integer parameter */
 
384
    {
 
385
      register int iarg;
 
386
      size_t length2;
 
387
      char buff[17];
 
388
 
 
389
      iarg = va_arg(args, int);
 
390
      if (*fmt == 'd')
 
391
        length2= (size_t) (int10_to_str((long) iarg,buff, -10) - buff);
 
392
      else
 
393
        length2= (uint) (int10_to_str((long) (uint) iarg,buff,10)- buff);
 
394
 
 
395
      /* minimum width padding */
 
396
      if (minimum_width > length2) 
 
397
      {
 
398
        const size_t buflen = minimum_width - length2;
 
399
        unsigned char *buffz= my_alloca(buflen);
 
400
        memset(buffz, is_zero_padded ? '0' : ' ', buflen);
 
401
        my_b_write(info, buffz, buflen);
 
402
        my_afree(buffz);
 
403
      }
 
404
 
 
405
      out_length+= length2;
 
406
      if (my_b_write(info, (const unsigned char *)buff, length2))
 
407
        goto err;
 
408
    }
 
409
    else if ((*fmt == 'l' && fmt[1] == 'd') || fmt[1] == 'u')
 
410
      /* long parameter */
 
411
    {
 
412
      register long iarg;
 
413
      size_t length2;
 
414
      char buff[17];
 
415
 
 
416
      iarg = va_arg(args, long);
 
417
      if (*++fmt == 'd')
 
418
        length2= (size_t) (int10_to_str(iarg,buff, -10) - buff);
 
419
      else
 
420
        length2= (size_t) (int10_to_str(iarg,buff,10)- buff);
 
421
      out_length+= length2;
 
422
      if (my_b_write(info, (unsigned char*) buff, length2))
 
423
        goto err;
 
424
    }
 
425
    else
 
426
    {
 
427
      /* %% or unknown code */
 
428
      if (my_b_write(info, (const unsigned char*) backtrack, (size_t) (fmt-backtrack)))
 
429
        goto err;
 
430
      out_length+= fmt-backtrack;
 
431
    }
 
432
  }
 
433
  return out_length;
 
434
 
 
435
err:
 
436
  return (size_t) -1;
 
437
}