Added test for libavcodec/avpacket.c
[ffmpeg.git] / libavcodec / tests / avpacket.c
1 /*
2  * This file is part of FFmpeg.
3  *
4  * FFmpeg is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * FFmpeg is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * Lesser General Public License for more details.
13  *
14  * You should have received a copy of the GNU Lesser General Public
15  * License along with FFmpeg; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17  */
18
19 #include <stdio.h>
20 #include <stdlib.h>
21 #include <inttypes.h>
22 #include <string.h>
23 #include "libavcodec/avcodec.h"
24 #include "libavutil/error.h"
25
26
27
28 static int setup_side_data_entry(AVPacket* avpkt)
29 {
30     const uint8_t *data_name = NULL;
31     int ret = 0, bytes;
32     uint8_t *extra_data = NULL;
33
34
35     /* get side_data_name string */
36     data_name = av_packet_side_data_name(AV_PKT_DATA_NEW_EXTRADATA);
37
38     /* Allocate a memory bloc */
39     bytes = strlen(data_name);
40
41     if(!(extra_data = av_malloc(bytes))){
42         ret = AVERROR(ENOMEM);
43         fprintf(stderr, "Error occurred: %s\n", av_err2str(ret));
44         exit(1);
45     }
46     /* copy side_data_name to extra_data array */
47     memcpy(extra_data, data_name, bytes);
48
49     /* create side data for AVPacket */
50     ret = av_packet_add_side_data(avpkt, AV_PKT_DATA_NEW_EXTRADATA,
51                                         extra_data, bytes);
52     if(ret < 0){
53         fprintf(stderr,
54                 "Error occurred in av_packet_add_side_data: %s\n",
55                 av_err2str(ret));
56     }
57
58     return ret;
59 }
60
61 static int initializations(AVPacket* avpkt)
62 {
63     const static uint8_t* data = "selftest for av_packet_clone(...)";
64     int ret = 0;
65
66     /* initialize avpkt */
67     av_init_packet(avpkt);
68
69     /* set values for avpkt */
70     avpkt->pts = 17;
71     avpkt->dts = 2;
72     avpkt->data = (uint8_t*)data;
73     avpkt->size = strlen(data);
74     avpkt->flags = AV_PKT_FLAG_DISCARD;
75     avpkt->duration = 100;
76     avpkt->pos = 3;
77
78     ret = setup_side_data_entry(avpkt);
79
80     return ret;
81 }
82
83 int main(void)
84 {
85     AVPacket avpkt;
86     AVPacket *avpkt_clone = NULL;
87     int ret = 0;
88
89     if(initializations(&avpkt) < 0){
90         printf("failed to initialize variables\n");
91         return 1;
92     }
93     /* test av_packet_clone*/
94     avpkt_clone = av_packet_clone(&avpkt);
95
96     if(!avpkt_clone) {
97         av_log(NULL, AV_LOG_ERROR,"av_packet_clone failed to clone AVPacket\n");
98         return 1;
99     }
100     /*test av_grow_packet*/
101     if(av_grow_packet(avpkt_clone, 20) < 0){
102         av_log(NULL, AV_LOG_ERROR, "av_grow_packet failed\n");
103         return 1;
104     }
105     if(av_grow_packet(avpkt_clone, INT_MAX) == 0){
106         printf( "av_grow_packet failed to return error "
107                 "when \"grow_by\" parameter is too large.\n" );
108         ret = 1;
109     }
110     /* test size error check in av_new_packet*/
111     if(av_new_packet(avpkt_clone, INT_MAX) == 0){
112         printf( "av_new_packet failed to return error "
113                 "when \"size\" parameter is too large.\n" );
114         ret = 1;
115     }
116     /*test size error check in av_packet_from_data*/
117     if(av_packet_from_data(avpkt_clone, avpkt_clone->data, INT_MAX) == 0){
118         printf("av_packet_from_data failed to return error "
119                 "when \"size\" parameter is too large.\n" );
120         ret = 1;
121     }
122     /*clean up*/
123     av_packet_free(&avpkt_clone);
124     av_packet_unref(&avpkt);
125
126
127     return ret;
128 }