php扩展开发笔记（10）自定义 libpng 库中的 IO 函数，将图片写入内存

在开发这个生成二维码扩展 dcode 的时候，需要将生成的二维码 png 图片以字符串的方式返回给调用者，而不是直接生成文件，这样比较方便的是不用去操作文件，将文件的操作完全交给用户。

生成图片采用了 libpng 的库，关于 libpng 的文档大家可以到 这里 png 文档 看。我使用这个库在 Ubuntu14.04 上编译我的扩展的时候还有个小问题 png_create_write_struct in Unknown on line 0 on ubuntu 14，到网上一搜索，还是非常常见的。

下面简单的列一下代码：

/** {{{ dcode_png_writer()
 * function is custom png_write callback function
 * Return void */staticvoid dcode_png_writer(png_structp png_ptr, png_bytep data, png_size_t length)
{
    png_mem_encode* p = (png_mem_encode*) png_get_io_ptr(png_ptr);
    size_t nsize = p->size + length;

    if (p->buffer)
        p->buffer = erealloc(p->buffer, nsize);
    else
        p->buffer = emalloc(nsize);

    if (!p->buffer)
    {
        png_error(png_ptr, "PNG allocate memory error");
        exit(FAILURE);
    }

    memcpy(p->buffer + p->size, data, length);
    p->size += length;
}
/* }}} */

/** {{{ dcode_write_to_png()
 * write qrcode struct to memory
 * Return char* */staticchar* dcode_write_to_png(QRcode *qrcode, int size, int margin, int *pp_len)
{

    png_structp png_ptr;
    png_infop info_ptr;

    unsignedchar *row, *p, *q;
    int x, y, xx, yy, bit;
    int realwidth;

    realwidth = (qrcode->width + margin * 2) * size;
    int row_fill_len = (realwidth + 7) / 8;

    png_ptr = png_create_write_struct(PNG_LIBPNG_VER_STRING, NULL, NULL, NULL);
    if (png_ptr == NULL)
    {
        php_error(E_ERROR, "Failed to initialize PNG writer");
        return NULL;
    }

    info_ptr = png_create_info_struct(png_ptr);
    if (info_ptr == NULL)
    {
        php_error(E_ERROR, "Failed to initialize PNG info");
        return NULL;
    }

    if (setjmp(png_jmpbuf(png_ptr)))
    {
        png_destroy_write_struct(&png_ptr, &info_ptr);
        php_error(E_ERROR, "Failed to set PNG jmpbuf");
        return NULL;
    }

    row = (unsignedchar *) emalloc(row_fill_len);
    if (row == NULL)
    {
        png_destroy_write_struct(&png_ptr, &info_ptr);
        php_error(E_ERROR, "Failed to allocate memory");
        return NULL;
    }

    png_mem_encode state = {NULL, 0};
    png_set_write_fn(png_ptr, &state, &dcode_png_writer, NULL);

    png_set_IHDR(png_ptr,
                info_ptr,
                realwidth,
                realwidth,
                1,
                PNG_COLOR_TYPE_GRAY,
                PNG_INTERLACE_NONE,
                PNG_COMPRESSION_TYPE_DEFAULT,
                PNG_FILTER_TYPE_DEFAULT);

    png_write_info(png_ptr, info_ptr);
    memset(row, 0xff, (realwidth + 7) / 8);
    for(y = 0; y data;
    for(y = 0; y width; y ++) {
        bit = 7;
        memset(row, 0xff, (realwidth + 7) / 8);
        q = row;
        q += margin * size / 8;
        bit = 7 - (margin * size % 8);
        for(x = 0; x width; x ++) {
            for(xx = 0; xx 1