mirror of
https://gitea.home.endeavr.de/Marcel/Tarstats-pp.git
synced 2025-12-10 00:39:01 +01:00
this added output to text file and proper parsing of command line arguments. this also cleaned up some more stuff.
This commit is contained in:
Marcel Nowicki
141
main.cpp
141
main.cpp
@@ -3,7 +3,6 @@
|
||||
*/
|
||||
|
||||
#include <iostream>
|
||||
#include <fstream>
|
||||
#include <string>
|
||||
#include <vector>
|
||||
#include <map>
|
||||
@@ -12,71 +11,109 @@
|
||||
#include "tarfunc.h"
|
||||
|
||||
int main(int argc, char** argv) {
|
||||
// Trivial check for arguments.
|
||||
if (argc < 2) {
|
||||
tar::printhelp();
|
||||
return 9;
|
||||
}
|
||||
|
||||
// GENERAL VARIABLES
|
||||
//count the types of all items
|
||||
std::map<std::string, uintmax_t> typecount{
|
||||
{tarconstant::typeFile, 0}, {tarconstant::typeDir, 0}, {tarconstant::typeSym, 0},
|
||||
{tarconstant::typeHard, 0}, {tarconstant::typeOther, 0}
|
||||
};
|
||||
|
||||
uint64_t sizeof_allfiles{}; // total size of all files in the archive
|
||||
|
||||
// Trivial check for arguments. Errorprone and has to be changed.
|
||||
if (argc < 2) {
|
||||
std::cout << tarconstant::helptext << '\n';
|
||||
return 9;
|
||||
// Load command line parameters into vector, read them and set variables
|
||||
std::vector<std::string> cmdparam{};
|
||||
for (int i = 1; i < argc; i++){
|
||||
cmdparam.emplace_back(argv[i]);
|
||||
}
|
||||
// Getting name from argument lists on startup. Trivial and errorprone. Placeholder for now.
|
||||
std::string archiveFilename(argv[1]);
|
||||
bool toFile = false;
|
||||
bool toJSON = false;
|
||||
|
||||
for (int i = 0; i < argc; i++){
|
||||
std::cout << argv[i] << '\n';
|
||||
std::vector<std::string> archiveFilename {};
|
||||
for (auto i : cmdparam) {
|
||||
if (i[0] != '-' && i.size() > 1) {
|
||||
archiveFilename.push_back(i);
|
||||
}
|
||||
if (i[0] == '-') {
|
||||
for (auto j : i) {
|
||||
switch (j) {
|
||||
case 'j':
|
||||
toJSON = true;
|
||||
break;
|
||||
case 'f':
|
||||
toFile = true;
|
||||
break;
|
||||
case 'h':
|
||||
tar::printhelp();
|
||||
return 9;
|
||||
case '-':
|
||||
break;
|
||||
default:
|
||||
std::cout << "Invalid optional argument '" << j << "'!" << '\n' << '\n';
|
||||
tar::printhelp();
|
||||
return 9;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
//Open tar File.
|
||||
std::ifstream file(archiveFilename, std::ios::binary);
|
||||
if(!file) {
|
||||
std::cout << "Error opening file!" << '\n' << '\n';
|
||||
std::cout << tarconstant::helptext << '\n';
|
||||
return 9;
|
||||
}
|
||||
for (auto &archiveName : archiveFilename) {
|
||||
//count the types of all items
|
||||
std::map<std::string, uintmax_t> typecount{
|
||||
{tarconstant::typeFile, 0}, {tarconstant::typeDir, 0}, {tarconstant::typeSym, 0},
|
||||
{tarconstant::typeHard, 0}, {tarconstant::typeOther, 0}
|
||||
};
|
||||
|
||||
if (!tar::validTar(file)) {
|
||||
std::cout << archiveFilename << " is not a valid tar file for tarstats-pp" << '\n';
|
||||
return 9;
|
||||
}
|
||||
uint64_t sizeof_allfiles{}; // total size of all files in the archive
|
||||
|
||||
while (file) {
|
||||
//Read header of next item in tar archive
|
||||
char *headbuffer = new char[tarconstant::blocksize];
|
||||
file.read(headbuffer, tarconstant::blocksize);
|
||||
|
||||
// tar file ends with 2 512byte blocks of 0. As no block should ever be 0 unless at the end, we check only once.
|
||||
if (tar::eof(headbuffer)) {
|
||||
break;
|
||||
//Open tar File.
|
||||
std::ifstream file(archiveName, std::ios::binary);
|
||||
if (!file) {
|
||||
std::cout << "Error opening file " << archiveName << "!" << '\n' << '\n';
|
||||
tar::printhelp();
|
||||
return 9;
|
||||
}
|
||||
|
||||
// Read type of item
|
||||
std::string itemtype = tar::getitemtype(headbuffer[tarconstant::itemtypeByte]);
|
||||
typecount[itemtype] += 1;
|
||||
|
||||
// read itemsize and add to total
|
||||
sizeof_allfiles += tar::getitemsize(headbuffer);
|
||||
|
||||
// ignore file content. we want to get to the next header. item types != FILE have no content blocks (0 byte)
|
||||
if (tar::getitemsize(headbuffer)!=0) {
|
||||
file.ignore((tar::getitemsize(headbuffer) / tarconstant::blocksize)
|
||||
* tarconstant::blocksize + tarconstant::blocksize);
|
||||
if (!tar::validTar(file)) {
|
||||
std::cout << "!!!!!!!" << '\n';
|
||||
std::cout << archiveName << " is not a valid tar file for tarstats-pp" << '\n';
|
||||
std::cout << "!!!!!!!" << '\n' << '\n';
|
||||
}
|
||||
else {
|
||||
while (file) {
|
||||
//Read header of next item in tar archive
|
||||
char *headbuffer = new char[tarconstant::blocksize];
|
||||
file.read(headbuffer, tarconstant::blocksize);
|
||||
|
||||
// empty the headerbuffer
|
||||
delete[] headbuffer;
|
||||
// tar file ends with 2 512byte blocks of 0. As no block should ever be 0 unless at the end, we check only once.
|
||||
if (tar::eof(headbuffer)) {
|
||||
break;
|
||||
}
|
||||
|
||||
// Read type of item
|
||||
std::string itemtype = tar::getitemtype(headbuffer[tarconstant::itemtypeByte]);
|
||||
typecount[itemtype] += 1;
|
||||
|
||||
// read itemsize and add to total
|
||||
sizeof_allfiles += tar::getitemsize(headbuffer);
|
||||
|
||||
// ignore file content. we want to get to the next header. item types != FILE have no content blocks (0 byte)
|
||||
if (tar::getitemsize(headbuffer) != 0) {
|
||||
file.ignore((tar::getitemsize(headbuffer) / tarconstant::blocksize)
|
||||
* tarconstant::blocksize + tarconstant::blocksize);
|
||||
}
|
||||
|
||||
// empty the headerbuffer
|
||||
delete[] headbuffer;
|
||||
}
|
||||
file.close();
|
||||
|
||||
std::cout << toJSON << '\n';
|
||||
tar::consolestats(typecount, std::filesystem::file_size(archiveName), sizeof_allfiles);
|
||||
if (toFile) {
|
||||
tar::txtfilestats(typecount, std::filesystem::file_size(archiveName), sizeof_allfiles,
|
||||
archiveName);
|
||||
}
|
||||
}
|
||||
}
|
||||
file.close();
|
||||
|
||||
tar::consolestats(typecount, std::filesystem::file_size(archiveFilename), sizeof_allfiles);
|
||||
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user