← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to plugin/md5/md5.cc

  • Committer: Monty Taylor
  • Date: 2009-07-11 08:59:58 UTC
  • mto: (1093.1.12 captain)
  • mto: This revision was merged to the branch mainline in revision 1097.
  • Revision ID: mordred@inaugust.com-20090711085958-182jngk7bbe020q4
Removed dangerous asserts... mainly to upset Stewart.

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
 
1
/* Copyright (C) 2006 MySQL AB
3
2
 
4
3
   This program is free software; you can redistribute it and/or modify
5
4
   it under the terms of the GNU General Public License as published by
15
14
   Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA */
16
15
 
17
16
#include <drizzled/server_includes.h>
18
 
#include <drizzled/plugin/function.h>
 
17
#include <drizzled/sql_udf.h>
19
18
#include <drizzled/item/func.h>
20
19
#include <drizzled/function/str/strfunc.h>
21
20
 
22
 
#if defined(HAVE_LIBGNUTLS_OPENSSL)
 
21
#if defined(HAVE_GNUTLS_OPENSSL)
23
22
# include <gnutls/openssl.h>
24
23
#else
25
24
# include <openssl/md5.h>
28
27
#include <stdio.h>
29
28
 
30
29
using namespace std;
31
 
using namespace drizzled;
32
30
 
33
 
class Md5Function : public Item_str_func
 
31
class Item_func_md5 : public Item_str_func
34
32
{
35
33
public:
36
 
  Md5Function() : Item_str_func() {}
 
34
  Item_func_md5() : Item_str_func() {}
 
35
  const char *func_name() const { return "md5"; }
37
36
  String *val_str(String*);
38
 
 
39
 
  void fix_length_and_dec() 
40
 
  {
41
 
    max_length= 32;
 
37
  void fix_length_and_dec() {
 
38
    max_length=32;
42
39
    args[0]->collation.set(
43
40
      get_charset_by_csname(args[0]->collation.collation->csname,
44
41
                            MY_CS_BINSORT), DERIVATION_COERCIBLE);
45
42
  }
46
 
 
47
 
  const char *func_name() const 
48
 
  { 
49
 
    return "md5"; 
50
 
  }
51
 
 
52
 
  bool check_argument_count(int n) 
53
 
  { 
54
 
    return (n == 1); 
55
 
  }
 
43
  bool check_argument_count(int n) { return (n==1); }
56
44
};
57
45
 
58
46
 
59
 
String *Md5Function::val_str(String *str)
 
47
String *Item_func_md5::val_str(String *str)
60
48
{
61
 
  assert(fixed == true);
62
 
 
63
 
  String *sptr= args[0]->val_str(str);
64
 
  if (sptr == NULL || str->alloc(32)) 
65
 
  {
66
 
    null_value= true;
67
 
    return 0;
68
 
  }
69
 
 
70
 
  null_value= false;
71
 
 
72
 
  unsigned char digest[16];
 
49
  assert(fixed == 1);
 
50
  String * sptr= args[0]->val_str(str);
73
51
  str->set_charset(&my_charset_bin);
74
 
  MD5_CTX context;
75
 
  MD5_Init(&context);
76
 
  MD5_Update(&context, (unsigned char *) sptr->ptr(), sptr->length());
77
 
  MD5_Final(digest, &context);
78
 
 
79
 
  snprintf((char *) str->ptr(), 33,
80
 
    "%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x",
81
 
    digest[0], digest[1], digest[2], digest[3],
82
 
    digest[4], digest[5], digest[6], digest[7],
83
 
    digest[8], digest[9], digest[10], digest[11],
84
 
    digest[12], digest[13], digest[14], digest[15]);
85
 
  str->length((uint32_t) 32);
86
 
 
87
 
  return str;
88
 
}
89
 
 
90
 
 
91
 
plugin::Create_function<Md5Function> *md5udf= NULL;
92
 
 
93
 
static int initialize(plugin::Registry &registry)
94
 
{
95
 
  md5udf= new plugin::Create_function<Md5Function>("md5");
96
 
  registry.add(md5udf);
 
52
  if (sptr)
 
53
  {
 
54
    MD5_CTX context;
 
55
    unsigned char digest[16];
 
56
 
 
57
    null_value=0;
 
58
    MD5_Init (&context);
 
59
    MD5_Update (&context,(unsigned char *) sptr->ptr(), sptr->length());
 
60
    MD5_Final (digest, &context);
 
61
    if (str->alloc(32))                         // Ensure that memory is free
 
62
    {
 
63
      null_value=1;
 
64
      return 0;
 
65
    }
 
66
    snprintf((char *) str->ptr(), 33,
 
67
            "%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x",
 
68
            digest[0], digest[1], digest[2], digest[3],
 
69
            digest[4], digest[5], digest[6], digest[7],
 
70
            digest[8], digest[9], digest[10], digest[11],
 
71
            digest[12], digest[13], digest[14], digest[15]);
 
72
    str->length((uint) 32);
 
73
    return str;
 
74
  }
 
75
  null_value=1;
97
76
  return 0;
98
77
}
99
78
 
100
 
static int finalize(plugin::Registry &registry)
 
79
 
 
80
Create_function<Item_func_md5> md5udf(string("md5"));
 
81
 
 
82
static int md5udf_plugin_init(PluginRegistry &registry)
101
83
{
102
 
  registry.remove(md5udf);
103
 
  delete md5udf;
 
84
  registry.add(&md5udf);
104
85
  return 0;
105
86
}
106
87
 
111
92
  "Stewart Smith",
112
93
  "UDF for computing md5sum",
113
94
  PLUGIN_LICENSE_GPL,
114
 
  initialize, /* Plugin Init */
115
 
  finalize,   /* Plugin Deinit */
 
95
  md5udf_plugin_init, /* Plugin Init */
 
96
  NULL,   /* Plugin Deinit */
116
97
  NULL,   /* status variables */
117
98
  NULL,   /* system variables */
118
99
  NULL    /* config options */