summaryrefslogtreecommitdiffstats
path: root/src/utils/mime_database.cpp
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;
}