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
|
#include "pak.h"
#include <boost/filesystem.hpp>
#include <algorithm>
#include <cstring>
#include <functional>
#include <iostream>
#include <iterator>
#include <stdexcept>
namespace {
void usage(char const* progname) {
std::cerr << "Usage: " << progname << " file.pak --extract [files]" << std::endl;
std::cerr << " " << progname << " file.pak --dump file" << std::endl;
std::cerr << " " << progname << " file.pak --list" << std::endl;
}
struct Extract {
Extract(Pak& p): m_p(p) {}
void operator()(std::string const& filename) { operator()(make_pair(filename, m_p[filename])); }
void operator()() {
for (Pak::files_t::const_iterator it = m_p.files().begin(); it != m_p.files().end(); ++it) operator()(*it);
}
void operator()(std::pair<std::string, PakFile> const& fp) {
std::string filename = fp.first;
// Remove path elements from m_path until it matches the filename's beginning
while (m_path != filename.substr(0, m_path.size())) {
std::string::size_type pos = m_path.rfind('/');
if (pos == std::string::npos) m_path.clear();
else m_path.erase(pos);
}
// Try to create new folders as required
for (std::string::size_type pos; (pos = filename.find('/', m_path.size() + 1)) != std::string::npos;) {
m_path = filename.substr(0, pos);
boost::filesystem::create_directory(m_path);
}
// Extract the file
std::ofstream f(filename.c_str(), std::ios::binary);
if (!f.is_open()) throw std::runtime_error("Unable to create file: " + filename);
std::cout << filename << std::flush;
std::vector<char> buf;
fp.second.get(buf);
f.write(&buf[0], buf.size());
std::cout << " " << buf.size() << " bytes" << std::endl;
}
private:
Pak& m_p;
std::string m_path;
};
}
int main(int argc, char** argv) {
std::ios::sync_with_stdio(false);
if( argc < 3 ) { usage(argv[0]); return EXIT_FAILURE; }
try {
Pak p(argv[1]);
if (!strcmp(argv[2],"--list")) std::cout << p.files();
else if (!strcmp(argv[2],"--dump")) {
if (argc != 4) { usage(argv[0]); return EXIT_FAILURE; }
std::vector<char> buf;
p[argv[3]].get(buf);
std::cout.write(&buf[0], buf.size());
} else if (!strcmp(argv[2],"--extract")) {
if (argc == 3) { (Extract(p))(); }
else std::for_each(argv + 3, argv + argc, Extract(p));
} else { usage(argv[0]); return EXIT_FAILURE; }
} catch (std::exception& e) {
std::cerr << "Error: " << e.what() << std::endl;
return EXIT_FAILURE;
}
}
|