probetest.c 6.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202
  1. /*
  2. * copyright (c) 2009 Michael Niedermayer <michaelni@gmx.at>
  3. *
  4. * This file is part of FFmpeg.
  5. *
  6. * FFmpeg is free software; you can redistribute it and/or
  7. * modify it under the terms of the GNU Lesser General Public
  8. * License as published by the Free Software Foundation; either
  9. * version 2.1 of the License, or (at your option) any later version.
  10. *
  11. * FFmpeg is distributed in the hope that it will be useful,
  12. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  14. * Lesser General Public License for more details.
  15. *
  16. * You should have received a copy of the GNU Lesser General Public
  17. * License along with FFmpeg; if not, write to the Free Software
  18. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  19. */
  20. #include <stdlib.h>
  21. #include "libavformat/avformat.h"
  22. #include "libavcodec/put_bits.h"
  23. #include "libavutil/lfg.h"
  24. #include "libavutil/timer.h"
  25. #define MAX_FORMATS 1000 //this must be larger than the number of formats
  26. static int score_array[MAX_FORMATS];
  27. static int64_t time_array[MAX_FORMATS];
  28. static int failures = 0;
  29. static const char *single_format;
  30. #ifndef AV_READ_TIME
  31. #define AV_READ_TIME(x) 0
  32. #endif
  33. static void probe(AVProbeData *pd, int type, int p, int size)
  34. {
  35. int i = 0;
  36. const AVInputFormat *fmt = NULL;
  37. void *fmt_opaque = NULL;
  38. while ((fmt = av_demuxer_iterate(&fmt_opaque))) {
  39. if (fmt->flags & AVFMT_NOFILE)
  40. continue;
  41. if (fmt->read_probe &&
  42. (!single_format || !strcmp(single_format, fmt->name))
  43. ) {
  44. int score;
  45. int64_t start = AV_READ_TIME();
  46. score = fmt->read_probe(pd);
  47. time_array[i] += AV_READ_TIME() - start;
  48. if (score > score_array[i] && score > AVPROBE_SCORE_MAX / 4) {
  49. score_array[i] = score;
  50. fprintf(stderr,
  51. "Failure of %s probing code with score=%d type=%d p=%X size=%d\n",
  52. fmt->name, score, type, p, size);
  53. failures++;
  54. }
  55. }
  56. i++;
  57. }
  58. }
  59. static void print_times(void)
  60. {
  61. int i = 0;
  62. const AVInputFormat *fmt = NULL;
  63. void *fmt_opaque = NULL;
  64. while ((fmt = av_demuxer_iterate(&fmt_opaque))) {
  65. if (fmt->flags & AVFMT_NOFILE)
  66. continue;
  67. if (time_array[i] > 1000000) {
  68. fprintf(stderr, "%12"PRIu64" cycles, %12s\n",
  69. time_array[i], fmt->name);
  70. }
  71. i++;
  72. }
  73. }
  74. static int read_int(char *arg) {
  75. int ret;
  76. if (!arg || !*arg)
  77. return -1;
  78. ret = strtol(arg, &arg, 0);
  79. if (*arg)
  80. return -1;
  81. return ret;
  82. }
  83. int main(int argc, char **argv)
  84. {
  85. unsigned int p, i, type, size, retry;
  86. AVProbeData pd = { 0 };
  87. AVLFG state;
  88. PutBitContext pb;
  89. int retry_count= 4097;
  90. int max_size = 65537;
  91. int j;
  92. for (j = i = 1; i<argc; i++) {
  93. if (!strcmp(argv[i], "-f") && i+1<argc && !single_format) {
  94. single_format = argv[++i];
  95. } else if (read_int(argv[i])>0 && j == 1) {
  96. retry_count = read_int(argv[i]);
  97. j++;
  98. } else if (read_int(argv[i])>0 && j == 2) {
  99. max_size = read_int(argv[i]);
  100. j++;
  101. } else {
  102. fprintf(stderr, "probetest [-f <input format>] [<retry_count> [<max_size>]]\n");
  103. return 1;
  104. }
  105. }
  106. if (max_size > 1000000000U/8) {
  107. fprintf(stderr, "max_size out of bounds\n");
  108. return 1;
  109. }
  110. if (retry_count > 1000000000U) {
  111. fprintf(stderr, "retry_count out of bounds\n");
  112. return 1;
  113. }
  114. av_lfg_init(&state, 0xdeadbeef);
  115. pd.buf = NULL;
  116. for (size = 1; size < max_size; size *= 2) {
  117. pd.buf_size = size;
  118. pd.buf = av_realloc(pd.buf, size + AVPROBE_PADDING_SIZE);
  119. pd.filename = "";
  120. if (!pd.buf) {
  121. fprintf(stderr, "out of memory\n");
  122. return 1;
  123. }
  124. memset(pd.buf, 0, size + AVPROBE_PADDING_SIZE);
  125. fprintf(stderr, "testing size=%d\n", size);
  126. for (retry = 0; retry < retry_count; retry += FFMAX(size, 32)) {
  127. for (type = 0; type < 4; type++) {
  128. for (p = 0; p < 4096; p++) {
  129. unsigned hist = 0;
  130. init_put_bits(&pb, pd.buf, size);
  131. switch (type) {
  132. case 0:
  133. for (i = 0; i < size * 8; i++)
  134. put_bits(&pb, 1, (av_lfg_get(&state) & 0xFFFFFFFF) > p << 20);
  135. break;
  136. case 1:
  137. for (i = 0; i < size * 8; i++) {
  138. unsigned int p2 = hist ? p & 0x3F : (p >> 6);
  139. unsigned int v = (av_lfg_get(&state) & 0xFFFFFFFF) > p2 << 26;
  140. put_bits(&pb, 1, v);
  141. hist = v;
  142. }
  143. break;
  144. case 2:
  145. for (i = 0; i < size * 8; i++) {
  146. unsigned int p2 = (p >> (hist * 3)) & 7;
  147. unsigned int v = (av_lfg_get(&state) & 0xFFFFFFFF) > p2 << 29;
  148. put_bits(&pb, 1, v);
  149. hist = (2 * hist + v) & 3;
  150. }
  151. break;
  152. case 3:
  153. for (i = 0; i < size; i++) {
  154. int c = 0;
  155. while (p & 63) {
  156. c = (av_lfg_get(&state) & 0xFFFFFFFF) >> 24;
  157. if (c >= 'a' && c <= 'z' && (p & 1))
  158. break;
  159. else if (c >= 'A' && c <= 'Z' && (p & 2))
  160. break;
  161. else if (c >= '0' && c <= '9' && (p & 4))
  162. break;
  163. else if (c == ' ' && (p & 8))
  164. break;
  165. else if (c == 0 && (p & 16))
  166. break;
  167. else if (c == 1 && (p & 32))
  168. break;
  169. }
  170. pd.buf[i] = c;
  171. }
  172. }
  173. flush_put_bits(&pb);
  174. probe(&pd, type, p, size);
  175. }
  176. }
  177. }
  178. }
  179. if(AV_READ_TIME())
  180. print_times();
  181. return failures;
  182. }