forked from chundongwang/untrunc
-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.cpp
77 lines (64 loc) · 1.9 KB
/
main.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
/*
Untrunc - main.cpp
Untrunc is GPL software; you can freely distribute,
redistribute, modify & use under the terms of the GNU General
Public License; either version 2 or its successor.
Untrunc is distributed under the GPL "AS IS", without
any warranty; without the implied warranty of merchantability
or fitness for either an expressed or implied particular purpose.
Please see the included GNU General Public License (GPL) for
your rights and further details; see the file COPYING. If you
cannot, write to the Free Software Foundation, 59 Temple Place
Suite 330, Boston, MA 02111-1307, USA. Or www.fsf.org
Copyright 2010 Federico Ponchio
*/
#include "mp4.h"
#include "atom.h"
#include <iostream>
#include <string>
using namespace std;
void usage() {
cerr << "Usage: untrunc [options] <ok.mp4> [<corrupt.mp4>]\n\n";
}
int main(int argc, char *argv[]) {
bool info = false;
bool analyze = false;
int i = 1;
for(; i < argc; i++) {
string arg(argv[i]);
if(arg[0] == '-') {
if(arg[1] == 'i') info = true;
if(arg[1] == 'a') analyze = true;
} else
break;
}
if(argc == i) {
usage();
return -1;
}
string ok = argv[i];
string corrupt;
i++;
if(i < argc)
corrupt = argv[i];
cout << "Reading: " << ok << endl;
Mp4 mp4;
try {
mp4.open(ok);
if(info) {
mp4.printMediaInfo();
mp4.printAtoms();
}
if(analyze) {
mp4.analyze();
}
if(corrupt.size()) {
mp4.repair(corrupt);
mp4.saveVideo(corrupt + "_fixed.mp4");
}
} catch(string e) {
cerr << e << endl;
return -1;
}
return 0;
}