← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to plugin/compress/compressudf.cc

  • Committer: Monty Taylor
  • Date: 2008-07-01 14:33:36 UTC
  • mto: (28.1.12 backport_patch)
  • mto: This revision was merged to the branch mainline in revision 34.
  • Revision ID: monty@inaugust.com-20080701143336-8uihm7dhpu92rt0q
Somehow missed moving password.c. Duh.

Show diffs side-by-side

added added

removed removed

Lines of Context:
plugin/compress/compressudf.cc
1
 
/* Copyright (C) 2006 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., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA */
15
 
 
16
 
#include <drizzled/common_includes.h>
17
 
#include <drizzled/item_func.h>
18
 
#include <drizzled/item_strfunc.h>
19
 
#include <drizzled/drizzled_error_messages.h>
20
 
#include <zlib.h>
21
 
 
22
 
class Item_func_compress: public Item_str_func
23
 
{
24
 
  String buffer;
25
 
public:
26
 
  Item_func_compress():Item_str_func(){}
27
 
  void fix_length_and_dec(){max_length= (args[0]->max_length*120)/100+12;}
28
 
  const char *func_name() const{return "compress";}
29
 
  String *val_str(String *) ;
30
 
};
31
 
 
32
 
String *Item_func_compress::val_str(String *str)
33
 
{
34
 
  int err= Z_OK, code;
35
 
  ulong new_size;
36
 
  String *res;
37
 
  Byte *body;
38
 
  char *tmp, *last_char;
39
 
  assert(fixed == 1);
40
 
 
41
 
  if (!(res= args[0]->val_str(str)))
42
 
  {
43
 
    null_value= 1;
44
 
    return 0;
45
 
  }
46
 
  null_value= 0;
47
 
  if (res->is_empty()) return res;
48
 
 
49
 
  /*
50
 
    Citation from zlib.h (comment for compress function):
51
 
 
52
 
    Compresses the source buffer into the destination buffer.  sourceLen is
53
 
    the byte length of the source buffer. Upon entry, destLen is the total
54
 
    size of the destination buffer, which must be at least 0.1% larger than
55
 
    sourceLen plus 12 bytes.
56
 
    We assume here that the buffer can't grow more than .25 %.
57
 
  */
58
 
  new_size= res->length() + res->length() / 5 + 12;
59
 
 
60
 
  // Check new_size overflow: new_size <= res->length()
61
 
  if (((uint32_t) (new_size+5) <= res->length()) || 
62
 
      buffer.realloc((uint32_t) new_size + 4 + 1))
63
 
  {
64
 
    null_value= 1;
65
 
    return 0;
66
 
  }
67
 
 
68
 
  body= ((Byte*)buffer.ptr()) + 4;
69
 
 
70
 
  // As far as we have checked res->is_empty() we can use ptr()
71
 
  if ((err= compress(body, &new_size,
72
 
                     (const Bytef*)res->ptr(), res->length())) != Z_OK)
73
 
  {
74
 
    code= err==Z_MEM_ERROR ? ER_ZLIB_Z_MEM_ERROR : ER_ZLIB_Z_BUF_ERROR;
75
 
    push_warning(current_thd, DRIZZLE_ERROR::WARN_LEVEL_ERROR, code, ER(code));
76
 
    null_value= 1;
77
 
    return 0;
78
 
  }
79
 
 
80
 
  tmp= (char*)buffer.ptr(); // int4store is a macro; avoid side effects
81
 
  int4store(tmp, res->length() & 0x3FFFFFFF);
82
 
 
83
 
  /* This is to ensure that things works for CHAR fields, which trim ' ': */
84
 
  last_char= ((char*)body)+new_size-1;
85
 
  if (*last_char == ' ')
86
 
  {
87
 
    *++last_char= '.';
88
 
    new_size++;
89
 
  }
90
 
 
91
 
  buffer.length((uint32_t)new_size + 4);
92
 
  return &buffer;
93
 
}
94
 
 
95
 
 
96
 
Item_func* create_compressudf_item(MEM_ROOT* m)
97
 
{
98
 
  return  new (m) Item_func_compress();
99
 
}
100
 
 
101
 
static struct udf_func compressudf = {
102
 
  { C_STRING_WITH_LEN("compress") },
103
 
  create_compressudf_item
104
 
};
105
 
 
106
 
static int compressudf_plugin_init(void *p)
107
 
{
108
 
  udf_func **f = (udf_func**) p;
109
 
 
110
 
  *f= &compressudf;
111
 
 
112
 
  return 0;
113
 
}
114
 
 
115
 
static int compressudf_plugin_deinit(void *p)
116
 
{
117
 
  udf_func *udff = (udf_func *) p;
118
 
  (void)udff;
119
 
  return 0;
120
 
}
121
 
 
122
 
mysql_declare_plugin(compress)
123
 
{
124
 
  DRIZZLE_UDF_PLUGIN,
125
 
  "compress",
126
 
  "1.0",
127
 
  "Stewart Smith",
128
 
  "UDF for compress()",
129
 
  PLUGIN_LICENSE_GPL,
130
 
  compressudf_plugin_init, /* Plugin Init */
131
 
  compressudf_plugin_deinit, /* Plugin Deinit */
132
 
  NULL,   /* status variables */
133
 
  NULL,   /* system variables */
134
 
  NULL    /* config options */
135
 
}
136
 
mysql_declare_plugin_end;