lib_mysqludf_getpwuid.c

/*
 * Skeleton libary for MySQL.
 * A set of MySQL user defined functions (UDF) to [DESCRIPTION]
 *
 * Copyright (C) [YYYY YOUR NAME <YOU@EXAMPLE.COM>]
 *
 * This library is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation; either version 2.1 of the License, or (at
 * your option) any later version.
 *
 * This library is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser
 * General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this library; if not, write to the Free Software Foundation, Inc.,
 * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
 */

#include "mysqludf.h"

/* For Windows, define PACKAGE_STRING in the VS project */
#ifndef __WIN__
#include "config.h"
#endif

/* These must be right or mysqld will not find the symbol! */
#ifdef	__cplusplus
extern "C" {
#endif
	DLLEXP my_bool lib_mysqludf_getpwuid_init(
            UDF_INIT *initid, UDF_ARGS *args, char *message
            );
	DLLEXP void lib_mysqludf_getpwuid_deinit(UDF_INIT *initid);
	DLLEXP char *lib_mysqludf_getpwuid(UDF_INIT *initid, UDF_ARGS *args, char *result, unsigned long *length, char *is_null, char *error);

#ifdef	__cplusplus
}
#endif


#include <pwd.h>

my_bool lib_mysqludf_getpwuid_init(UDF_INIT *initid, UDF_ARGS *args, char *message)
{
    if ( args->arg_count != 1 )
    {
        strcpy(message, "Incorrect number of arguments to getpwuid");
        return 1;
    }
    if ( args->arg_type[0] != INT_RESULT )
    {
        strcpy(message, "Argument must be integer");
        return 1;
    }
	return 0;
}

void lib_mysqludf_getpwuid_deinit(UDF_INIT *initid)
{
}

/* For functions that return STRING or DECIMAL */ 
char* lib_mysqludf_getpwuid(
        UDF_INIT *initid,
        UDF_ARGS *args,
        char* result,
        unsigned long* length,
        char *is_null,
        char *error
        )
{
    struct passwd pwent;
    struct passwd *pwentp;
    long long uid;
    char buf[1024];
    int s;

    strcpy(result, "unknown");
    uid = *((long long*) args->args[0] );
    s = getpwuid_r( uid, &pwent, buf, sizeof buf, &pwentp );
    if ( pwentp == NULL) {
        strcpy(result, "unknown");
    } else {
        strcpy(result, pwent.pw_name);
    }
    *length = strlen(result);
    return result;
}