sql_string.h 8.91 KB
Newer Older
monty@hundin.mysql.fi's avatar
monty@hundin.mysql.fi 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,
bk@work.mysql.com's avatar
bk@work.mysql.com committed
9
   but WITHOUT ANY WARRANTY; without even the implied warranty of
monty@hundin.mysql.fi's avatar
monty@hundin.mysql.fi 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 */
bk@work.mysql.com's avatar
bk@work.mysql.com 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);
monty@mashka.mysql.fi's avatar
monty@mashka.mysql.fi committed
30 31 32
uint32 copy_and_convert(char *to, uint32 to_length, CHARSET_INFO *to_cs,
			const char *from, uint32 from_length,
			CHARSET_INFO *from_cs);
33

bk@work.mysql.com's avatar
bk@work.mysql.com committed
34 35 36 37 38
class String
{
  char *Ptr;
  uint32 str_length,Alloced_length;
  bool alloced;
39
  CHARSET_INFO *str_charset;
bk@work.mysql.com's avatar
bk@work.mysql.com committed
40 41
public:
  String()
42 43
  { 
    Ptr=0; str_length=Alloced_length=0; alloced=0; 
bar@bar.mysql.r18.ru's avatar
bar@bar.mysql.r18.ru committed
44
    str_charset= &my_charset_bin; 
45
  }
bk@work.mysql.com's avatar
bk@work.mysql.com committed
46
  String(uint32 length_arg)
47 48
  { 
    alloced=0; Alloced_length=0; (void) real_alloc(length_arg); 
bar@bar.mysql.r18.ru's avatar
bar@bar.mysql.r18.ru 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
  }
bk@work.mysql.com's avatar
bk@work.mysql.com 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 74 75
  static void *operator new(size_t size)
  { return (void*) sql_alloc((uint) size); }
  static void *operator new(size_t size, MEM_ROOT *mem_root)
  { return (void*) alloc_root(mem_root, (uint) size); }
bk@work.mysql.com's avatar
bk@work.mysql.com committed
76 77 78 79
  static void operator delete(void *ptr_arg,size_t size) /*lint -e715 */
    { sql_element_free(ptr_arg); }
  ~String() { free(); }

80
  inline void set_charset(CHARSET_INFO *charset) { str_charset= charset; }
81
  inline CHARSET_INFO *charset() const { return str_charset; }
bk@work.mysql.com's avatar
bk@work.mysql.com committed
82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108
  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;
  }

  void set(String &str,uint32 offset,uint32 arg_length)
  {
    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;
109
    str_charset=str.str_charset;
bk@work.mysql.com's avatar
bk@work.mysql.com committed
110
  }
111
  inline void set(char *str,uint32 arg_length, CHARSET_INFO *cs)
bk@work.mysql.com's avatar
bk@work.mysql.com committed
112 113 114
  {
    free();
    Ptr=(char*) str; str_length=Alloced_length=arg_length ; alloced=0;
115
    str_charset=cs;
bk@work.mysql.com's avatar
bk@work.mysql.com committed
116
  }
117
  inline void set(const char *str,uint32 arg_length, CHARSET_INFO *cs)
bk@work.mysql.com's avatar
bk@work.mysql.com committed
118 119 120
  {
    free();
    Ptr=(char*) str; str_length=arg_length; Alloced_length=0 ; alloced=0;
121
    str_charset=cs;
bk@work.mysql.com's avatar
bk@work.mysql.com committed
122
  }
123
  bool set_ascii(const char *str, uint32 arg_length);
124
  inline void set_quick(char *str,uint32 arg_length, CHARSET_INFO *cs)
bk@work.mysql.com's avatar
bk@work.mysql.com committed
125 126 127 128 129
  {
    if (!alloced)
    {
      Ptr=(char*) str; str_length=Alloced_length=arg_length;
    }
130
    str_charset=cs;
bk@work.mysql.com's avatar
bk@work.mysql.com committed
131
  }
132 133 134
  bool set(longlong num, CHARSET_INFO *cs);
  bool set(ulonglong num, CHARSET_INFO *cs);
  bool set(double num,uint decimals, CHARSET_INFO *cs);
bk@work.mysql.com's avatar
bk@work.mysql.com committed
135
  inline void free()
monty@narttu.mysql.fi's avatar
monty@narttu.mysql.fi committed
136 137
  {
    if (alloced)
bk@work.mysql.com's avatar
bk@work.mysql.com committed
138
    {
monty@narttu.mysql.fi's avatar
monty@narttu.mysql.fi committed
139 140 141 142 143
      alloced=0;
      Alloced_length=0;
      my_free(Ptr,MYF(0));
      Ptr=0;
      str_length=0;				/* Safety */
bk@work.mysql.com's avatar
bk@work.mysql.com committed
144
    }
monty@narttu.mysql.fi's avatar
monty@narttu.mysql.fi committed
145
  }
bk@work.mysql.com's avatar
bk@work.mysql.com committed
146 147 148 149 150 151 152 153 154 155 156 157 158 159 160
  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))))
      {
161
	Alloced_length = 0;
bk@work.mysql.com's avatar
bk@work.mysql.com committed
162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184
	real_alloc(arg_length);
      }
      else
      {
	Ptr=new_ptr;
	Alloced_length=arg_length;
      }
    }
  }
  bool is_alloced() { return alloced; }
  inline String& operator = (const String &s)
  {
    if (&s != this)
    {
      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
185
  bool copy(const char *s,uint32 arg_length, CHARSET_INFO *cs);	// Allocate new string
186 187 188 189 190
  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);
191
  bool set_or_copy_aligned(const char *s, uint32 arg_length, CHARSET_INFO *cs);
192 193
  bool copy(const char*s,uint32 arg_length, CHARSET_INFO *csfrom,
	    CHARSET_INFO *csto);
bk@work.mysql.com's avatar
bk@work.mysql.com committed
194 195
  bool append(const String &s);
  bool append(const char *s,uint32 arg_length=0);
196
  bool append(const char *s,uint32 arg_length, CHARSET_INFO *cs);
197
  bool append(IO_CACHE* file, uint32 arg_length);
198 199
  bool append_with_prefill(const char *s, uint32 arg_length, 
			   uint32 full_length, char fill_char);
bk@work.mysql.com's avatar
bk@work.mysql.com committed
200
  int strstr(const String &search,uint32 offset=0); // Returns offset to substring or -1
201
  int strstr_case(const String &s,uint32 offset=0);
bk@work.mysql.com's avatar
bk@work.mysql.com committed
202
  int strrstr(const String &search,uint32 offset=0); // Returns offset to substring or -1
203
  bool replace(uint32 offset,uint32 arg_length,const char *to,uint32 length);
bk@work.mysql.com's avatar
bk@work.mysql.com committed
204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220
  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();
221 222
  inline void caseup() { my_caseup(str_charset,Ptr,str_length); }
  inline void casedn() { my_casedn(str_charset,Ptr,str_length); }
223
  friend int sortcmp(const String *a,const String *b, CHARSET_INFO *cs);
monty@mysql.com's avatar
monty@mysql.com committed
224
  friend int stringcmp(const String *a,const String *b);
bk@work.mysql.com's avatar
bk@work.mysql.com committed
225 226 227
  friend String *copy_if_not_alloced(String *a,String *b,uint32 arg_length);
  uint32 numchars();
  int charpos(int i,uint32 offset=0);
228 229 230 231 232 233 234

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

235 236 237 238 239
  /*
    The following append operations do NOT check alloced memory
    q_*** methods writes values of parameters itself
    qs_*** methods writes string representation of value
  */
240
  void q_append(const char c)
241 242 243
  {
    Ptr[str_length++] = c;
  }
hf@deer.mysql.r18.ru's avatar
hf@deer.mysql.r18.ru committed
244
  void q_append(const uint32 n)
245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264
  {
    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;
  }

265
  void write_at_position(int position, uint32 value)
266 267 268 269
  {
    int4store(Ptr + position,value);
  }

270
  void qs_append(const char *str, uint32 len);
271 272
  void qs_append(double d);
  void qs_append(double *d);
273 274 275 276 277
  inline void qs_append(const char c)
  {
     Ptr[str_length]= c;
     str_length++;
  }
278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302

  /* 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;
  }
bell@sanja.is.com.ua's avatar
bell@sanja.is.com.ua committed
303
  void print(String *print);
bk@work.mysql.com's avatar
bk@work.mysql.com committed
304
};