utils.c

   1 /*
   2  * Copyright (c) 2006 Teodor Sigaev <teodor@sigaev.ru>
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *        notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *        notice, this list of conditions and the following disclaimer in the
  12  *        documentation and/or other materials provided with the distribution.
  13  * 3. Neither the name of the author nor the names of any co-contributors
  14  *        may be used to endorse or promote products derived from this software
  15  *        without specific prior written permission.
  16  *
  17  * THIS SOFTWARE IS PROVIDED BY CONTRIBUTORS ``AS IS'' AND ANY EXPRESS
  18  * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  19  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  20  * ARE DISCLAIMED. IN NO EVENT SHALL CONTRIBUTORS BE LIABLE FOR ANY
  21  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  22  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE
  23  * GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  24  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER
  25  * IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
  26  * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN
  27  * IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  28  */
  29
  30 #include <string.h>
  31 #include <stdio.h>
  32 #include <stdlib.h>
  33
  34 #include "ftsbench.h"
  35
  36 void
  37 sb_add(StringBuf *b, char *s, int length)
  38 {
  39         if ( length <=0 )
  40                 length = strlen(s);
  41
  42         if ( b->strlen + length + 1 >= b->length ) {
  43                 if ( b->str == NULL || b->length == 0 ) {
  44                         b->length = (length + 1)*4;
  45                         b->str = (char*)malloc( sizeof(char) * b->length );
  46                 } else {
  47                         do
  48                                 b->length *= 2;
  49                         while( b->strlen + length + 1 >= b->length );
  50                         b->str = (char*)realloc( b->str, sizeof(char) * b->length );
  51                 }
  52
  53                 if (!b->str) {
  54                         fprintf(stderr,"Not enough memory (%d bytes)\n", b->length);
  55                         exit(1);
  56                 }
  57         }
  58
  59         memcpy(b->str + b->strlen, s, length);
  60         b->strlen += length;
  61         b->str[ b->strlen ] = '\0';
  62 }
  63
  64