diff options
Diffstat (limited to 'lib/Utils')
-rw-r--r-- | lib/Utils/DebugDump.cpp | 247 | ||||
-rw-r--r-- | lib/Utils/DebugDump.h | 72 | ||||
-rw-r--r-- | lib/Utils/Settings.cpp | 64 | ||||
-rw-r--r-- | lib/Utils/Settings.h | 13 |
4 files changed, 396 insertions, 0 deletions
diff --git a/lib/Utils/DebugDump.cpp b/lib/Utils/DebugDump.cpp new file mode 100644 index 00000000..e35ec4dd --- /dev/null +++ b/lib/Utils/DebugDump.cpp @@ -0,0 +1,247 @@ +/* + DebugDump.cpp + + Copyright (C) 2009 Zdenek Prikryl (zprikryl@redhat.com) + Copyright (C) 2009 RedHat inc. + + 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., + 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. + */ + +#include "DebugDump.h" +#include <fstream> +#include <iostream> +#include <sstream> +#include <sys/types.h> +#include <sys/stat.h> +#include <dirent.h> +#include <sys/utsname.h> +#include <limits.h> +#include <fcntl.h> +#include <sys/procfs.h> +#include <ctype.h> + +CDebugDump::CDebugDump() : + m_sDebugDumpDir("") +{} + +void CDebugDump::Open(const std::string& pDir) +{ + m_sDebugDumpDir = pDir; + if (!ExistFileDir(pDir)) + { + throw "CDebugDump::CDebugDump(): "+pDir+" does not exist."; + } +} + +bool CDebugDump::Exist(const std::string& pPath) +{ + std::string fullPath = m_sDebugDumpDir + "/" + pPath; + return ExistFileDir(fullPath); +} + + +bool CDebugDump::ExistFileDir(const std::string& pPath) +{ + struct stat buf; + if (stat(pPath.c_str(), &buf) == 0) + { + if (S_ISDIR(buf.st_mode) || S_ISREG(buf.st_mode)) + { + return true; + } + } + return false; +} + +void CDebugDump::Create(const std::string& pDir) +{ + m_sDebugDumpDir = pDir; + Delete(pDir); + if (mkdir(pDir.c_str(), 0755) == -1) + { + throw "CDebugDump::Create(): Cannot create dir: " + pDir; + } + SaveEnvironment(); +} + +void CDebugDump::Delete(const std::string& pDir) +{ + if (!ExistFileDir(pDir)) + { + return; + } + + DIR *dir = opendir(pDir.c_str()); + std::string fullPath; + struct dirent *dent = NULL; + if (dir != NULL) + { + while ((dent = readdir(dir)) != NULL) + { + if (std::string(dent->d_name) != "." && std::string(dent->d_name) != "..") + { + fullPath = pDir + "/" + dent->d_name; + if (dent->d_type == DT_DIR) + { + Delete(fullPath); + } + if (remove(fullPath.c_str()) == -1) + { + throw "CDebugDump::DeleteDir(): Cannot remove file: " + fullPath; + } + } + } + closedir(dir); + if (remove(pDir.c_str()) == -1) + { + throw "CDebugDump::DeleteDir(): Cannot remove dir: " + fullPath; + } + } +} + +void CDebugDump::SaveEnvironment() +{ + struct utsname buf; + if (uname(&buf) == 0) + { + SaveText(FILENAME_KERNEL, buf.release); + SaveText(FILENAME_ARCHITECTURE, buf.machine); + } +} + +void CDebugDump::LoadTextFile(const std::string& pPath, std::string& pData) +{ + std::ifstream fIn; + pData = ""; + fIn.open(pPath.c_str()); + if (fIn.is_open()) + { + std::string line; + while (!fIn.eof()) + { + getline (fIn,line); + pData += line; + } + fIn.close(); + } + else + { + throw "CDebugDump: LoadTextFile(): Cannot open file " + pPath; + } +} + +void CDebugDump::LoadBinaryFile(const std::string& pPath, char** pData, unsigned int* pSize) +{ + std::ifstream fIn; + fIn.open(pPath.c_str(), std::ios::binary | std::ios::ate); + unsigned int size; + if (fIn.is_open()) + { + size = fIn.tellg(); + char *data = new char [size]; + fIn.read(data, size); + + *pData = data; + *pSize = size; + + fIn.close(); + } + else + { + throw "CDebugDump: LoadBinaryFile(): Cannot open file " + pPath; + } +} + + +void CDebugDump::SaveTextFile(const std::string& pPath, const std::string& pData) +{ + std::ofstream fOut; + fOut.open(pPath.c_str()); + if (fOut.is_open()) + { + fOut << pData; + fOut.close(); + } + else + { + throw "CDebugDump: SaveTextFile(): Cannot open file " + pPath; + } +} + +void CDebugDump::SaveBinaryFile(const std::string& pPath, const char* pData, const unsigned pSize) +{ + std::ofstream fOut; + fOut.open(pPath.c_str(), std::ios::binary); + if (fOut.is_open()) + { + fOut.write(pData, pSize); + fOut.close(); + } + else + { + throw "CDebugDump: SaveBinaryFile(): Cannot open file " + pPath; + } +} + +void CDebugDump::LoadText(const std::string& pName, std::string& pData) +{ + std::string fullPath = m_sDebugDumpDir + "/" + pName; + LoadTextFile(fullPath, pData); +} +void CDebugDump::LoadBinary(const std::string& pName, char** pData, unsigned int* pSize) +{ + std::string fullPath = m_sDebugDumpDir + "/" + pName; + LoadBinaryFile(fullPath, pData, pSize); +} + +void CDebugDump::SaveText(const std::string& pName, const std::string& pData) +{ + std::string fullPath = m_sDebugDumpDir + "/" + pName; + SaveTextFile(fullPath, pData); +} +void CDebugDump::SaveBinary(const std::string& pName, const char* pData, const unsigned int pSize) +{ + std::string fullPath = m_sDebugDumpDir + "/" + pName; + SaveBinaryFile(fullPath, pData, pSize); +} + + +void CDebugDump::SaveProc(const std::string& pPID) +{ + std::string path = "/proc/"+pPID+"/exe"; + std::string data; + char executable[PATH_MAX]; + + if (readlink(path.c_str(), executable, PATH_MAX) == 0) + { + SaveText(FILENAME_EXECUTABLE, executable); + } + + path = "/proc/"+pPID+"/status"; + std::string uid = "0"; + int ii = 0; + + LoadTextFile(path, data); + data = data.substr(data.find("Uid:")+5); + + while (!isspace(data[ii])) + { + uid += data[ii]; + ii++; + } + SaveText(FILENAME_UID, uid); + + // TODO: Use packagekit +} diff --git a/lib/Utils/DebugDump.h b/lib/Utils/DebugDump.h new file mode 100644 index 00000000..c8eef157 --- /dev/null +++ b/lib/Utils/DebugDump.h @@ -0,0 +1,72 @@ +/* + DebugDump.h - header file for the library caring of writing new reports + to the specific directory + + Copyright (C) 2009 Zdenek Prikryl (zprikryl@redhat.com) + Copyright (C) 2009 RedHat inc. + + 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., + 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. + */ + +#ifndef DEBUGDUMP_H_ +#define DEBUGDUMP_H_ + +#include <string> + +#define FILENAME_ARCHITECTURE "architecture" +#define FILENAME_KERNEL "kernel" +#define FILENAME_EXECUTABLE "executable" +#define FILENAME_TIME "time" +#define FILENAME_UID "uid" +#define FILENAME_PACKAGE "package" +#define FILENAME_HASH "hash" +#define FILENAME_LANGUAGE "language" +#define FILENAME_APPLICATION "application" +#define FILENAME_TEXTDATA1 "text_data1" +#define FILENAME_BINARYDATA1 "binary_data1" + +class CDebugDump +{ + private: + std::string m_sDebugDumpDir; + + void SaveEnvironment(); + + + void LoadTextFile(const std::string& pName, std::string& pData); + void LoadBinaryFile(const std::string& pName, char** pData, unsigned int* pSize); + + void SaveTextFile(const std::string& pName, const std::string& pData); + void SaveBinaryFile(const std::string& pName, const char* pData, const unsigned int pSize); + bool ExistFileDir(const std::string& pPath); + + public: + + CDebugDump(); + void Open(const std::string& pDir); + void Create(const std::string& pDir); + void Delete(const std::string& pDir); + bool Exist(const std::string& pPath); + + void LoadText(const std::string& pName, std::string& pData); + void LoadBinary(const std::string& pName, char** pData, unsigned int* pSize); + + void SaveText(const std::string& pName, const std::string& pData); + void SaveBinary(const std::string& pName, const char* pData, const unsigned int pSize); + + void SaveProc(const std::string& pPID); +}; + +#endif /*DEBUGDUMP_H_*/ diff --git a/lib/Utils/Settings.cpp b/lib/Utils/Settings.cpp new file mode 100644 index 00000000..866adc57 --- /dev/null +++ b/lib/Utils/Settings.cpp @@ -0,0 +1,64 @@ + +#include "Settings.h" +#include <fstream> + + +void load_settings(const std::string& path, map_settings_t& settings) +{ + std::ifstream fIn; + fIn.open(path.c_str()); + if (fIn.is_open()) + { + std::string line; + while (!fIn.eof()) + { + getline(fIn, line); + + int ii; + bool is_value = false; + bool valid = false; + std::string key = ""; + std::string value = ""; + for (ii = 0; ii < line.length(); ii++) + { + if (isspace(line[ii])) + { + continue; + } + if (line[ii] == '#') + { + break; + } + else if (line[ii] == '=') + { + is_value = true; + } + else if (line[ii] == '=' && is_value) + { + key = ""; + value = ""; + break; + } + else if (!is_value) + { + key += line[ii]; + } + else + { + valid = true; + value += line[ii]; + } + } + if (valid) + { + settings[key] = value; + } + } + fIn.close(); + } +} + +void save_settings(const std::string& path, const map_settings_t& settings) +{ + //TODO: write this +} diff --git a/lib/Utils/Settings.h b/lib/Utils/Settings.h new file mode 100644 index 00000000..ec30a27a --- /dev/null +++ b/lib/Utils/Settings.h @@ -0,0 +1,13 @@ +#ifndef SETTINGSFUNC_H_ +#define SETTINGSFUNC_H_ + +#include "Settings.h" +#include <string> +#include <map> + +typedef std::map<std::string, std::string> map_settings_t; + +void load_settings(const std::string& path, map_settings_t& settings); +void save_settings(const std::string& path, const map_settings_t& settings); + +#endif /* SETTINGSFUNC_H_ */ |