← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to plugin/compress/compressudf.cc

  • Committer: Brian Aker
  • Date: 2008-11-04 15:39:09 UTC
  • mfrom: (575.1.2 devel)
  • Revision ID: brian@tangent.org-20081104153909-c72hn65udxs1ccal
Merge of Monty's work

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/error.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_session, 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;