| /* zenutils - Utilities for working with creative firmwares. |
| * Copyright 2007 (c) Rasmus Ry <rasmus.ry{at}gmail.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, write to the Free Software |
| * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA |
| */ |
| |
| #include <iostream> |
| #include <ctime> |
| #include <getpot/GetPot> |
| #include <utils.h> |
| #include <firmware.h> |
| |
| |
| static const char VERSION[] = "0.1"; |
| |
| |
| void print_version() |
| { |
| std::cout |
| << "firmware_extract - Extracts files from a Creative firmware." |
| << std::endl |
| << "Version " << VERSION << std::endl |
| << "Copyright (c) 2007 Rasmus Ry" << std::endl; |
| } |
| |
| void print_help() |
| { |
| print_version(); |
| std::cout |
| << "Usage: firmware_extract [command] [options]" << std::endl |
| << std::endl |
| << " Commands:" << std::endl |
| << " -h,--help" << std::endl |
| << " prints this message." << std::endl |
| << " -f,--firmware [file]" << std::endl |
| << " specifies the firmware arhive file name." << std::endl |
| << std::endl |
| << " Options:" << std::endl |
| << " -V,--verbose" << std::endl |
| << " prints verbose messages." << std::endl |
| << " -p,--prefix [prefix]" << std::endl |
| << " specifies a file name prefix for the extracted files." << std::endl |
| << std::endl |
| ; |
| } |
| |
| |
| struct save_entry_functor |
| { |
| save_entry_functor(const std::string& fileprefix) |
| : _fileprefix(fileprefix) {} |
| |
| bool operator()(const zen::firmware_entry& entry) |
| { |
| std::string filename = _fileprefix + entry.get_content_name(); |
| std::ofstream ofs; |
| ofs.open(filename.c_str(), std::ios::binary); |
| if (!ofs) |
| false; |
| |
| size_t off = entry.get_content_offset(); |
| std::streamsize size = entry.get_bytes().size() - off; |
| ofs.write((const char*)&entry.get_bytes()[off], size); |
| |
| return ofs.good(); |
| } |
| |
| const std::string& _fileprefix; |
| }; //struct save_entry_functor |
| |
| struct print_entry_functor |
| { |
| print_entry_functor(std::ostream& os, const std::string& fileprefix) |
| : _os(os), _fileprefix(fileprefix), num(0) {} |
| |
| bool operator()(const zen::firmware_entry& entry) |
| { |
| std::string filename = _fileprefix + entry.get_content_name(); |
| if (!num) |
| _os << "[./" << num++ << "]" << std::endl; |
| else |
| _os << "[../" << num++ << "]" << std::endl; |
| _os << "tag = " << entry.get_name() << std::endl; |
| |
| if (entry.get_content_offset()) |
| _os << "name = " << entry.get_content_name() << std::endl; |
| |
| _os << "file = \'" << shared::double_quote(filename) << "\'" |
| << std::endl; |
| |
| return _os.good(); |
| } |
| |
| std::ostream& _os; |
| const std::string& _fileprefix; |
| int num; |
| }; //struct print_entry_functor |
| |
| |
| int process_arguments(int argc, char* argv[]) |
| { |
| //-------------------------------------------------------------------- |
| // Parse input variables. |
| //-------------------------------------------------------------------- |
| |
| GetPot cl(argc, argv); |
| if (cl.size() == 1 || cl.search(2, "-h", "--help")) |
| { |
| print_help(); |
| return 1; |
| } |
| |
| std::string firmwarename; |
| if (cl.search("-f") || cl.search("--firmware")) |
| firmwarename = cl.next(""); |
| if (firmwarename.empty()) |
| { |
| std::cerr << "Firmware archive must be specified." << std::endl; |
| return 2; |
| } |
| |
| bool verbose = false; |
| if (cl.search("-V") || cl.search("--verbose")) |
| verbose = true; |
| |
| std::string prefixname = shared::remove_extension(firmwarename) + "_"; |
| if (cl.search("-p") || cl.search("--prefix")) |
| prefixname = cl.next(prefixname.c_str()); |
| |
| |
| //-------------------------------------------------------------------- |
| // Read the firmware archive. |
| //-------------------------------------------------------------------- |
| |
| if (verbose) |
| std::cout << "[*] Reading firmware archive..." << std::endl; |
| |
| zen::firmware_archive archive(false); |
| std::ifstream ifs; |
| ifs.open(firmwarename.c_str(), std::ios::binary); |
| if (!ifs) |
| { |
| std::cerr << "Failed to open the firmware archive." << std::endl; |
| return 3; |
| } |
| |
| if (!archive.read(ifs)) |
| { |
| std::cerr << "Failed to read the firmware archive." << std::endl; |
| return 4; |
| } |
| |
| |
| //-------------------------------------------------------------------- |
| // Generate a make file for the extracted firmware archive. |
| //-------------------------------------------------------------------- |
| |
| // Get make filename for the given input file. |
| std::string makefile = shared::replace_extension(firmwarename, ".mk"); |
| |
| if (verbose) |
| std::cout << "[*] Producing make file..." << std::endl; |
| |
| |
| // Produce make file for the given input file. |
| std::ofstream ofs; |
| ofs.open(makefile.c_str(), std::ios::binary); |
| if (!ofs) |
| { |
| std::cerr << "Failed to create firmware archive make file." |
| << std::endl; |
| return 5; |
| } |
| |
| time_t timeval = time(NULL); |
| ofs << "# Make file generated at: " << ctime(&timeval); |
| ofs << "endian = " << (archive.is_big_endian() ? "big" : "little") |
| << std::endl; |
| ofs << "signed = " << (archive.is_signed() ? "true" : "false") |
| << std::endl; |
| |
| ofs << "[children]" << std::endl; |
| ofs << "count = " << archive.get_children().size() << std::endl; |
| |
| std::for_each(archive.get_children().begin(), |
| archive.get_children().end(), |
| print_entry_functor(ofs, prefixname)); |
| |
| ofs << "[neighbours]" << std::endl; |
| ofs << "count = " << archive.get_neighbours().size() << std::endl; |
| std::for_each(archive.get_neighbours().begin(), |
| archive.get_neighbours().end(), |
| print_entry_functor(ofs, prefixname)); |
| |
| |
| //-------------------------------------------------------------------- |
| // Save firmware entries. |
| //-------------------------------------------------------------------- |
| |
| if (verbose) |
| std::cout << "[*] Saving firmware entries..." << std::endl; |
| |
| std::for_each(archive.get_children().begin(), |
| archive.get_children().end(), |
| save_entry_functor(prefixname)); |
| |
| std::for_each(archive.get_neighbours().begin(), |
| archive.get_neighbours().end(), |
| save_entry_functor(prefixname)); |
| |
| return 0; |
| } |
| |
| int main(int argc, char* argv[]) |
| { |
| try |
| { |
| return process_arguments(argc, argv); |
| } |
| catch (const std::exception& xcpt) |
| { |
| std::cerr << "Exception caught: " << xcpt.what() << std::endl; |
| return -1; |
| } |
| catch (...) |
| { |
| std::cerr << "Unknown exception caught." << std::endl; |
| return -2; |
| } |
| return -3; |
| } |