123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101 |
- /*
- * Copyright (C) 2023 Teluu Inc. (http://www.teluu.com)
- *
- * This program is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
- * (at your option) any later version.
- *
- * This program 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 General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
- */
- #include <stdio.h>
- #include <stdint.h>
- #include <stdlib.h>
- #include <pjlib.h>
- #include <pjmedia-codec/h264_packetizer.h>
- #if defined(PJMEDIA_HAS_VIDEO) && (PJMEDIA_HAS_VIDEO != 0)
- #define kMinInputLength 10
- #define kMaxInputLength 5120
- pj_pool_factory *mem;
- int h264_unpacketizer(const uint8_t *data, size_t size,
- uint8_t *output, size_t output_size)
- {
- int ret = 0;
- pj_pool_t *pool;
- pj_status_t status;
- pjmedia_h264_packetizer_cfg cfg;
- pjmedia_h264_packetizer *pktz;
- unsigned bits_pos = 0;
- pool = pj_pool_create(mem, "h264_test", 1000, 1000, NULL);
- pj_bzero(&cfg, sizeof(cfg));
- cfg.mtu = 1500;
- cfg.unpack_nal_start = 4;
- cfg.mode = PJMEDIA_H264_PACKETIZER_MODE_NON_INTERLEAVED;
- status = pjmedia_h264_packetizer_create(pool, &cfg, &pktz);
- if (status == PJ_SUCCESS) {
- status = pjmedia_h264_unpacketize(pktz, data, size, output,
- output_size, &bits_pos);
- }
- pj_pool_release(pool);
- return ret;
- }
- extern int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size)
- {
- int ret = 0;
- uint8_t *data;
- uint8_t *output;
- size_t output_size;
- pj_caching_pool caching_pool;
- if (Size < kMinInputLength || Size > kMaxInputLength) {
- return 1;
- }
- /* Add null termination for the data */
- data = (uint8_t *)calloc((Size+1), sizeof(uint8_t));
- memcpy((void *)data, (void *)Data, Size);
- output_size = Size + 32;
- output = (uint8_t *)calloc(output_size, sizeof(uint8_t));
- /* Init */
- pj_init();
- pj_caching_pool_init(&caching_pool, &pj_pool_factory_default_policy, 0);
- pj_log_set_level(0);
- mem = &caching_pool.factory;
- /* Fuzz */
- ret = h264_unpacketizer(data, Size, output, output_size);
- free(data);
- free(output);
- pj_caching_pool_destroy(&caching_pool);
- return ret;
- }
- #else
- extern int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size)
- {
- PJ_UNUSED_ARG(Data);
- PJ_UNUSED_ARG(Size);
- return 0;
- }
- #endif
|