blob: e79616e6a33ade4980bf7d8973f3f20e24db4aaa (
plain)
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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
/*
Copyright (C) 2005-2009 Michel de Boer <michel@twinklephone.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, see <https://www.gnu.org/licenses/>.
*/
#include "mime_database.h"
#include <cassert>
#include "log.h"
#include "sys_settings.h"
using namespace utils;
//////////////////////////
// class t_mime_db_record
//////////////////////////
bool t_mime_db_record::create_file_record(vector<string> &v) const {
// The mime database is read only. So this function should
// never be called.
assert(false);
return false;
}
bool t_mime_db_record::populate_from_file_record(const vector<string> &v) {
// Check number of fields
if (v.size() != 2) return false;
mimetype = v[0];
file_glob = v[1];
return true;
}
//////////////////////////
// class t_mime_database
//////////////////////////
t_mime_database::t_mime_database() {
set_separator(':');
set_filename(sys_config->get_mime_shared_database());
mime_magic_ = magic_open(MAGIC_MIME | MAGIC_ERROR);
if (mime_magic_ == (magic_t)NULL) {
log_file->write_report("Failed to open magic number database",
"t_mime_database::t_mime_database", LOG_NORMAL, LOG_WARNING);
return;
}
magic_load(mime_magic_, NULL);
}
t_mime_database::~t_mime_database() {
magic_close(mime_magic_);
}
void t_mime_database::add_record(const t_mime_db_record &record) {
map_mime2glob_.insert(make_pair(record.mimetype, record.file_glob));
}
string t_mime_database::get_glob(const string &mimetype) const {
map<string, string>::const_iterator it = map_mime2glob_.find(mimetype);
if (it != map_mime2glob_.end()) {
return it->second;
}
return "";
}
string t_mime_database::get_mimetype(const string &filename) const {
const char *mime_desc = magic_file(mime_magic_, filename.c_str());
if (!mime_desc) return "";
// Sometimes the magic libary adds additional info to the
// returned mime type. Strip this info.
string mime_type(mime_desc);
string::size_type end_of_mime = mime_type.find_first_not_of(
"abcdefghijklmnopqrstuvwxyz"
"ABCDEFGHIJKLMNOPQSTUVWXYZ"
"0123456789-.!%*_+`'~/");
if (end_of_mime != string::npos) {
mime_type = mime_type.substr(0, end_of_mime);
}
return mime_type;
}
|
