← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to plugin/md5/md5.cc

  • Committer: brian
  • Date: 2008-06-25 05:29:13 UTC
  • Revision ID: brian@localhost.localdomain-20080625052913-6upwo0jsrl4lnapl
clean slate

Show diffs side-by-side

added added

removed removed

Lines of Context:
plugin/md5/md5.cc
1
 
/* vim: expandtab:shiftwidth=2:tabstop=2:smarttab: 
2
 
   Copyright (C) 2006 MySQL AB
3
 
 
4
 
   This program is free software; you can redistribute it and/or modify
5
 
   it under the terms of the GNU General Public License as published by
6
 
   the Free Software Foundation; version 2 of the License.
7
 
 
8
 
   This program is distributed in the hope that it will be useful,
9
 
   but WITHOUT ANY WARRANTY; without even the implied warranty of
10
 
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
 
   GNU General Public License for more details.
12
 
 
13
 
   You should have received a copy of the GNU General Public License
14
 
   along with this program; if not, write to the Free Software
15
 
   Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA */
16
 
 
17
 
#include "config.h"
18
 
 
19
 
#include <cstdio>
20
 
#include <cstddef>
21
 
 
22
 
#include <gcrypt.h>
23
 
 
24
 
#include <drizzled/plugin/function.h>
25
 
#include <drizzled/item/func.h>
26
 
#include "drizzled/charset.h"
27
 
#include <drizzled/function/str/strfunc.h>
28
 
 
29
 
using namespace std;
30
 
using namespace drizzled;
31
 
 
32
 
class Md5Function : public Item_str_func
33
 
{
34
 
public:
35
 
  Md5Function() : Item_str_func() {}
36
 
  String *val_str(String*);
37
 
 
38
 
  void fix_length_and_dec() 
39
 
  {
40
 
    max_length= 32;
41
 
    args[0]->collation.set(
42
 
      get_charset_by_csname(args[0]->collation.collation->csname,
43
 
                            MY_CS_BINSORT), DERIVATION_COERCIBLE);
44
 
  }
45
 
 
46
 
  const char *func_name() const 
47
 
  { 
48
 
    return "md5"; 
49
 
  }
50
 
 
51
 
  bool check_argument_count(int n) 
52
 
  { 
53
 
    return (n == 1); 
54
 
  }
55
 
};
56
 
 
57
 
 
58
 
String *Md5Function::val_str(String *str)
59
 
{
60
 
  assert(fixed == true);
61
 
 
62
 
  String *sptr= args[0]->val_str(str);
63
 
  if (sptr == NULL || str->alloc(32)) 
64
 
  {
65
 
    null_value= true;
66
 
    return 0;
67
 
  }
68
 
 
69
 
  null_value= false;
70
 
 
71
 
  str->set_charset(&my_charset_bin);
72
 
 
73
 
  gcry_md_hd_t md5_context;
74
 
  gcry_md_open(&md5_context, GCRY_MD_MD5, 0);
75
 
  gcry_md_write(md5_context, sptr->ptr(), sptr->length());  
76
 
  unsigned char *digest= gcry_md_read(md5_context, 0);
77
 
 
78
 
  snprintf((char *) str->ptr(), 33,
79
 
    "%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x",
80
 
    digest[0], digest[1], digest[2], digest[3],
81
 
    digest[4], digest[5], digest[6], digest[7],
82
 
    digest[8], digest[9], digest[10], digest[11],
83
 
    digest[12], digest[13], digest[14], digest[15]);
84
 
  str->length((uint32_t) 32);
85
 
 
86
 
  gcry_md_close(md5_context);
87
 
 
88
 
  return str;
89
 
}
90
 
 
91
 
 
92
 
plugin::Create_function<Md5Function> *md5udf= NULL;
93
 
 
94
 
static int initialize(module::Context &context)
95
 
{
96
 
  /* Initialize libgcrypt */
97
 
  if (not gcry_check_version(GCRYPT_VERSION))
98
 
  {
99
 
    errmsg_printf(ERRMSG_LVL_ERROR, _("libgcrypt library version mismatch\n"));
100
 
    return 1;
101
 
  }
102
 
  /* Disable secure memory.  */
103
 
  gcry_control (GCRYCTL_DISABLE_SECMEM, 0);
104
 
 
105
 
  /* Tell Libgcrypt that initialization has completed. */
106
 
  gcry_control (GCRYCTL_INITIALIZATION_FINISHED, 0);
107
 
 
108
 
  md5udf= new plugin::Create_function<Md5Function>("md5");
109
 
  context.add(md5udf);
110
 
  return 0;
111
 
}
112
 
 
113
 
DRIZZLE_DECLARE_PLUGIN
114
 
{
115
 
  DRIZZLE_VERSION_ID,
116
 
  "md5",
117
 
  "1.0",
118
 
  "Stewart Smith",
119
 
  "UDF for computing md5sum",
120
 
  PLUGIN_LICENSE_GPL,
121
 
  initialize, /* Plugin Init */
122
 
  NULL,   /* system variables */
123
 
  NULL    /* config options */
124
 
}
125
 
DRIZZLE_DECLARE_PLUGIN_END;