sql_string.h 10.4 KB
Newer Older
unknown's avatar
unknown committed
1 2 3 4 5 6 7 8
/* Copyright (C) 2000 MySQL AB

   This program is free software; you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
   the Free Software Foundation; either version 2 of the License, or
   (at your option) any later version.

   This program is distributed in the hope that it will be useful,
unknown's avatar
unknown committed
9
   but WITHOUT ANY WARRANTY; without even the implied warranty of
unknown's avatar
unknown committed
10 11 12 13 14 15
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.

   You should have received a copy of the GNU General Public License
   along with this program; if not, write to the Free Software
   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
unknown's avatar
unknown committed
16 17 18 19 20 21 22 23 24 25 26

/* This file is originally from the mysql distribution. Coded by monty */

#ifdef __GNUC__
#pragma interface			/* gcc class implementation */
#endif

#ifndef NOT_FIXED_DEC
#define NOT_FIXED_DEC			31
#endif

27
class String;
28
int sortcmp(const String *a,const String *b, CHARSET_INFO *cs);
29
String *copy_if_not_alloced(String *a,String *b,uint32 arg_length);
unknown's avatar
unknown committed
30 31
uint32 copy_and_convert(char *to, uint32 to_length, CHARSET_INFO *to_cs,
			const char *from, uint32 from_length,
32
			CHARSET_INFO *from_cs, uint *errors);
33

unknown's avatar
unknown committed
34 35 36 37 38
class String
{
  char *Ptr;
  uint32 str_length,Alloced_length;
  bool alloced;
39
  CHARSET_INFO *str_charset;
unknown's avatar
unknown committed
40 41
public:
  String()
42 43
  { 
    Ptr=0; str_length=Alloced_length=0; alloced=0; 
unknown's avatar
unknown committed
44
    str_charset= &my_charset_bin; 
45
  }
unknown's avatar
unknown committed
46
  String(uint32 length_arg)
47 48
  { 
    alloced=0; Alloced_length=0; (void) real_alloc(length_arg); 
unknown's avatar
unknown committed
49
    str_charset= &my_charset_bin;
50
  }
51
  String(const char *str, CHARSET_INFO *cs)
52 53
  { 
    Ptr=(char*) str; str_length=(uint) strlen(str); Alloced_length=0; alloced=0;
54
    str_charset=cs;
55
  }
56
  String(const char *str,uint32 len, CHARSET_INFO *cs)
57 58
  { 
    Ptr=(char*) str; str_length=len; Alloced_length=0; alloced=0;
59
    str_charset=cs;
60
  }
61
  String(char *str,uint32 len, CHARSET_INFO *cs)
62 63
  { 
    Ptr=(char*) str; Alloced_length=str_length=len; alloced=0;
64
    str_charset=cs;
65
  }
unknown's avatar
unknown committed
66
  String(const String &str)
67 68 69 70 71
  { 
    Ptr=str.Ptr ; str_length=str.str_length ;
    Alloced_length=str.Alloced_length; alloced=0; 
    str_charset=str.str_charset;
  }
72 73
  static void *operator new(size_t size, MEM_ROOT *mem_root)
  { return (void*) alloc_root(mem_root, (uint) size); }
74
  static void operator delete(void *ptr_arg,size_t size)
unknown's avatar
unknown committed
75
  { TRASH(ptr_arg, size); }
76
  static void operator delete(void *ptr_arg, MEM_ROOT *mem_root)
unknown's avatar
unknown committed
77
  { /* never called */ }
unknown's avatar
unknown committed
78 79
  ~String() { free(); }

unknown's avatar
unknown committed
80
  inline void set_charset(CHARSET_INFO *charset) { str_charset= charset; }
81
  inline CHARSET_INFO *charset() const { return str_charset; }
unknown's avatar
unknown committed
82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99
  inline uint32 length() const { return str_length;}
  inline uint32 alloced_length() const { return Alloced_length;}
  inline char& operator [] (uint32 i) const { return Ptr[i]; }
  inline void length(uint32 len) { str_length=len ; }
  inline bool is_empty() { return (str_length == 0); }
  inline const char *ptr() const { return Ptr; }
  inline char *c_ptr()
  {
    if (!Ptr || Ptr[str_length])		/* Should be safe */
      (void) realloc(str_length);
    return Ptr;
  }
  inline char *c_ptr_quick()
  {
    if (Ptr && str_length < Alloced_length)
      Ptr[str_length]=0;
    return Ptr;
  }
unknown's avatar
unknown committed
100 101 102 103 104 105 106 107
  inline char *c_ptr_safe()
  {
    if (Ptr && str_length < Alloced_length)
      Ptr[str_length]=0;
    else
      (void) realloc(str_length);
    return Ptr;
  }
unknown's avatar
unknown committed
108 109 110

  void set(String &str,uint32 offset,uint32 arg_length)
  {
111
    DBUG_ASSERT(&str != this);
unknown's avatar
unknown committed
112 113 114 115 116 117
    free();
    Ptr=(char*) str.ptr()+offset; str_length=arg_length; alloced=0;
    if (str.Alloced_length)
      Alloced_length=str.Alloced_length-offset;
    else
      Alloced_length=0;
118
    str_charset=str.str_charset;
unknown's avatar
unknown committed
119
  }
120
  inline void set(char *str,uint32 arg_length, CHARSET_INFO *cs)
unknown's avatar
unknown committed
121 122 123
  {
    free();
    Ptr=(char*) str; str_length=Alloced_length=arg_length ; alloced=0;
124
    str_charset=cs;
unknown's avatar
unknown committed
125
  }
126
  inline void set(const char *str,uint32 arg_length, CHARSET_INFO *cs)
unknown's avatar
unknown committed
127 128 129
  {
    free();
    Ptr=(char*) str; str_length=arg_length; Alloced_length=0 ; alloced=0;
130
    str_charset=cs;
unknown's avatar
unknown committed
131
  }
132
  bool set_ascii(const char *str, uint32 arg_length);
133
  inline void set_quick(char *str,uint32 arg_length, CHARSET_INFO *cs)
unknown's avatar
unknown committed
134 135 136 137 138
  {
    if (!alloced)
    {
      Ptr=(char*) str; str_length=Alloced_length=arg_length;
    }
139
    str_charset=cs;
unknown's avatar
unknown committed
140
  }
141 142 143
  bool set(longlong num, CHARSET_INFO *cs);
  bool set(ulonglong num, CHARSET_INFO *cs);
  bool set(double num,uint decimals, CHARSET_INFO *cs);
144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171

  /*
    PMG 2004.11.12
    This is a method that works the same as perl's "chop". It simply
    drops the last character of a string. This is useful in the case
    of the federated storage handler where I'm building a unknown
    number, list of values and fields to be used in a sql insert
    statement to be run on the remote server, and have a comma after each.
    When the list is complete, I "chop" off the trailing comma

    ex. 
      String stringobj; 
      stringobj.append("VALUES ('foo', 'fi', 'fo',");
      stringobj.chop();
      stringobj.append(")");

    In this case, the value of string was:

    VALUES ('foo', 'fi', 'fo',
    VALUES ('foo', 'fi', 'fo'
    VALUES ('foo', 'fi', 'fo')
      
  */
  inline void chop()
  {
    Ptr[str_length--]= '\0'; 
  }

unknown's avatar
unknown committed
172
  inline void free()
unknown's avatar
unknown committed
173 174
  {
    if (alloced)
unknown's avatar
unknown committed
175
    {
unknown's avatar
unknown committed
176 177 178 179 180
      alloced=0;
      Alloced_length=0;
      my_free(Ptr,MYF(0));
      Ptr=0;
      str_length=0;				/* Safety */
unknown's avatar
unknown committed
181
    }
unknown's avatar
unknown committed
182
  }
unknown's avatar
unknown committed
183 184 185 186 187 188 189 190 191 192 193 194 195 196 197
  inline bool alloc(uint32 arg_length)
  {
    if (arg_length < Alloced_length)
      return 0;
    return real_alloc(arg_length);
  }
  bool real_alloc(uint32 arg_length);			// Empties old string
  bool realloc(uint32 arg_length);
  inline void shrink(uint32 arg_length)		// Shrink buffer
  {
    if (arg_length < Alloced_length)
    {
      char *new_ptr;
      if (!(new_ptr=(char*) my_realloc(Ptr,arg_length,MYF(0))))
      {
unknown's avatar
unknown committed
198
	Alloced_length = 0;
unknown's avatar
unknown committed
199 200 201 202 203 204 205 206 207
	real_alloc(arg_length);
      }
      else
      {
	Ptr=new_ptr;
	Alloced_length=arg_length;
      }
    }
  }
208 209 210 211
  inline void shrink_to_length()
  {
    Alloced_length= str_length;
  }
unknown's avatar
unknown committed
212 213 214 215 216
  bool is_alloced() { return alloced; }
  inline String& operator = (const String &s)
  {
    if (&s != this)
    {
217 218 219 220 221
      /*
        It is forbidden to do assignments like 
        some_string = substring_of_that_string
       */
      DBUG_ASSERT(!s.uses_buffer_owned_by(this));
unknown's avatar
unknown committed
222 223 224 225 226 227 228 229 230
      free();
      Ptr=s.Ptr ; str_length=s.str_length ; Alloced_length=s.Alloced_length;
      alloced=0;
    }
    return *this;
  }

  bool copy();					// Alloc string if not alloced
  bool copy(const String &s);			// Allocate new string
231
  bool copy(const char *s,uint32 arg_length, CHARSET_INFO *cs);	// Allocate new string
unknown's avatar
unknown committed
232 233 234 235 236
  static bool needs_conversion(uint32 arg_length,
  			       CHARSET_INFO *cs_from, CHARSET_INFO *cs_to,
			       uint32 *offset);
  bool copy_aligned(const char *s, uint32 arg_length, uint32 offset,
		    CHARSET_INFO *cs);
unknown's avatar
unknown committed
237
  bool set_or_copy_aligned(const char *s, uint32 arg_length, CHARSET_INFO *cs);
unknown's avatar
unknown committed
238
  bool copy(const char*s,uint32 arg_length, CHARSET_INFO *csfrom,
239
	    CHARSET_INFO *csto, uint *errors);
unknown's avatar
unknown committed
240
  bool append(const String &s);
241 242
  bool append(const char *s);
  bool append(const char *s,uint32 arg_length);
243
  bool append(const char *s,uint32 arg_length, CHARSET_INFO *cs);
244
  bool append(IO_CACHE* file, uint32 arg_length);
unknown's avatar
unknown committed
245 246
  bool append_with_prefill(const char *s, uint32 arg_length, 
			   uint32 full_length, char fill_char);
unknown's avatar
unknown committed
247 248
  int strstr(const String &search,uint32 offset=0); // Returns offset to substring or -1
  int strrstr(const String &search,uint32 offset=0); // Returns offset to substring or -1
249
  bool replace(uint32 offset,uint32 arg_length,const char *to,uint32 length);
unknown's avatar
unknown committed
250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266
  bool replace(uint32 offset,uint32 arg_length,const String &to);
  inline bool append(char chr)
  {
    if (str_length < Alloced_length)
    {
      Ptr[str_length++]=chr;
    }
    else
    {
      if (realloc(str_length+1))
	return 1;
      Ptr[str_length++]=chr;
    }
    return 0;
  }
  bool fill(uint32 max_length,char fill);
  void strip_sp();
267 268
  inline void caseup() { my_caseup(str_charset,Ptr,str_length); }
  inline void casedn() { my_casedn(str_charset,Ptr,str_length); }
269
  friend int sortcmp(const String *a,const String *b, CHARSET_INFO *cs);
unknown's avatar
unknown committed
270
  friend int stringcmp(const String *a,const String *b);
unknown's avatar
unknown committed
271 272 273
  friend String *copy_if_not_alloced(String *a,String *b,uint32 arg_length);
  uint32 numchars();
  int charpos(int i,uint32 offset=0);
unknown's avatar
unknown committed
274 275 276 277 278 279 280

  int reserve(uint32 space_needed)
  {
    return realloc(str_length + space_needed);
  }
  int reserve(uint32 space_needed, uint32 grow_by);

unknown's avatar
unknown committed
281 282 283 284 285
  /*
    The following append operations do NOT check alloced memory
    q_*** methods writes values of parameters itself
    qs_*** methods writes string representation of value
  */
unknown's avatar
unknown committed
286
  void q_append(const char c)
unknown's avatar
unknown committed
287 288 289
  {
    Ptr[str_length++] = c;
  }
unknown's avatar
unknown committed
290
  void q_append(const uint32 n)
unknown's avatar
unknown committed
291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310
  {
    int4store(Ptr + str_length, n);
    str_length += 4;
  }
  void q_append(double d)
  {
    float8store(Ptr + str_length, d);
    str_length += 8;
  }
  void q_append(double *d)
  {
    float8store(Ptr + str_length, *d);
    str_length += 8;
  }
  void q_append(const char *data, uint32 data_len)
  {
    memcpy(Ptr + str_length, data, data_len);
    str_length += data_len;
  }

unknown's avatar
unknown committed
311
  void write_at_position(int position, uint32 value)
unknown's avatar
unknown committed
312 313 314 315
  {
    int4store(Ptr + position,value);
  }

unknown's avatar
unknown committed
316
  void qs_append(const char *str, uint32 len);
unknown's avatar
unknown committed
317 318
  void qs_append(double d);
  void qs_append(double *d);
unknown's avatar
unknown committed
319 320 321 322 323
  inline void qs_append(const char c)
  {
     Ptr[str_length]= c;
     str_length++;
  }
324 325
  void qs_append(int i);
  void qs_append(uint i);
326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350

  /* Inline (general) functions used by the protocol functions */

  inline char *prep_append(uint32 arg_length, uint32 step_alloc)
  {
    uint32 new_length= arg_length + str_length;
    if (new_length > Alloced_length)
    {
      if (realloc(new_length + step_alloc))
        return 0;
    }
    uint32 old_length= str_length;
    str_length+= arg_length;
    return Ptr+ old_length;			/* Area to use */
  }

  inline bool append(const char *s, uint32 arg_length, uint32 step_alloc)
  {
    uint32 new_length= arg_length + str_length;
    if (new_length > Alloced_length && realloc(new_length + step_alloc))
      return TRUE;
    memcpy(Ptr+str_length, s, arg_length);
    str_length+= arg_length;
    return FALSE;
  }
unknown's avatar
unknown committed
351
  void print(String *print);
352 353 354

  /* Swap two string objects. Efficient way to exchange data without memcpy. */
  void swap(String &s);
355 356 357 358 359

  inline bool uses_buffer_owned_by(const String *s) const
  {
    return (s->alloced && Ptr >= s->Ptr && Ptr < s->Ptr + s->str_length);
  }
unknown's avatar
unknown committed
360
};