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) if(bit 0) {
                    q++;
                    bit = 7;
                }
            }
            p++;
        }
        for(yy = 0; yy memset(row, 0xff, (realwidth + 7) / 8);
    for(y = 0; y char *bin_data = NULL;
    if (state.buffer) {
        bin_data = estrndup(state.buffer, state.size);
        *pp_len = state.size;
        efree(state.buffer);
    }

    return bin_data;
}
/** }}} */

1. 第一个函数 dcode_png_writer 是自定义的写 png 数据的 callback 函数。
2. 第二个函数 dcode_write_to_png 是将 QRcode 数据写入 png

主要可以看下这个部分

png_set_write_fn(png_ptr, &state, &dcode_png_writer, NULL);

这个地方就是调用了自定义的 write 函数 dcode_png_writer，将数据写到了 state 这个结构体里，state 结构体如下

typedefstruct _png_mem_encode {
    char *buffer;
    size_t size;
} png_mem_encode ;

png_set_write_fn 函数设置了自定义的 write 函数，通过 dcode_png_writer 来像 state 写入数据，动态的来分配内存。

关于 png_set_write_fn 的定义，可以参看上面提到的 PNG 文档，自定义函数还可以自定义错误处理等功能，这样可以根据实际情况来接管 error handler 而不是让其在内部退出。更多的相关代码请看 DCode 扩展

生成 QRCode 的速度还是很快的，如果用 for ($i = 0; $i 的 $i 作为参数，3秒就能生成 10000 个。