mbuf.c File Reference

#include "postgres.h"
#include "mbuf.h"
#include "px.h"

Include dependency graph for mbuf.c:

Go to the source code of this file.

Data Structures
struct	MBuf
struct	PullFilter
struct	PushFilter

Macros
#define	STEP   (16*1024)

Functions
int	mbuf_avail (MBuf *mbuf)
int	mbuf_size (MBuf *mbuf)
int	mbuf_free (MBuf *mbuf)
static void	prepare_room (MBuf *mbuf, int block_len)
int	mbuf_append (MBuf *dst, const uint8 *buf, int len)
MBuf *	mbuf_create (int len)
MBuf *	mbuf_create_from_data (uint8 *data, int len)
int	mbuf_grab (MBuf *mbuf, int len, uint8 **data_p)
int	mbuf_steal_data (MBuf *mbuf, uint8 **data_p)
int	pullf_create (PullFilter **pf_p, const PullFilterOps *op, void *init_arg, PullFilter *src)
void	pullf_free (PullFilter *pf)
int	pullf_read (PullFilter *pf, int len, uint8 **data_p)
int	pullf_read_max (PullFilter *pf, int len, uint8 **data_p, uint8 *tmpbuf)
int	pullf_read_fixed (PullFilter *src, int len, uint8 *dst)
static int	pull_from_mbuf (void *arg, PullFilter *src, int len, uint8 **data_p, uint8 *buf, int buflen)
int	pullf_create_mbuf_reader (PullFilter **mp_p, MBuf *src)
int	pushf_create (PushFilter **mp_p, const PushFilterOps *op, void *init_arg, PushFilter *next)
void	pushf_free (PushFilter *mp)
void	pushf_free_all (PushFilter *mp)
static int	wrap_process (PushFilter *mp, const uint8 *data, int len)
int	pushf_write (PushFilter *mp, const uint8 *data, int len)
int	pushf_flush (PushFilter *mp)
static int	push_into_mbuf (PushFilter *next, void *arg, const uint8 *data, int len)
int	pushf_create_mbuf_writer (PushFilter **res, MBuf *dst)

Variables
static const struct PullFilterOps	mbuf_reader
static const struct PushFilterOps	mbuf_filter

Macro Definition Documentation

STEP

#define STEP   (16*1024)

Definition at line 37 of file mbuf.c.

Function Documentation

mbuf_append()

int mbuf_append	(	MBuf *	dst,
		const uint8 *	buf,
		int	len
	)

Definition at line 94 of file mbuf.c.

{
    if (dst->no_write)
    {
        px_debug("mbuf_append: no_write");
        return PXE_BUG;
    }
 
    prepare_room(dst, len);
 
    memcpy(dst->data_end, buf, len);
    dst->data_end += len;
 
    return 0;
}

References buf, MBuf::data_end, len, MBuf::no_write, prepare_room(), px_debug(), and PXE_BUG.

Referenced by copy_crlf(), decrypt_internal(), encrypt_internal(), parse_literal_data(), and push_into_mbuf().

mbuf_avail()

int mbuf_avail

(

MBuf *

mbuf

)

Definition at line 50 of file mbuf.c.

{
    return mbuf->data_end - mbuf->read_pos;
}

References MBuf::data_end, and MBuf::read_pos.

Referenced by mbuf_grab(), and pgp_encrypt().

mbuf_create()

MBuf * mbuf_create

(

int

len

)

Definition at line 111 of file mbuf.c.

{
    MBuf       *mbuf;
 
    if (!len)
        len = 8192;
 
    mbuf = palloc(sizeof *mbuf);
    mbuf->data = palloc(len);
    mbuf->buf_end = mbuf->data + len;
    mbuf->data_end = mbuf->data;
    mbuf->read_pos = mbuf->data;
 
    mbuf->no_write = false;
    mbuf->own_data = true;
 
    return mbuf;
}

References MBuf::buf_end, MBuf::data, MBuf::data_end, len, MBuf::no_write, MBuf::own_data, palloc(), and MBuf::read_pos.

Referenced by decrypt_internal(), and encrypt_internal().

mbuf_create_from_data()

MBuf * mbuf_create_from_data	(	uint8 *	data,
		int	len
	)

Definition at line 131 of file mbuf.c.

{
    MBuf       *mbuf;
 
    mbuf = palloc(sizeof *mbuf);
    mbuf->data = (uint8 *) data;
    mbuf->buf_end = mbuf->data + len;
    mbuf->data_end = mbuf->data + len;
    mbuf->read_pos = mbuf->data;
 
    mbuf->no_write = true;
    mbuf->own_data = false;
 
    return mbuf;
}

References MBuf::buf_end, MBuf::data, data, MBuf::data_end, len, MBuf::no_write, MBuf::own_data, palloc(), and MBuf::read_pos.

Referenced by create_mbuf_from_vardata(), and decrypt_internal().

mbuf_free()

int mbuf_free

(

MBuf *

mbuf

)

Definition at line 62 of file mbuf.c.

{
    if (mbuf->own_data)
    {
        px_memset(mbuf->data, 0, mbuf->buf_end - mbuf->data);
        pfree(mbuf->data);
    }
    pfree(mbuf);
    return 0;
}

References MBuf::buf_end, MBuf::data, MBuf::own_data, pfree(), and px_memset().

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

mbuf_grab()

int mbuf_grab	(	MBuf *	mbuf,
		int	len,
		uint8 **	data_p
	)

Definition at line 149 of file mbuf.c.

{
    if (len > mbuf_avail(mbuf))
        len = mbuf_avail(mbuf);
 
    mbuf->no_write = true;
 
    *data_p = mbuf->read_pos;
    mbuf->read_pos += len;
    return len;
}

References len, mbuf_avail(), MBuf::no_write, and MBuf::read_pos.

Referenced by pgp_encrypt(), and pull_from_mbuf().

mbuf_size()

int mbuf_size

(

MBuf *

mbuf

)

Definition at line 56 of file mbuf.c.

{
    return mbuf->data_end - mbuf->data;
}

References MBuf::data, and MBuf::data_end.

Referenced by mbuf_steal_data().

mbuf_steal_data()

int mbuf_steal_data	(	MBuf *	mbuf,
		uint8 **	data_p
	)

Definition at line 162 of file mbuf.c.

{
    int         len = mbuf_size(mbuf);
 
    mbuf->no_write = true;
    mbuf->own_data = false;
 
    *data_p = mbuf->data;
 
    mbuf->data = mbuf->data_end = mbuf->read_pos = mbuf->buf_end = NULL;
 
    return len;
}

References MBuf::buf_end, MBuf::data, MBuf::data_end, len, mbuf_size(), MBuf::no_write, MBuf::own_data, and MBuf::read_pos.

Referenced by decrypt_internal(), and encrypt_internal().

prepare_room()

static void prepare_room ( MBuf *  mbuf, int  block_len  )

static

Definition at line 74 of file mbuf.c.

{
    uint8      *newbuf;
    unsigned    newlen;
 
    if (mbuf->data_end + block_len <= mbuf->buf_end)
        return;
 
    newlen = (mbuf->buf_end - mbuf->data)
        + ((block_len + STEP + STEP - 1) & -STEP);
 
    newbuf = repalloc(mbuf->data, newlen);
 
    mbuf->buf_end = newbuf + newlen;
    mbuf->data_end = newbuf + (mbuf->data_end - mbuf->data);
    mbuf->read_pos = newbuf + (mbuf->read_pos - mbuf->data);
    mbuf->data = newbuf;
}

References MBuf::buf_end, buf_end, MBuf::data, MBuf::data_end, MBuf::read_pos, repalloc(), and STEP.

Referenced by mbuf_append().

pull_from_mbuf()

static int pull_from_mbuf ( void *  arg, PullFilter *  src, int  len, uint8 **  data_p, uint8 *  buf, int  buflen  )

static

Definition at line 323 of file mbuf.c.

{
    MBuf       *mbuf = arg;
 
    return mbuf_grab(mbuf, len, data_p);
}

References arg, len, and mbuf_grab().

pullf_create()

int pullf_create	(	PullFilter **	pf_p,
		const PullFilterOps *	op,
		void *	init_arg,
		PullFilter *	src
	)

Definition at line 191 of file mbuf.c.

{
    PullFilter *pf;
    void       *priv;
    int         res;
 
    if (op->init != NULL)
    {
        res = op->init(&priv, init_arg, src);
        if (res < 0)
            return res;
    }
    else
    {
        priv = init_arg;
        res = 0;
    }
 
    pf = palloc0(sizeof(*pf));
    pf->buflen = res;
    pf->op = op;
    pf->priv = priv;
    pf->src = src;
    if (pf->buflen > 0)
    {
        pf->buf = palloc(pf->buflen);
        pf->pos = 0;
    }
    else
    {
        pf->buf = NULL;
        pf->pos = 0;
    }
    *pf_p = pf;
    return 0;
}

References PullFilter::buf, PullFilter::buflen, PullFilterOps::init, PullFilter::op, palloc(), palloc0(), PullFilter::pos, PullFilter::priv, and PullFilter::src.

Referenced by parse_symenc_data(), parse_symenc_mdc_data(), pgp_create_pkt_reader(), process_data_packets(), process_secret_key(), and pullf_create_mbuf_reader().

pullf_create_mbuf_reader()

int pullf_create_mbuf_reader	(	PullFilter **	mp_p,
		MBuf *	src
	)

Definition at line 336 of file mbuf.c.

{
    return pullf_create(mp_p, &mbuf_reader, src, NULL);
}

References mbuf_reader, and pullf_create().

Referenced by pgp_decrypt(), pgp_get_keyid(), and pgp_set_pubkey().

pullf_free()

void pullf_free

(

PullFilter *

pf

)

Definition at line 229 of file mbuf.c.

{
    if (pf->op->free)
        pf->op->free(pf->priv);
 
    if (pf->buf)
    {
        px_memset(pf->buf, 0, pf->buflen);
        pfree(pf->buf);
    }
 
    px_memset(pf, 0, sizeof(*pf));
    pfree(pf);
}

References PullFilter::buf, PullFilter::buflen, PullFilterOps::free, PullFilter::op, pfree(), PullFilter::priv, and px_memset().

Referenced by internal_read_key(), parse_compressed_data(), parse_symenc_data(), parse_symenc_mdc_data(), pgp_decrypt(), pgp_get_keyid(), pgp_set_pubkey(), process_data_packets(), and process_secret_key().

pullf_read()

int pullf_read	(	PullFilter *	pf,
		int	len,
		uint8 **	data_p
	)

Definition at line 246 of file mbuf.c.

{
    int         res;
 
    if (pf->op->pull)
    {
        if (pf->buflen && len > pf->buflen)
            len = pf->buflen;
        res = pf->op->pull(pf->priv, pf->src, len, data_p,
                           pf->buf, pf->buflen);
    }
    else
        res = pullf_read(pf->src, len, data_p);
    return res;
}

References PullFilter::buf, PullFilter::buflen, len, PullFilter::op, PullFilter::priv, PullFilterOps::pull, pullf_read(), and PullFilter::src.

Referenced by decrypt_read(), mdc_read(), mdcbuf_refill(), parse_compressed_data(), parse_literal_data(), pgp_expect_packet_end(), pgp_parse_pkt_hdr(), pgp_skip_packet(), pktreader_pull(), pullf_read(), and pullf_read_max().

pullf_read_fixed()

int pullf_read_fixed	(	PullFilter *	src,
		int	len,
		uint8 *	dst
	)

Definition at line 301 of file mbuf.c.

{
    int         res;
    uint8      *p;
 
    res = pullf_read_max(src, len, &p, dst);
    if (res < 0)
        return res;
    if (res != len)
    {
        px_debug("pullf_read_fixed: need=%d got=%d", len, res);
        return PXE_PGP_CORRUPT_DATA;
    }
    if (p != dst)
        memcpy(dst, p, len);
    return 0;
}

References len, pullf_read_max(), px_debug(), and PXE_PGP_CORRUPT_DATA.

Referenced by _pgp_read_public_key(), check_key_cksum(), check_key_sha1(), pgp_mpi_read(), pgp_parse_pubenc_sesskey(), pgp_s2k_read(), process_secret_key(), and read_pubenc_keyid().

pullf_read_max()

int pullf_read_max	(	PullFilter *	pf,
		int	len,
		uint8 **	data_p,
		uint8 *	tmpbuf
	)

Definition at line 263 of file mbuf.c.

{
    int         res,
                total;
    uint8      *tmp;
 
    res = pullf_read(pf, len, data_p);
    if (res <= 0 || res == len)
        return res;
 
    /* read was shorter, use tmpbuf */
    memcpy(tmpbuf, *data_p, res);
    *data_p = tmpbuf;
    len -= res;
    total = res;
 
    while (len > 0)
    {
        res = pullf_read(pf, len, &tmp);
        if (res < 0)
        {
            /* so the caller must clear only on success */
            px_memset(tmpbuf, 0, total);
            return res;
        }
        if (res == 0)
            break;
        memcpy(tmpbuf + total, tmp, res);
        total += res;
        len -= res;
    }
    return total;
}

References len, pullf_read(), px_memset(), and tmpbuf.

Referenced by mdc_finish(), parse_literal_data(), parse_symenc_sesskey(), prefix_init(), and pullf_read_fixed().

push_into_mbuf()

static int push_into_mbuf ( PushFilter *  next, void *  arg, const uint8 *  data, int  len  )

static

Definition at line 529 of file mbuf.c.

{
    int         res = 0;
    MBuf       *mbuf = arg;
 
    if (len > 0)
        res = mbuf_append(mbuf, data, len);
    return res < 0 ? res : 0;
}

References arg, data, len, and mbuf_append().

pushf_create()

int pushf_create	(	PushFilter **	mp_p,
		const PushFilterOps *	op,
		void *	init_arg,
		PushFilter *	next
	)

Definition at line 357 of file mbuf.c.

{
    PushFilter *mp;
    void       *priv;
    int         res;
 
    if (op->init != NULL)
    {
        res = op->init(next, init_arg, &priv);
        if (res < 0)
            return res;
    }
    else
    {
        priv = init_arg;
        res = 0;
    }
 
    mp = palloc0(sizeof(*mp));
    mp->block_size = res;
    mp->op = op;
    mp->priv = priv;
    mp->next = next;
    if (mp->block_size > 0)
    {
        mp->buf = palloc(mp->block_size);
        mp->pos = 0;
    }
    else
    {
        mp->buf = NULL;
        mp->pos = 0;
    }
    *mp_p = mp;
    return 0;
}

References PushFilter::block_size, PushFilter::buf, PushFilterOps::init, next, PushFilter::next, PushFilter::op, palloc(), palloc0(), PushFilter::pos, and PushFilter::priv.

Referenced by init_compress(), init_encdata_packet(), init_litdata_packet(), pgp_create_pkt_writer(), pgp_encrypt(), and pushf_create_mbuf_writer().

pushf_create_mbuf_writer()

int pushf_create_mbuf_writer	(	PushFilter **	res,
		MBuf *	dst
	)

Definition at line 544 of file mbuf.c.

{
    return pushf_create(res, &mbuf_filter, dst, NULL);
}

References mbuf_filter, and pushf_create().

Referenced by pgp_encrypt().

pushf_flush()

int pushf_flush

(

PushFilter *

mp

)

Definition at line 499 of file mbuf.c.

{
    int         res;
 
    while (mp)
    {
        if (mp->block_size > 0)
        {
            res = wrap_process(mp, mp->buf, mp->pos);
            if (res < 0)
                return res;
        }
 
        if (mp->op->flush)
        {
            res = mp->op->flush(mp->next, mp->priv);
            if (res < 0)
                return res;
        }
 
        mp = mp->next;
    }
    return 0;
}

References PushFilter::block_size, PushFilter::buf, PushFilterOps::flush, PushFilter::next, PushFilter::op, PushFilter::pos, PushFilter::priv, and wrap_process().

Referenced by pgp_encrypt(), and pgp_write_pubenc_sesskey().

pushf_free()

void pushf_free

(

PushFilter *

mp

)

Definition at line 395 of file mbuf.c.

{
    if (mp->op->free)
        mp->op->free(mp->priv);
 
    if (mp->buf)
    {
        px_memset(mp->buf, 0, mp->block_size);
        pfree(mp->buf);
    }
 
    px_memset(mp, 0, sizeof(*mp));
    pfree(mp);
}

References PushFilter::block_size, PushFilter::buf, PushFilterOps::free, PushFilter::op, pfree(), PushFilter::priv, and px_memset().

Referenced by init_compress(), init_litdata_packet(), pgp_write_pubenc_sesskey(), and pushf_free_all().

pushf_free_all()

void pushf_free_all

(

PushFilter *

mp

)

Definition at line 411 of file mbuf.c.

{
    PushFilter *tmp;
 
    while (mp)
    {
        tmp = mp->next;
        pushf_free(mp);
        mp = tmp;
    }
}

References PushFilter::next, and pushf_free().

Referenced by pgp_encrypt().

pushf_write()

int pushf_write	(	PushFilter *	mp,
		const uint8 *	data,
		int	len
	)

Definition at line 439 of file mbuf.c.

{
    int         need,
                res;
 
    /*
     * no buffering
     */
    if (mp->block_size <= 0)
        return wrap_process(mp, data, len);
 
    /*
     * try to empty buffer
     */
    need = mp->block_size - mp->pos;
    if (need > 0)
    {
        if (len < need)
        {
            memcpy(mp->buf + mp->pos, data, len);
            mp->pos += len;
            return 0;
        }
        memcpy(mp->buf + mp->pos, data, need);
        len -= need;
        data += need;
    }
 
    /*
     * buffer full, process
     */
    res = wrap_process(mp, mp->buf, mp->block_size);
    if (res < 0)
        return res;
    mp->pos = 0;
 
    /*
     * now process directly from data
     */
    while (len > 0)
    {
        if (len > mp->block_size)
        {
            res = wrap_process(mp, data, mp->block_size);
            if (res < 0)
                return res;
            data += mp->block_size;
            len -= mp->block_size;
        }
        else
        {
            memcpy(mp->buf, data, len);
            mp->pos += len;
            break;
        }
    }
    return 0;
}

References PushFilter::block_size, PushFilter::buf, data, len, PushFilter::pos, and wrap_process().

Referenced by crlf_process(), encrypt_init(), encrypt_process(), init_compress(), init_litdata_packet(), mdc_flush(), mdc_write(), pgp_encrypt(), pgp_mpi_write(), pgp_write_pubenc_sesskey(), pkt_stream_flush(), pkt_stream_process(), wrap_process(), write_normal_header(), write_prefix(), write_symenc_sesskey(), and write_tag_only().

wrap_process()

static int wrap_process ( PushFilter *  mp, const uint8 *  data, int  len  )

static

Definition at line 424 of file mbuf.c.

{
    int         res;
 
    if (mp->op->push != NULL)
        res = mp->op->push(mp->next, mp->priv, data, len);
    else
        res = pushf_write(mp->next, data, len);
    if (res > 0)
        return PXE_BUG;
    return res;
}

References data, len, PushFilter::next, PushFilter::op, PushFilter::priv, PushFilterOps::push, pushf_write(), and PXE_BUG.

Referenced by pushf_flush(), and pushf_write().

Variable Documentation

mbuf_filter

const struct PushFilterOps mbuf_filter

static

Initial value:

= {
    NULL, push_into_mbuf, NULL, NULL
}

Definition at line 539 of file mbuf.c.

Referenced by pushf_create_mbuf_writer().

mbuf_reader

const struct PullFilterOps mbuf_reader

static

Initial value:

= {
    NULL, pull_from_mbuf, NULL
}

Definition at line 331 of file mbuf.c.

Referenced by pullf_create_mbuf_reader().