pgp-pgsql.c File Reference

#include "postgres.h"
#include "catalog/pg_type.h"
#include "common/string.h"
#include "funcapi.h"
#include "lib/stringinfo.h"
#include "mb/pg_wchar.h"
#include "mbuf.h"
#include "pgp.h"
#include "px.h"
#include "utils/array.h"
#include "utils/builtins.h"

Include dependency graph for pgp-pgsql.c:

Go to the source code of this file.

Data Structures
struct	debug_expect
struct	pgp_armor_headers_state

Macros
#define	EX_MSG(arg)
#define	EX_CHECK(arg)

Functions
	PG_FUNCTION_INFO_V1 (pgp_sym_encrypt_bytea)
	PG_FUNCTION_INFO_V1 (pgp_sym_encrypt_text)
	PG_FUNCTION_INFO_V1 (pgp_sym_decrypt_bytea)
	PG_FUNCTION_INFO_V1 (pgp_sym_decrypt_text)
	PG_FUNCTION_INFO_V1 (pgp_pub_encrypt_bytea)
	PG_FUNCTION_INFO_V1 (pgp_pub_encrypt_text)
	PG_FUNCTION_INFO_V1 (pgp_pub_decrypt_bytea)
	PG_FUNCTION_INFO_V1 (pgp_pub_decrypt_text)
	PG_FUNCTION_INFO_V1 (pgp_key_id_w)
	PG_FUNCTION_INFO_V1 (pg_armor)
	PG_FUNCTION_INFO_V1 (pg_dearmor)
	PG_FUNCTION_INFO_V1 (pgp_armor_headers)
static text *	convert_charset (text *src, int cset_from, int cset_to)
static text *	convert_from_utf8 (text *src)
static text *	convert_to_utf8 (text *src)
static void	clear_and_pfree (text *p)
static void	fill_expect (struct debug_expect *ex, int text_mode)
static void	check_expect (PGP_Context *ctx, struct debug_expect *ex)
static void	show_debug (const char *msg)
static int	set_arg (PGP_Context *ctx, char *key, char *val, struct debug_expect *ex)
static char *	getword (char *p, char **res_p, int *res_len)
static char *	downcase_convert (const uint8 *s, int len)
static int	parse_args (PGP_Context *ctx, uint8 *args, int arg_len, struct debug_expect *ex)
static MBuf *	create_mbuf_from_vardata (text *data)
static void	init_work (PGP_Context **ctx_p, int is_text, text *args, struct debug_expect *ex)
static bytea *	encrypt_internal (int is_pubenc, int is_text, text *data, text *key, text *args)
static bytea *	decrypt_internal (int is_pubenc, int need_text, text *data, text *key, text *keypsw, text *args)
Datum	pgp_sym_encrypt_bytea (PG_FUNCTION_ARGS)
Datum	pgp_sym_encrypt_text (PG_FUNCTION_ARGS)
Datum	pgp_sym_decrypt_bytea (PG_FUNCTION_ARGS)
Datum	pgp_sym_decrypt_text (PG_FUNCTION_ARGS)
Datum	pgp_pub_encrypt_bytea (PG_FUNCTION_ARGS)
Datum	pgp_pub_encrypt_text (PG_FUNCTION_ARGS)
Datum	pgp_pub_decrypt_bytea (PG_FUNCTION_ARGS)
Datum	pgp_pub_decrypt_text (PG_FUNCTION_ARGS)
static int	parse_key_value_arrays (ArrayType *key_array, ArrayType *val_array, char ***p_keys, char ***p_values)
Datum	pg_armor (PG_FUNCTION_ARGS)
Datum	pg_dearmor (PG_FUNCTION_ARGS)
Datum	pgp_armor_headers (PG_FUNCTION_ARGS)
Datum	pgp_key_id_w (PG_FUNCTION_ARGS)

Macro Definition Documentation

EX_CHECK

#define EX_CHECK

(

arg

)

Value:

        do { \
        if (ex->arg >= 0 && ex->arg != ctx->arg) EX_MSG(arg); \
    } while (0)

Definition at line 142 of file pgp-pgsql.c.

EX_MSG

#define EX_MSG

(

arg

)

Value:

    ereport(NOTICE, (errmsg( \
        "pgp_decrypt: unexpected %s: expected %d got %d", \
        CppAsString(arg), ex->arg, ctx->arg)))

Definition at line 137 of file pgp-pgsql.c.

Function Documentation

check_expect()

static void check_expect ( PGP_Context *  ctx, struct debug_expect *  ex  )

static

Definition at line 147 of file pgp-pgsql.c.

{
    EX_CHECK(cipher_algo);
    EX_CHECK(s2k_mode);
    EX_CHECK(s2k_count);
    EX_CHECK(s2k_digest_algo);
    EX_CHECK(use_sess_key);
    if (ctx->use_sess_key)
        EX_CHECK(s2k_cipher_algo);
    EX_CHECK(disable_mdc);
    EX_CHECK(compress_algo);
    EX_CHECK(unicode_mode);
}

References EX_CHECK, and PGP_Context::use_sess_key.

Referenced by decrypt_internal().

clear_and_pfree()

static void clear_and_pfree ( text *  p )

static

Definition at line 97 of file pgp-pgsql.c.

{
    px_memset(p, 0, VARSIZE_ANY(p));
    pfree(p);
}

References pfree(), px_memset(), and VARSIZE_ANY.

Referenced by decrypt_internal(), and encrypt_internal().

convert_charset()

static text* convert_charset ( text *  src, int  cset_from, int  cset_to  )

static

Definition at line 68 of file pgp-pgsql.c.

{
    int         src_len = VARSIZE_ANY_EXHDR(src);
    unsigned char *dst;
    unsigned char *csrc = (unsigned char *) VARDATA_ANY(src);
    text       *res;
 
    dst = pg_do_encoding_conversion(csrc, src_len, cset_from, cset_to);
    if (dst == csrc)
        return src;
 
    res = cstring_to_text((char *) dst);
    pfree(dst);
    return res;
}

References cstring_to_text(), pfree(), pg_do_encoding_conversion(), res, VARDATA_ANY, and VARSIZE_ANY_EXHDR.

Referenced by convert_from_utf8(), and convert_to_utf8().

convert_from_utf8()

static text* convert_from_utf8 ( text *  src )

static

Definition at line 85 of file pgp-pgsql.c.

{
    return convert_charset(src, PG_UTF8, GetDatabaseEncoding());
}

References convert_charset(), GetDatabaseEncoding(), and PG_UTF8.

Referenced by decrypt_internal().

convert_to_utf8()

static text* convert_to_utf8 ( text *  src )

static

Definition at line 91 of file pgp-pgsql.c.

{
    return convert_charset(src, GetDatabaseEncoding(), PG_UTF8);
}

References convert_charset(), GetDatabaseEncoding(), and PG_UTF8.

Referenced by encrypt_internal().

create_mbuf_from_vardata()

static MBuf* create_mbuf_from_vardata ( text *  data )

static

Definition at line 344 of file pgp-pgsql.c.

{
    return mbuf_create_from_data((uint8 *) VARDATA_ANY(data),
                                 VARSIZE_ANY_EXHDR(data));
}

References data, mbuf_create_from_data(), VARDATA_ANY, and VARSIZE_ANY_EXHDR.

Referenced by decrypt_internal(), encrypt_internal(), and pgp_key_id_w().

decrypt_internal()

static bytea* decrypt_internal ( int  is_pubenc, int  need_text, text *  data, text *  key, text *  keypsw, text *  args  )

static

Definition at line 458 of file pgp-pgsql.c.

{
    int         err;
    MBuf       *src = NULL,
               *dst = NULL;
    uint8       tmp[VARHDRSZ];
    uint8      *restmp;
    bytea      *res;
    int         res_len;
    PGP_Context *ctx = NULL;
    struct debug_expect ex;
    int         got_unicode = 0;
 
 
    init_work(&ctx, need_text, args, &ex);
 
    src = mbuf_create_from_data((uint8 *) VARDATA_ANY(data),
                                VARSIZE_ANY_EXHDR(data));
    dst = mbuf_create(VARSIZE_ANY(data) + 2048);
 
    /*
     * reserve room for header
     */
    mbuf_append(dst, tmp, VARHDRSZ);
 
    /*
     * set key
     */
    if (is_pubenc)
    {
        uint8      *psw = NULL;
        int         psw_len = 0;
        MBuf       *kbuf;
 
        if (keypsw)
        {
            psw = (uint8 *) VARDATA_ANY(keypsw);
            psw_len = VARSIZE_ANY_EXHDR(keypsw);
        }
        kbuf = create_mbuf_from_vardata(key);
        err = pgp_set_pubkey(ctx, kbuf, psw, psw_len, 1);
        mbuf_free(kbuf);
    }
    else
        err = pgp_set_symkey(ctx, (uint8 *) VARDATA_ANY(key),
                             VARSIZE_ANY_EXHDR(key));
 
    /* decrypt */
    if (err >= 0)
    {
        err = pgp_decrypt(ctx, src, dst);
 
        if (ex.expect)
            check_expect(ctx, &ex);
 
        /* remember the setting */
        got_unicode = pgp_get_unicode_mode(ctx);
    }
 
    mbuf_free(src);
    pgp_free(ctx);
 
    if (err)
    {
        px_set_debug_handler(NULL);
        mbuf_free(dst);
        px_THROW_ERROR(err);
    }
 
    res_len = mbuf_steal_data(dst, &restmp);
    mbuf_free(dst);
 
    /* res_len includes VARHDRSZ */
    res = (bytea *) restmp;
    SET_VARSIZE(res, res_len);
 
    if (need_text && got_unicode)
    {
        text       *utf = convert_from_utf8(res);
 
        if (utf != res)
        {
            clear_and_pfree(res);
            res = utf;
        }
    }
    px_set_debug_handler(NULL);
 
    return res;
}

References generate_unaccent_rules::args, check_expect(), clear_and_pfree(), convert_from_utf8(), create_mbuf_from_vardata(), data, err(), debug_expect::expect, init_work(), sort-test::key, mbuf_append(), mbuf_create(), mbuf_create_from_data(), mbuf_free(), mbuf_steal_data(), pgp_decrypt(), pgp_free(), pgp_get_unicode_mode(), pgp_set_pubkey(), pgp_set_symkey(), px_set_debug_handler(), px_THROW_ERROR(), res, SET_VARSIZE, VARDATA_ANY, VARHDRSZ, VARSIZE_ANY, and VARSIZE_ANY_EXHDR.

Referenced by pgp_pub_decrypt_bytea(), pgp_pub_decrypt_text(), pgp_sym_decrypt_bytea(), and pgp_sym_decrypt_text().

downcase_convert()

static char* downcase_convert ( const uint8 *  s, int  len  )

static

Definition at line 288 of file pgp-pgsql.c.

{
    int         c,
                i;
    char       *res = palloc(len + 1);
 
    for (i = 0; i < len; i++)
    {
        c = s[i];
        if (c >= 'A' && c <= 'Z')
            c += 'a' - 'A';
        res[i] = c;
    }
    res[len] = 0;
    return res;
}

References i, len, palloc(), and res.

Referenced by parse_args().

encrypt_internal()

static bytea* encrypt_internal ( int  is_pubenc, int  is_text, text *  data, text *  key, text *  args  )

static

Definition at line 372 of file pgp-pgsql.c.

{
    MBuf       *src,
               *dst;
    uint8       tmp[VARHDRSZ];
    uint8      *restmp;
    bytea      *res;
    int         res_len;
    PGP_Context *ctx;
    int         err;
    struct debug_expect ex;
    text       *tmp_data = NULL;
 
    init_work(&ctx, is_text, args, &ex);
 
    if (is_text && pgp_get_unicode_mode(ctx))
    {
        tmp_data = convert_to_utf8(data);
        if (tmp_data == data)
            tmp_data = NULL;
        else
            data = tmp_data;
    }
 
    src = create_mbuf_from_vardata(data);
    dst = mbuf_create(VARSIZE_ANY(data) + 128);
 
    /*
     * reserve room for header
     */
    mbuf_append(dst, tmp, VARHDRSZ);
 
    /*
     * set key
     */
    if (is_pubenc)
    {
        MBuf       *kbuf = create_mbuf_from_vardata(key);
 
        err = pgp_set_pubkey(ctx, kbuf,
                             NULL, 0, 0);
        mbuf_free(kbuf);
    }
    else
        err = pgp_set_symkey(ctx, (uint8 *) VARDATA_ANY(key),
                             VARSIZE_ANY_EXHDR(key));
 
    /*
     * encrypt
     */
    if (err >= 0)
        err = pgp_encrypt(ctx, src, dst);
 
    /*
     * check for error
     */
    if (err)
    {
        if (ex.debug)
            px_set_debug_handler(NULL);
        if (tmp_data)
            clear_and_pfree(tmp_data);
        pgp_free(ctx);
        mbuf_free(src);
        mbuf_free(dst);
        px_THROW_ERROR(err);
    }
 
    /* res_len includes VARHDRSZ */
    res_len = mbuf_steal_data(dst, &restmp);
    res = (bytea *) restmp;
    SET_VARSIZE(res, res_len);
 
    if (tmp_data)
        clear_and_pfree(tmp_data);
    pgp_free(ctx);
    mbuf_free(src);
    mbuf_free(dst);
 
    px_set_debug_handler(NULL);
 
    return res;
}

References generate_unaccent_rules::args, clear_and_pfree(), convert_to_utf8(), create_mbuf_from_vardata(), data, debug_expect::debug, err(), init_work(), sort-test::key, mbuf_append(), mbuf_create(), mbuf_free(), mbuf_steal_data(), pgp_encrypt(), pgp_free(), pgp_get_unicode_mode(), pgp_set_pubkey(), pgp_set_symkey(), px_set_debug_handler(), px_THROW_ERROR(), res, SET_VARSIZE, VARDATA_ANY, VARHDRSZ, VARSIZE_ANY, and VARSIZE_ANY_EXHDR.

Referenced by pgp_pub_encrypt_bytea(), pgp_pub_encrypt_text(), pgp_sym_encrypt_bytea(), and pgp_sym_encrypt_text().

fill_expect()

static void fill_expect ( struct debug_expect *  ex, int  text_mode  )

static

Definition at line 122 of file pgp-pgsql.c.

{
    ex->debug = 0;
    ex->expect = 0;
    ex->cipher_algo = -1;
    ex->s2k_mode = -1;
    ex->s2k_count = -1;
    ex->s2k_cipher_algo = -1;
    ex->s2k_digest_algo = -1;
    ex->compress_algo = -1;
    ex->use_sess_key = -1;
    ex->disable_mdc = -1;
    ex->unicode_mode = -1;
}

References debug_expect::cipher_algo, debug_expect::compress_algo, debug_expect::debug, debug_expect::disable_mdc, debug_expect::expect, debug_expect::s2k_cipher_algo, debug_expect::s2k_count, debug_expect::s2k_digest_algo, debug_expect::s2k_mode, debug_expect::unicode_mode, and debug_expect::use_sess_key.

Referenced by init_work().

getword()

static char* getword ( char *  p, char **  res_p, int *  res_len  )

static

Definition at line 259 of file pgp-pgsql.c.

{
    /* whitespace at start */
    while (*p && (*p == ' ' || *p == '\t' || *p == '\n'))
        p++;
 
    /* word data */
    *res_p = p;
    if (*p == '=' || *p == ',')
        p++;
    else
        while (*p && !(*p == ' ' || *p == '\t' || *p == '\n'
                       || *p == '=' || *p == ','))
            p++;
 
    /* word end */
    *res_len = p - *res_p;
 
    /* whitespace at end */
    while (*p && (*p == ' ' || *p == '\t' || *p == '\n'))
        p++;
 
    return p;
}

Referenced by parse_args().

init_work()

static void init_work ( PGP_Context **  ctx_p, int  is_text, text *  args, struct debug_expect *  ex  )

static

Definition at line 351 of file pgp-pgsql.c.

{
    int         err = pgp_init(ctx_p);
 
    fill_expect(ex, is_text);
 
    if (err == 0 && args != NULL)
        err = parse_args(*ctx_p, (uint8 *) VARDATA_ANY(args),
                         VARSIZE_ANY_EXHDR(args), ex);
 
    if (err)
        px_THROW_ERROR(err);
 
    if (ex->debug)
        px_set_debug_handler(show_debug);
 
    pgp_set_text_mode(*ctx_p, is_text);
}

References generate_unaccent_rules::args, debug_expect::debug, err(), fill_expect(), parse_args(), pgp_init(), pgp_set_text_mode(), px_set_debug_handler(), px_THROW_ERROR(), show_debug(), VARDATA_ANY, and VARSIZE_ANY_EXHDR.

Referenced by decrypt_internal(), and encrypt_internal().

parse_args()

static int parse_args ( PGP_Context *  ctx, uint8 *  args, int  arg_len, struct debug_expect *  ex  )

static

Definition at line 306 of file pgp-pgsql.c.

{
    char       *str = downcase_convert(args, arg_len);
    char       *key,
               *val;
    int         key_len,
                val_len;
    int         res = 0;
    char       *p = str;
 
    while (*p)
    {
        res = PXE_ARGUMENT_ERROR;
        p = getword(p, &key, &key_len);
        if (*p++ != '=')
            break;
        p = getword(p, &val, &val_len);
        if (*p == '\0')
            ;
        else if (*p++ != ',')
            break;
 
        if (*key == 0 || *val == 0 || val_len == 0)
            break;
 
        key[key_len] = 0;
        val[val_len] = 0;
 
        res = set_arg(ctx, key, val, ex);
        if (res < 0)
            break;
    }
    pfree(str);
    return res;
}

References generate_unaccent_rules::args, downcase_convert(), getword(), sort-test::key, pfree(), PXE_ARGUMENT_ERROR, res, set_arg(), str, and val.

Referenced by init_work().

parse_key_value_arrays()

static int parse_key_value_arrays ( ArrayType *  key_array, ArrayType *  val_array, char ***  p_keys, char ***  p_values  )

static

Definition at line 755 of file pgp-pgsql.c.

{
    int         nkdims = ARR_NDIM(key_array);
    int         nvdims = ARR_NDIM(val_array);
    char      **keys,
              **values;
    Datum      *key_datums,
               *val_datums;
    bool       *key_nulls,
               *val_nulls;
    int         key_count,
                val_count;
    int         i;
 
    if (nkdims > 1 || nkdims != nvdims)
        ereport(ERROR,
                (errcode(ERRCODE_ARRAY_SUBSCRIPT_ERROR),
                 errmsg("wrong number of array subscripts")));
    if (nkdims == 0)
        return 0;
 
    deconstruct_array_builtin(key_array, TEXTOID, &key_datums, &key_nulls, &key_count);
    deconstruct_array_builtin(val_array, TEXTOID, &val_datums, &val_nulls, &val_count);
 
    if (key_count != val_count)
        ereport(ERROR,
                (errcode(ERRCODE_ARRAY_SUBSCRIPT_ERROR),
                 errmsg("mismatched array dimensions")));
 
    keys = (char **) palloc(sizeof(char *) * key_count);
    values = (char **) palloc(sizeof(char *) * val_count);
 
    for (i = 0; i < key_count; i++)
    {
        char       *v;
 
        /* Check that the key doesn't contain anything funny */
        if (key_nulls[i])
            ereport(ERROR,
                    (errcode(ERRCODE_NULL_VALUE_NOT_ALLOWED),
                     errmsg("null value not allowed for header key")));
 
        v = TextDatumGetCString(key_datums[i]);
 
        if (!pg_is_ascii(v))
            ereport(ERROR,
                    (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
                     errmsg("header key must not contain non-ASCII characters")));
        if (strstr(v, ": "))
            ereport(ERROR,
                    (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
                     errmsg("header key must not contain \": \"")));
        if (strchr(v, '\n'))
            ereport(ERROR,
                    (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
                     errmsg("header key must not contain newlines")));
        keys[i] = v;
 
        /* And the same for the value */
        if (val_nulls[i])
            ereport(ERROR,
                    (errcode(ERRCODE_NULL_VALUE_NOT_ALLOWED),
                     errmsg("null value not allowed for header value")));
 
        v = TextDatumGetCString(val_datums[i]);
 
        if (!pg_is_ascii(v))
            ereport(ERROR,
                    (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
                     errmsg("header value must not contain non-ASCII characters")));
        if (strchr(v, '\n'))
            ereport(ERROR,
                    (errcode(ERRCODE_INVALID_PARAMETER_VALUE),
                     errmsg("header value must not contain newlines")));
 
        values[i] = v;
    }
 
    *p_keys = keys;
    *p_values = values;
    return key_count;
}

References ARR_NDIM, deconstruct_array_builtin(), ereport, errcode(), errmsg(), ERROR, i, palloc(), pg_is_ascii(), TextDatumGetCString, and values.

Referenced by pg_armor().

pg_armor()

Datum pg_armor

(

PG_FUNCTION_ARGS

)

Definition at line 840 of file pgp-pgsql.c.

{
    bytea      *data;
    text       *res;
    int         data_len;
    StringInfoData buf;
    int         num_headers;
    char      **keys = NULL,
              **values = NULL;
 
    data = PG_GETARG_BYTEA_PP(0);
    data_len = VARSIZE_ANY_EXHDR(data);
    if (PG_NARGS() == 3)
    {
        num_headers = parse_key_value_arrays(PG_GETARG_ARRAYTYPE_P(1),
                                             PG_GETARG_ARRAYTYPE_P(2),
                                             &keys, &values);
    }
    else if (PG_NARGS() == 1)
        num_headers = 0;
    else
        elog(ERROR, "unexpected number of arguments %d", PG_NARGS());
 
    initStringInfo(&buf);
 
    pgp_armor_encode((uint8 *) VARDATA_ANY(data), data_len, &buf,
                     num_headers, keys, values);
 
    res = palloc(VARHDRSZ + buf.len);
    SET_VARSIZE(res, VARHDRSZ + buf.len);
    memcpy(VARDATA(res), buf.data, buf.len);
    pfree(buf.data);
 
    PG_FREE_IF_COPY(data, 0);
    PG_RETURN_TEXT_P(res);
}

References buf, data, elog, ERROR, initStringInfo(), palloc(), parse_key_value_arrays(), pfree(), PG_FREE_IF_COPY, PG_GETARG_ARRAYTYPE_P, PG_GETARG_BYTEA_PP, PG_NARGS, PG_RETURN_TEXT_P, pgp_armor_encode(), res, SET_VARSIZE, values, VARDATA, VARDATA_ANY, VARHDRSZ, and VARSIZE_ANY_EXHDR.

pg_dearmor()

Datum pg_dearmor

(

PG_FUNCTION_ARGS

)

Definition at line 878 of file pgp-pgsql.c.

{
    text       *data;
    bytea      *res;
    int         data_len;
    int         ret;
    StringInfoData buf;
 
    data = PG_GETARG_TEXT_PP(0);
    data_len = VARSIZE_ANY_EXHDR(data);
 
    initStringInfo(&buf);
 
    ret = pgp_armor_decode((uint8 *) VARDATA_ANY(data), data_len, &buf);
    if (ret < 0)
        px_THROW_ERROR(ret);
    res = palloc(VARHDRSZ + buf.len);
    SET_VARSIZE(res, VARHDRSZ + buf.len);
    memcpy(VARDATA(res), buf.data, buf.len);
    pfree(buf.data);
 
    PG_FREE_IF_COPY(data, 0);
    PG_RETURN_TEXT_P(res);
}

References buf, data, initStringInfo(), palloc(), pfree(), PG_FREE_IF_COPY, PG_GETARG_TEXT_PP, PG_RETURN_TEXT_P, pgp_armor_decode(), px_THROW_ERROR(), res, SET_VARSIZE, VARDATA, VARDATA_ANY, VARHDRSZ, and VARSIZE_ANY_EXHDR.

PG_FUNCTION_INFO_V1() [1/12]

PG_FUNCTION_INFO_V1

(

pg_armor

)

PG_FUNCTION_INFO_V1() [2/12]

PG_FUNCTION_INFO_V1

(

pg_dearmor

)

PG_FUNCTION_INFO_V1() [3/12]

PG_FUNCTION_INFO_V1

(

pgp_armor_headers

)

PG_FUNCTION_INFO_V1() [4/12]

PG_FUNCTION_INFO_V1

(

pgp_key_id_w

)

PG_FUNCTION_INFO_V1() [5/12]

PG_FUNCTION_INFO_V1

(

pgp_pub_decrypt_bytea

)

PG_FUNCTION_INFO_V1() [6/12]

PG_FUNCTION_INFO_V1

(

pgp_pub_decrypt_text

)

PG_FUNCTION_INFO_V1() [7/12]

PG_FUNCTION_INFO_V1

(

pgp_pub_encrypt_bytea

)

PG_FUNCTION_INFO_V1() [8/12]

PG_FUNCTION_INFO_V1

(

pgp_pub_encrypt_text

)

PG_FUNCTION_INFO_V1() [9/12]

PG_FUNCTION_INFO_V1

(

pgp_sym_decrypt_bytea

)

PG_FUNCTION_INFO_V1() [10/12]

PG_FUNCTION_INFO_V1

(

pgp_sym_decrypt_text

)

PG_FUNCTION_INFO_V1() [11/12]

PG_FUNCTION_INFO_V1

(

pgp_sym_encrypt_bytea

)

PG_FUNCTION_INFO_V1() [12/12]

PG_FUNCTION_INFO_V1

(

pgp_sym_encrypt_text

)

pgp_armor_headers()

Datum pgp_armor_headers

(

PG_FUNCTION_ARGS

)

Definition at line 912 of file pgp-pgsql.c.

{
    FuncCallContext *funcctx;
    pgp_armor_headers_state *state;
    char       *utf8key;
    char       *utf8val;
    HeapTuple   tuple;
    TupleDesc   tupdesc;
    AttInMetadata *attinmeta;
 
    if (SRF_IS_FIRSTCALL())
    {
        text       *data = PG_GETARG_TEXT_PP(0);
        int         res;
        MemoryContext oldcontext;
 
        funcctx = SRF_FIRSTCALL_INIT();
 
        /* we need the state allocated in the multi call context */
        oldcontext = MemoryContextSwitchTo(funcctx->multi_call_memory_ctx);
 
        /* Build a tuple descriptor for our result type */
        if (get_call_result_type(fcinfo, NULL, &tupdesc) != TYPEFUNC_COMPOSITE)
            elog(ERROR, "return type must be a row type");
 
        attinmeta = TupleDescGetAttInMetadata(tupdesc);
        funcctx->attinmeta = attinmeta;
 
        state = (pgp_armor_headers_state *) palloc(sizeof(pgp_armor_headers_state));
 
        res = pgp_extract_armor_headers((uint8 *) VARDATA_ANY(data),
                                        VARSIZE_ANY_EXHDR(data),
                                        &state->nheaders, &state->keys,
                                        &state->values);
        if (res < 0)
            px_THROW_ERROR(res);
 
        MemoryContextSwitchTo(oldcontext);
        funcctx->user_fctx = state;
    }
 
    funcctx = SRF_PERCALL_SETUP();
    state = (pgp_armor_headers_state *) funcctx->user_fctx;
 
    if (funcctx->call_cntr >= state->nheaders)
        SRF_RETURN_DONE(funcctx);
    else
    {
        char       *values[2];
 
        /* we assume that the keys (and values) are in UTF-8. */
        utf8key = state->keys[funcctx->call_cntr];
        utf8val = state->values[funcctx->call_cntr];
 
        values[0] = pg_any_to_server(utf8key, strlen(utf8key), PG_UTF8);
        values[1] = pg_any_to_server(utf8val, strlen(utf8val), PG_UTF8);
 
        /* build a tuple */
        tuple = BuildTupleFromCStrings(funcctx->attinmeta, values);
        SRF_RETURN_NEXT(funcctx, HeapTupleGetDatum(tuple));
    }
}

References FuncCallContext::attinmeta, BuildTupleFromCStrings(), FuncCallContext::call_cntr, data, elog, ERROR, get_call_result_type(), HeapTupleGetDatum(), if(), MemoryContextSwitchTo(), FuncCallContext::multi_call_memory_ctx, palloc(), pg_any_to_server(), PG_GETARG_TEXT_PP, PG_UTF8, pgp_extract_armor_headers(), px_THROW_ERROR(), res, SRF_FIRSTCALL_INIT, SRF_IS_FIRSTCALL, SRF_PERCALL_SETUP, SRF_RETURN_DONE, SRF_RETURN_NEXT, TupleDescGetAttInMetadata(), TYPEFUNC_COMPOSITE, FuncCallContext::user_fctx, values, VARDATA_ANY, and VARSIZE_ANY_EXHDR.

pgp_key_id_w()

Datum pgp_key_id_w

(

PG_FUNCTION_ARGS

)

Definition at line 982 of file pgp-pgsql.c.

{
    bytea      *data;
    text       *res;
    int         res_len;
    MBuf       *buf;
 
    data = PG_GETARG_BYTEA_PP(0);
    buf = create_mbuf_from_vardata(data);
    res = palloc(VARHDRSZ + 17);
 
    res_len = pgp_get_keyid(buf, VARDATA(res));
    mbuf_free(buf);
    if (res_len < 0)
        px_THROW_ERROR(res_len);
    SET_VARSIZE(res, VARHDRSZ + res_len);
 
    PG_FREE_IF_COPY(data, 0);
    PG_RETURN_TEXT_P(res);
}

References buf, create_mbuf_from_vardata(), data, mbuf_free(), palloc(), PG_FREE_IF_COPY, PG_GETARG_BYTEA_PP, PG_RETURN_TEXT_P, pgp_get_keyid(), px_THROW_ERROR(), res, SET_VARSIZE, VARDATA, and VARHDRSZ.

pgp_pub_decrypt_bytea()

Datum pgp_pub_decrypt_bytea

(

PG_FUNCTION_ARGS

)

Definition at line 692 of file pgp-pgsql.c.

{
    bytea      *data,
               *key;
    text       *psw = NULL,
               *arg = NULL;
    text       *res;
 
    data = PG_GETARG_BYTEA_PP(0);
    key = PG_GETARG_BYTEA_PP(1);
    if (PG_NARGS() > 2)
        psw = PG_GETARG_TEXT_PP(2);
    if (PG_NARGS() > 3)
        arg = PG_GETARG_TEXT_PP(3);
 
    res = decrypt_internal(1, 0, data, key, psw, arg);
 
    PG_FREE_IF_COPY(data, 0);
    PG_FREE_IF_COPY(key, 1);
    if (PG_NARGS() > 2)
        PG_FREE_IF_COPY(psw, 2);
    if (PG_NARGS() > 3)
        PG_FREE_IF_COPY(arg, 3);
    PG_RETURN_TEXT_P(res);
}

References arg, data, decrypt_internal(), sort-test::key, PG_FREE_IF_COPY, PG_GETARG_BYTEA_PP, PG_GETARG_TEXT_PP, PG_NARGS, PG_RETURN_TEXT_P, and res.

pgp_pub_decrypt_text()

Datum pgp_pub_decrypt_text

(

PG_FUNCTION_ARGS

)

Definition at line 719 of file pgp-pgsql.c.

{
    bytea      *data,
               *key;
    text       *psw = NULL,
               *arg = NULL;
    text       *res;
 
    data = PG_GETARG_BYTEA_PP(0);
    key = PG_GETARG_BYTEA_PP(1);
    if (PG_NARGS() > 2)
        psw = PG_GETARG_TEXT_PP(2);
    if (PG_NARGS() > 3)
        arg = PG_GETARG_TEXT_PP(3);
 
    res = decrypt_internal(1, 1, data, key, psw, arg);
 
    PG_FREE_IF_COPY(data, 0);
    PG_FREE_IF_COPY(key, 1);
    if (PG_NARGS() > 2)
        PG_FREE_IF_COPY(psw, 2);
    if (PG_NARGS() > 3)
        PG_FREE_IF_COPY(arg, 3);
    PG_RETURN_TEXT_P(res);
}

References arg, data, decrypt_internal(), sort-test::key, PG_FREE_IF_COPY, PG_GETARG_BYTEA_PP, PG_GETARG_TEXT_PP, PG_NARGS, PG_RETURN_TEXT_P, and res.

pgp_pub_encrypt_bytea()

Datum pgp_pub_encrypt_bytea

(

PG_FUNCTION_ARGS

)

Definition at line 647 of file pgp-pgsql.c.

{
    bytea      *data,
               *key;
    text       *arg = NULL;
    text       *res;
 
    data = PG_GETARG_BYTEA_PP(0);
    key = PG_GETARG_BYTEA_PP(1);
    if (PG_NARGS() > 2)
        arg = PG_GETARG_TEXT_PP(2);
 
    res = encrypt_internal(1, 0, data, key, arg);
 
    PG_FREE_IF_COPY(data, 0);
    PG_FREE_IF_COPY(key, 1);
    if (PG_NARGS() > 2)
        PG_FREE_IF_COPY(arg, 2);
    PG_RETURN_TEXT_P(res);
}

References arg, data, encrypt_internal(), sort-test::key, PG_FREE_IF_COPY, PG_GETARG_BYTEA_PP, PG_GETARG_TEXT_PP, PG_NARGS, PG_RETURN_TEXT_P, and res.

pgp_pub_encrypt_text()

Datum pgp_pub_encrypt_text

(

PG_FUNCTION_ARGS

)

Definition at line 669 of file pgp-pgsql.c.

{
    bytea      *key;
    text       *arg = NULL;
    text       *res,
               *data;
 
    data = PG_GETARG_TEXT_PP(0);
    key = PG_GETARG_BYTEA_PP(1);
    if (PG_NARGS() > 2)
        arg = PG_GETARG_TEXT_PP(2);
 
    res = encrypt_internal(1, 1, data, key, arg);
 
    PG_FREE_IF_COPY(data, 0);
    PG_FREE_IF_COPY(key, 1);
    if (PG_NARGS() > 2)
        PG_FREE_IF_COPY(arg, 2);
    PG_RETURN_TEXT_P(res);
}

References arg, data, encrypt_internal(), sort-test::key, PG_FREE_IF_COPY, PG_GETARG_BYTEA_PP, PG_GETARG_TEXT_PP, PG_NARGS, PG_RETURN_TEXT_P, and res.

pgp_sym_decrypt_bytea()

Datum pgp_sym_decrypt_bytea

(

PG_FUNCTION_ARGS

)

Definition at line 599 of file pgp-pgsql.c.

{
    bytea      *data;
    text       *arg = NULL;
    text       *res,
               *key;
 
    data = PG_GETARG_BYTEA_PP(0);
    key = PG_GETARG_TEXT_PP(1);
    if (PG_NARGS() > 2)
        arg = PG_GETARG_TEXT_PP(2);
 
    res = decrypt_internal(0, 0, data, key, NULL, arg);
 
    PG_FREE_IF_COPY(data, 0);
    PG_FREE_IF_COPY(key, 1);
    if (PG_NARGS() > 2)
        PG_FREE_IF_COPY(arg, 2);
    PG_RETURN_TEXT_P(res);
}

References arg, data, decrypt_internal(), sort-test::key, PG_FREE_IF_COPY, PG_GETARG_BYTEA_PP, PG_GETARG_TEXT_PP, PG_NARGS, PG_RETURN_TEXT_P, and res.

pgp_sym_decrypt_text()

Datum pgp_sym_decrypt_text

(

PG_FUNCTION_ARGS

)

Definition at line 621 of file pgp-pgsql.c.

{
    bytea      *data;
    text       *arg = NULL;
    text       *res,
               *key;
 
    data = PG_GETARG_BYTEA_PP(0);
    key = PG_GETARG_TEXT_PP(1);
    if (PG_NARGS() > 2)
        arg = PG_GETARG_TEXT_PP(2);
 
    res = decrypt_internal(0, 1, data, key, NULL, arg);
 
    PG_FREE_IF_COPY(data, 0);
    PG_FREE_IF_COPY(key, 1);
    if (PG_NARGS() > 2)
        PG_FREE_IF_COPY(arg, 2);
    PG_RETURN_TEXT_P(res);
}

References arg, data, decrypt_internal(), sort-test::key, PG_FREE_IF_COPY, PG_GETARG_BYTEA_PP, PG_GETARG_TEXT_PP, PG_NARGS, PG_RETURN_TEXT_P, and res.

pgp_sym_encrypt_bytea()

Datum pgp_sym_encrypt_bytea

(

PG_FUNCTION_ARGS

)

Definition at line 554 of file pgp-pgsql.c.

{
    bytea      *data;
    text       *arg = NULL;
    text       *res,
               *key;
 
    data = PG_GETARG_BYTEA_PP(0);
    key = PG_GETARG_TEXT_PP(1);
    if (PG_NARGS() > 2)
        arg = PG_GETARG_TEXT_PP(2);
 
    res = encrypt_internal(0, 0, data, key, arg);
 
    PG_FREE_IF_COPY(data, 0);
    PG_FREE_IF_COPY(key, 1);
    if (PG_NARGS() > 2)
        PG_FREE_IF_COPY(arg, 2);
    PG_RETURN_TEXT_P(res);
}

References arg, data, encrypt_internal(), sort-test::key, PG_FREE_IF_COPY, PG_GETARG_BYTEA_PP, PG_GETARG_TEXT_PP, PG_NARGS, PG_RETURN_TEXT_P, and res.

pgp_sym_encrypt_text()

Datum pgp_sym_encrypt_text

(

PG_FUNCTION_ARGS

)

Definition at line 576 of file pgp-pgsql.c.

{
    text       *data,
               *key;
    text       *arg = NULL;
    text       *res;
 
    data = PG_GETARG_TEXT_PP(0);
    key = PG_GETARG_TEXT_PP(1);
    if (PG_NARGS() > 2)
        arg = PG_GETARG_TEXT_PP(2);
 
    res = encrypt_internal(0, 1, data, key, arg);
 
    PG_FREE_IF_COPY(data, 0);
    PG_FREE_IF_COPY(key, 1);
    if (PG_NARGS() > 2)
        PG_FREE_IF_COPY(arg, 2);
    PG_RETURN_TEXT_P(res);
}

References arg, data, encrypt_internal(), sort-test::key, PG_FREE_IF_COPY, PG_GETARG_TEXT_PP, PG_NARGS, PG_RETURN_TEXT_P, and res.

set_arg()

static int set_arg ( PGP_Context *  ctx, char *  key, char *  val, struct debug_expect *  ex  )

static

Definition at line 168 of file pgp-pgsql.c.

{
    int         res = 0;
 
    if (strcmp(key, "cipher-algo") == 0)
        res = pgp_set_cipher_algo(ctx, val);
    else if (strcmp(key, "disable-mdc") == 0)
        res = pgp_disable_mdc(ctx, atoi(val));
    else if (strcmp(key, "sess-key") == 0)
        res = pgp_set_sess_key(ctx, atoi(val));
    else if (strcmp(key, "s2k-mode") == 0)
        res = pgp_set_s2k_mode(ctx, atoi(val));
    else if (strcmp(key, "s2k-count") == 0)
        res = pgp_set_s2k_count(ctx, atoi(val));
    else if (strcmp(key, "s2k-digest-algo") == 0)
        res = pgp_set_s2k_digest_algo(ctx, val);
    else if (strcmp(key, "s2k-cipher-algo") == 0)
        res = pgp_set_s2k_cipher_algo(ctx, val);
    else if (strcmp(key, "compress-algo") == 0)
        res = pgp_set_compress_algo(ctx, atoi(val));
    else if (strcmp(key, "compress-level") == 0)
        res = pgp_set_compress_level(ctx, atoi(val));
    else if (strcmp(key, "convert-crlf") == 0)
        res = pgp_set_convert_crlf(ctx, atoi(val));
    else if (strcmp(key, "unicode-mode") == 0)
        res = pgp_set_unicode_mode(ctx, atoi(val));
 
    /*
     * The remaining options are for debugging/testing and are therefore not
     * documented in the user-facing docs.
     */
    else if (ex != NULL && strcmp(key, "debug") == 0)
        ex->debug = atoi(val);
    else if (ex != NULL && strcmp(key, "expect-cipher-algo") == 0)
    {
        ex->expect = 1;
        ex->cipher_algo = pgp_get_cipher_code(val);
    }
    else if (ex != NULL && strcmp(key, "expect-disable-mdc") == 0)
    {
        ex->expect = 1;
        ex->disable_mdc = atoi(val);
    }
    else if (ex != NULL && strcmp(key, "expect-sess-key") == 0)
    {
        ex->expect = 1;
        ex->use_sess_key = atoi(val);
    }
    else if (ex != NULL && strcmp(key, "expect-s2k-mode") == 0)
    {
        ex->expect = 1;
        ex->s2k_mode = atoi(val);
    }
    else if (ex != NULL && strcmp(key, "expect-s2k-count") == 0)
    {
        ex->expect = 1;
        ex->s2k_count = atoi(val);
    }
    else if (ex != NULL && strcmp(key, "expect-s2k-digest-algo") == 0)
    {
        ex->expect = 1;
        ex->s2k_digest_algo = pgp_get_digest_code(val);
    }
    else if (ex != NULL && strcmp(key, "expect-s2k-cipher-algo") == 0)
    {
        ex->expect = 1;
        ex->s2k_cipher_algo = pgp_get_cipher_code(val);
    }
    else if (ex != NULL && strcmp(key, "expect-compress-algo") == 0)
    {
        ex->expect = 1;
        ex->compress_algo = atoi(val);
    }
    else if (ex != NULL && strcmp(key, "expect-unicode-mode") == 0)
    {
        ex->expect = 1;
        ex->unicode_mode = atoi(val);
    }
    else
        res = PXE_ARGUMENT_ERROR;
 
    return res;
}

References debug_expect::cipher_algo, debug_expect::compress_algo, debug_expect::debug, debug_expect::disable_mdc, debug_expect::expect, sort-test::key, pgp_disable_mdc(), pgp_get_cipher_code(), pgp_get_digest_code(), pgp_set_cipher_algo(), pgp_set_compress_algo(), pgp_set_compress_level(), pgp_set_convert_crlf(), pgp_set_s2k_cipher_algo(), pgp_set_s2k_count(), pgp_set_s2k_digest_algo(), pgp_set_s2k_mode(), pgp_set_sess_key(), pgp_set_unicode_mode(), PXE_ARGUMENT_ERROR, res, debug_expect::s2k_cipher_algo, debug_expect::s2k_count, debug_expect::s2k_digest_algo, debug_expect::s2k_mode, debug_expect::unicode_mode, debug_expect::use_sess_key, and val.

Referenced by parse_args().

show_debug()

static void show_debug ( const char *  msg )

static

Definition at line 162 of file pgp-pgsql.c.

{
    ereport(NOTICE, (errmsg("dbg: %s", msg)));
}

References ereport, errmsg(), and NOTICE.

Referenced by init_work().