encode YUV420 to jpeg, using libjpeg

#include
#include
#include "jpeglib.h"
#include
#include

gboolean jpeg_enc_yv12(guchar* buffer, gint width, gint height, gint quality, char* filename)
{
    struct jpeg_compress_struct cinfo;
    struct jpeg_error_mgr jerr;
    FILE *outfile = NULL;
    gboolean ret = TRUE;
    if(buffer == NULL || width <=0 || height <=0|| filename == NULL)
        return FALSE;
    if ((outfile = fopen(filename, "wb")) == NULL)
    {  
        return FALSE;
    }    
    cinfo.err = jpeg_std_error(&jerr);
    jpeg_create_compress(&cinfo);
    jpeg_stdio_dest(&cinfo, outfile);

    cinfo.image_width = width;
    cinfo.image_height = height;
    cinfo.input_components = 3;        
    cinfo.in_color_space = JCS_YCbCr;
    jpeg_set_defaults(&cinfo);
    jpeg_set_quality(&cinfo, quality, TRUE);
    cinfo.raw_data_in = TRUE;  
    
    {
        JSAMPARRAY pp[3];
        JSAMPROW *rpY = (JSAMPROW*)malloc(sizeof(JSAMPROW) * height);
        JSAMPROW *rpU = (JSAMPROW*)malloc(sizeof(JSAMPROW) * height);
        JSAMPROW *rpV = (JSAMPROW*)malloc(sizeof(JSAMPROW) * height);
        int k;
        if(rpY == NULL && rpU == NULL && rpV == NULL)
        {
            ret = FALSE;
            goto exit;
        }
        cinfo.comp_info[0].h_samp_factor =
        cinfo.comp_info[0].v_samp_factor = 2;
        cinfo.comp_info[1].h_samp_factor =
        cinfo.comp_info[1].v_samp_factor =
        cinfo.comp_info[2].h_samp_factor =
        cinfo.comp_info[2].v_samp_factor = 1;
        jpeg_start_compress(&cinfo, TRUE);

        for (k = 0; k < height; k+=2)
        {
            rpY[k]   = buffer + k*width;
            rpY[k+1] = buffer + (k+1)*width;
            rpU[k/2] = buffer+width*height + (k/2)*width/2;
            rpV[k/2] = buffer+width*height*5/4 + (k/2)*width/2;
        }
        for (k = 0; k < height; k+=2*DCTSIZE)
        {
            pp[0] = &rpY[k];
            pp[1] = &rpU[k/2];
            pp[2] = &rpV[k/2];
            jpeg_write_raw_data(&cinfo, pp, 2*DCTSIZE);
        }
        jpeg_finish_compress(&cinfo);
        free(rpY);
        free(rpU);
        free(rpV);
    }
exit:
    fclose(outfile);
    jpeg_destroy_compress(&cinfo);
    return ret;
}

int main()
{
    guchar *buffer = malloc(640*480*3/2);
    jpeg_enc_yv12(buffer, 640, 480, 90, "test.jpg");
    free(buffer);    
}