Mercurial > libavformat.hg
view jpeg.c @ 6:b681d968f227 libavformat
Fix a crash caused by a null coded_picture pointer in ffserver.
author | philipjsg |
---|---|
date | Mon, 09 Dec 2002 02:53:55 +0000 |
parents | 39c4c4336486 |
children | 45308962220f |
line wrap: on
line source
/* * JPEG based formats * Copyright (c) 2000, 2001 Fabrice Bellard. * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2 of the License, or (at your option) any later version. * * This library is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with this library; if not, write to the Free Software * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA */ #include "avformat.h" /* Multipart JPEG */ #define BOUNDARY_TAG "ffserver" static int mpjpeg_write_header(AVFormatContext *s) { UINT8 buf1[256]; snprintf(buf1, sizeof(buf1), "--%s\n", BOUNDARY_TAG); put_buffer(&s->pb, buf1, strlen(buf1)); put_flush_packet(&s->pb); return 0; } static int mpjpeg_write_packet(AVFormatContext *s, int stream_index, UINT8 *buf, int size, int force_pts) { UINT8 buf1[256]; snprintf(buf1, sizeof(buf1), "Content-type: image/jpeg\n\n"); put_buffer(&s->pb, buf1, strlen(buf1)); put_buffer(&s->pb, buf, size); snprintf(buf1, sizeof(buf1), "\n--%s\n", BOUNDARY_TAG); put_buffer(&s->pb, buf1, strlen(buf1)); put_flush_packet(&s->pb); return 0; } static int mpjpeg_write_trailer(AVFormatContext *s) { return 0; } static AVOutputFormat mpjpeg_format = { "mpjpeg", "Mime multipart JPEG format", "multipart/x-mixed-replace;boundary=" BOUNDARY_TAG, "mjpg", 0, CODEC_ID_NONE, CODEC_ID_MJPEG, mpjpeg_write_header, mpjpeg_write_packet, mpjpeg_write_trailer, }; /*************************************/ /* single frame JPEG */ static int single_jpeg_write_header(AVFormatContext *s) { return 0; } static int single_jpeg_write_packet(AVFormatContext *s, int stream_index, UINT8 *buf, int size, int force_pts) { put_buffer(&s->pb, buf, size); put_flush_packet(&s->pb); return 1; /* no more data can be sent */ } static int single_jpeg_write_trailer(AVFormatContext *s) { return 0; } static AVOutputFormat single_jpeg_format = { "singlejpeg", "single JPEG image", "image/jpeg", NULL, /* note: no extension to favorize jpeg multiple images match */ 0, CODEC_ID_NONE, CODEC_ID_MJPEG, single_jpeg_write_header, single_jpeg_write_packet, single_jpeg_write_trailer, }; /*************************************/ /* multiple jpeg images */ typedef struct JpegContext { char path[1024]; int img_number; } JpegContext; static int jpeg_write_header(AVFormatContext *s1) { JpegContext *s; s = av_mallocz(sizeof(JpegContext)); if (!s) return -1; s1->priv_data = s; pstrcpy(s->path, sizeof(s->path), s1->filename); s->img_number = 1; return 0; } static int jpeg_write_packet(AVFormatContext *s1, int stream_index, UINT8 *buf, int size, int force_pts) { JpegContext *s = s1->priv_data; char filename[1024]; ByteIOContext f1, *pb = &f1; if (get_frame_filename(filename, sizeof(filename), s->path, s->img_number) < 0) return -EIO; if (url_fopen(pb, filename, URL_WRONLY) < 0) return -EIO; put_buffer(pb, buf, size); put_flush_packet(pb); url_fclose(pb); s->img_number++; return 0; } static int jpeg_write_trailer(AVFormatContext *s1) { return 0; } /***/ static int jpeg_read_header(AVFormatContext *s1, AVFormatParameters *ap) { JpegContext *s; int i; char buf[1024]; ByteIOContext pb1, *f = &pb1; AVStream *st; s = av_mallocz(sizeof(JpegContext)); if (!s) return -1; s1->priv_data = s; pstrcpy(s->path, sizeof(s->path), s1->filename); s1->nb_streams = 1; st = av_mallocz(sizeof(AVStream)); if (!st) { av_free(s); return -ENOMEM; } avcodec_get_context_defaults(&st->codec); s1->streams[0] = st; s->img_number = 0; /* try to find the first image */ for(i=0;i<5;i++) { if (get_frame_filename(buf, sizeof(buf), s->path, s->img_number) < 0) goto fail; if (url_fopen(f, buf, URL_RDONLY) >= 0) break; s->img_number++; } if (i == 5) goto fail; url_fclose(f); st->codec.codec_type = CODEC_TYPE_VIDEO; st->codec.codec_id = CODEC_ID_MJPEG; if (!ap || !ap->frame_rate) st->codec.frame_rate = 25 * FRAME_RATE_BASE; else st->codec.frame_rate = ap->frame_rate; return 0; fail: av_free(s); return -EIO; } static int jpeg_read_packet(AVFormatContext *s1, AVPacket *pkt) { JpegContext *s = s1->priv_data; char filename[1024]; int size; ByteIOContext f1, *f = &f1; if (get_frame_filename(filename, sizeof(filename), s->path, s->img_number) < 0) return -EIO; f = &f1; if (url_fopen(f, filename, URL_RDONLY) < 0) return -EIO; size = url_seek(url_fileno(f), 0, SEEK_END); url_seek(url_fileno(f), 0, SEEK_SET); av_new_packet(pkt, size); pkt->stream_index = 0; get_buffer(f, pkt->data, size); url_fclose(f); s->img_number++; return 0; } static int jpeg_read_close(AVFormatContext *s1) { return 0; } static AVInputFormat jpeg_iformat = { "jpeg", "JPEG image", sizeof(JpegContext), NULL, jpeg_read_header, jpeg_read_packet, jpeg_read_close, NULL, .flags = AVFMT_NOFILE | AVFMT_NEEDNUMBER, .extensions = "jpg,jpeg", }; static AVOutputFormat jpeg_oformat = { "jpeg", "JPEG image", "image/jpeg", "jpg,jpeg", sizeof(JpegContext), CODEC_ID_NONE, CODEC_ID_MJPEG, jpeg_write_header, jpeg_write_packet, jpeg_write_trailer, .flags = AVFMT_NOFILE | AVFMT_NEEDNUMBER, }; int jpeg_init(void) { av_register_output_format(&mpjpeg_format); av_register_output_format(&single_jpeg_format); av_register_input_format(&jpeg_iformat); av_register_output_format(&jpeg_oformat); return 0; }