sql_printf.c:
- Solucionado bug de "Segmentation fault". Debian: - Añadida dependencia con MariaDB.
This commit is contained in:
parent
d390a0ff15
commit
f127871a21
|
@ -8,7 +8,7 @@ Homepage: http://www.verdnatura.es
|
|||
|
||||
Package: vn-mysql
|
||||
Architecture: any
|
||||
Depends: ${shlibs:Depends}, ${misc:Depends}, mysql-server (>= 5.5.27)
|
||||
Description: Verdnatura MySQL plugins
|
||||
Depends: ${shlibs:Depends}, ${misc:Depends}, mysql-server (>= 5.5.27) | mariadb-galera-server (>= 5.5.27) | mariadb-server (>= 5.5.27)
|
||||
Description: MySQL plugins
|
||||
This package contains some useful MySQL plugins
|
||||
developed by Verdnatura.
|
||||
|
|
|
@ -4,83 +4,93 @@
|
|||
#include <mysql/mysql.h>
|
||||
|
||||
#define NAME "multimax"
|
||||
#define PARAMC 1
|
||||
#define NPARAMS 1
|
||||
#define BLOCK_SIZE 1024
|
||||
|
||||
typedef struct
|
||||
{
|
||||
int * org_type;
|
||||
char * data;
|
||||
size_t len;
|
||||
size_t allocated;
|
||||
unsigned long len;
|
||||
unsigned long alloc;
|
||||
}
|
||||
Buffer;
|
||||
|
||||
static void buffer_append (Buffer *buffer, char *string, size_t len)
|
||||
static void buffer_append (Buffer *buffer, char *string, unsigned long len)
|
||||
{
|
||||
if (!string)
|
||||
return;
|
||||
|
||||
unsigned long new_len = buffer->len + len;
|
||||
|
||||
if ((buffer->len + len) > buffer->allocated)
|
||||
if (new_len > buffer->alloc)
|
||||
{
|
||||
size_t allocate = buffer->allocated + (len > BLOCK_SIZE ? len : BLOCK_SIZE);
|
||||
buffer->data = realloc (buffer->data, allocate);
|
||||
buffer->allocated = allocate;
|
||||
buffer->alloc = new_len + BLOCK_SIZE - (new_len % BLOCK_SIZE);
|
||||
buffer->data = realloc (buffer->data, sizeof (char) * buffer->alloc);
|
||||
}
|
||||
|
||||
strncpy (&buffer->data[buffer->len], string, len);
|
||||
buffer->len += len;
|
||||
buffer->len = new_len;
|
||||
}
|
||||
|
||||
char * sql_printf (UDF_INIT *initid, UDF_ARGS *args, char *result, unsigned long *length, char *is_null, char *error)
|
||||
{
|
||||
int n = 1;
|
||||
size_t len;
|
||||
char *arg;
|
||||
char *aux;
|
||||
char delimiter;
|
||||
int i = 1;
|
||||
unsigned long j = 0;
|
||||
Buffer *buffer = (Buffer *) initid->ptr;
|
||||
char *string = args->args[0];
|
||||
char *format = args->args[0];
|
||||
|
||||
if (!string)
|
||||
if (!format)
|
||||
{
|
||||
*is_null = 1;
|
||||
return NULL;
|
||||
}
|
||||
|
||||
buffer->len = 0;
|
||||
string[args->lengths[0]] = '\0';
|
||||
|
||||
while ((aux = strchr (string, '%')))
|
||||
while (1)
|
||||
{
|
||||
buffer_append (buffer, string, aux - string);
|
||||
string = aux + 1;
|
||||
char *delimiter = NULL;
|
||||
unsigned long aux = j;
|
||||
|
||||
switch (*string)
|
||||
while (j < args->lengths[0] && format[j] != '%')
|
||||
j++;
|
||||
|
||||
buffer_append (buffer, &format[aux], j - aux);
|
||||
|
||||
if (j == args->lengths[0])
|
||||
break;
|
||||
|
||||
char c = format[j+1];
|
||||
j += 2;
|
||||
|
||||
switch (c)
|
||||
{
|
||||
case 'v':
|
||||
if (buffer->org_type[n] == STRING_RESULT)
|
||||
delimiter = "'";
|
||||
else
|
||||
delimiter = NULL;
|
||||
break;
|
||||
case 't':
|
||||
delimiter = "`";
|
||||
break;
|
||||
case 's':
|
||||
delimiter = NULL;
|
||||
case 'v':
|
||||
if (buffer->org_type[i] == STRING_RESULT)
|
||||
delimiter = "'";
|
||||
break;
|
||||
default:
|
||||
string--;
|
||||
case 's':
|
||||
break;
|
||||
case '%':
|
||||
buffer_append (buffer, "%", 1);
|
||||
continue;
|
||||
default:
|
||||
*error = 1;
|
||||
return NULL;
|
||||
}
|
||||
|
||||
if (i >= args->arg_count)
|
||||
{
|
||||
*error = 1;
|
||||
return NULL;
|
||||
}
|
||||
|
||||
if (arg)
|
||||
{
|
||||
arg = args->args[n];
|
||||
len = args->lengths[n];
|
||||
}
|
||||
else
|
||||
char *arg = args->args[i];
|
||||
unsigned long len = args->lengths[i];
|
||||
|
||||
if (!arg)
|
||||
{
|
||||
arg = "NULL";
|
||||
len = strlen (arg);
|
||||
|
@ -90,32 +100,28 @@ char * sql_printf (UDF_INIT *initid, UDF_ARGS *args, char *result, unsigned long
|
|||
buffer_append (buffer, delimiter, 1);
|
||||
buffer_append (buffer, arg, len);
|
||||
buffer_append (buffer, delimiter, 1);
|
||||
|
||||
string++;
|
||||
n++;
|
||||
i++;
|
||||
}
|
||||
|
||||
buffer_append (buffer, string, strlen (string) + 1);
|
||||
*length = buffer->len - 1;
|
||||
*length = buffer->len;
|
||||
return buffer->data;
|
||||
}
|
||||
|
||||
my_bool sql_printf_init (UDF_INIT *initid, UDF_ARGS *args, char *message)
|
||||
{
|
||||
int count = args->arg_count;
|
||||
|
||||
if (count >= PARAMC)
|
||||
if (args->arg_count >= NPARAMS)
|
||||
{
|
||||
int n;
|
||||
int i;
|
||||
Buffer *buffer = malloc (sizeof (Buffer));
|
||||
buffer->org_type = malloc (sizeof (int) * count);
|
||||
buffer->data = malloc (sizeof (char) * BLOCK_SIZE);
|
||||
buffer->allocated = BLOCK_SIZE;
|
||||
buffer->org_type = malloc (sizeof (int) * args->arg_count);
|
||||
buffer->alloc = BLOCK_SIZE;
|
||||
buffer->data = malloc (sizeof (char) * buffer->alloc);
|
||||
buffer->len = 0;
|
||||
|
||||
for (n = 0; n < count; n++)
|
||||
for (i = 0; i < args->arg_count; i++)
|
||||
{
|
||||
buffer->org_type[n] = args->arg_type[n];
|
||||
args->arg_type[n] = STRING_RESULT;
|
||||
buffer->org_type[i] = args->arg_type[i];
|
||||
args->arg_type[i] = STRING_RESULT;
|
||||
}
|
||||
|
||||
initid->ptr = (void *) buffer;
|
||||
|
@ -125,7 +131,7 @@ my_bool sql_printf_init (UDF_INIT *initid, UDF_ARGS *args, char *message)
|
|||
}
|
||||
else
|
||||
{
|
||||
sprintf (message, "%s must have at least %d parameters", NAME, PARAMC);
|
||||
sprintf (message, "%s must have at least %d parameters", NAME, NPARAMS);
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
|
|
Reference in New Issue