158 lines
3.5 KiB
C++
158 lines
3.5 KiB
C++
/*
|
|
----
|
|
This file is part of SECONDO.
|
|
|
|
Copyright (C) 2004, University in Hagen, Department of Computer Science,
|
|
Database Systems for New Applications.
|
|
|
|
SECONDO 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.
|
|
|
|
SECONDO 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 SECONDO; if not, write to the Free Software
|
|
Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
----
|
|
|
|
*/
|
|
#include <Base64.h>
|
|
#include <fstream>
|
|
#include <iostream>
|
|
#include <unistd.h>
|
|
#include <assert.h>
|
|
#include <sstream>
|
|
#include <cstdlib>
|
|
|
|
using namespace std;
|
|
|
|
|
|
|
|
static void
|
|
usageMsg(char* progName) {
|
|
|
|
cout << "Usage: " << progName << " [-d -n<size>] filename" << endl;
|
|
cout << "Converts a file into base64 format and vice versa." << endl;
|
|
cout << "Options : " << endl;
|
|
cout << " -d : decode" << endl;
|
|
cout << " -n<size>: Don't use the stream based methods "
|
|
<< "of class Base64. " << endl;
|
|
cout << " Read the file into a buffer of size <size>" << endl;
|
|
}
|
|
|
|
|
|
int
|
|
main(int argc, char *argv[]) {
|
|
|
|
|
|
if (argc < 2 || argc > 4) {
|
|
usageMsg(argv[0]);
|
|
exit(1);
|
|
}
|
|
|
|
|
|
|
|
#ifdef _POSIX_OPT_H
|
|
int c = 0;
|
|
opterr = 0;
|
|
|
|
while ((c = getopt(argc,argv, "dn:")) != -1) {
|
|
switch (c) {
|
|
case 'd':
|
|
decode = true;
|
|
break;
|
|
case 'n':
|
|
nostream = true;
|
|
size = atoi(optarg);
|
|
break;
|
|
case '?':
|
|
cerr << "Unkown option " << (char) optopt << endl;
|
|
usageMsg(argv[0]);
|
|
default:
|
|
exit(1);
|
|
}
|
|
}
|
|
|
|
string fileName(argv[optind]);
|
|
string extension="";
|
|
cout << "file: " << fileName << ", size: " << size << endl;
|
|
decode ? extension=".bin" : extension=".base64";
|
|
|
|
|
|
ifstream inFile(fileName.c_str(), ios::binary);
|
|
ofstream outFile((fileName+extension).c_str(), ios::binary);
|
|
|
|
Base64* encoder = new Base64();
|
|
|
|
if (decode) {
|
|
|
|
if (nostream) {
|
|
string text="";
|
|
int bufSize = 0;
|
|
char* buffer=0;
|
|
int readFaults=0;
|
|
const int inLength=1024;
|
|
char textPart[inLength+1];
|
|
|
|
do {
|
|
|
|
inFile.read(textPart, inLength);
|
|
int noBytes = inFile.gcount();
|
|
if (noBytes != inLength) {
|
|
readFaults++;
|
|
}
|
|
assert( readFaults <= 1 );
|
|
|
|
textPart[noBytes]=0;
|
|
text += textPart;
|
|
|
|
} while (inFile.good());
|
|
|
|
bufSize = text.length();
|
|
bufSize=encoder->sizeDecoded(text.length());
|
|
cout << "Text: " << text.length() << ", bufSize: " << bufSize << endl;
|
|
buffer = new char[bufSize];
|
|
|
|
bufSize = encoder->decode(text, buffer);
|
|
outFile.write(buffer, bufSize);
|
|
delete [] buffer;
|
|
|
|
} else {
|
|
|
|
encoder->decodeStream(inFile,outFile);
|
|
}
|
|
|
|
} else {
|
|
|
|
char* buffer = 0;
|
|
if (nostream) {
|
|
cout << "nostreams ..." << endl;
|
|
buffer = new char[size+1];
|
|
inFile.read(buffer,size);
|
|
assert (inFile.gcount() == size);
|
|
string text = "";
|
|
encoder->encode(buffer, size, text);
|
|
outFile << text;
|
|
delete [] buffer;
|
|
|
|
} else {
|
|
encoder->encodeStream(inFile,outFile);
|
|
}
|
|
}
|
|
|
|
delete encoder;
|
|
|
|
#else
|
|
cerr << "You will need the posix getopt library!" << endl;
|
|
exit(1);
|
|
#endif
|
|
|
|
}
|
|
|
|
|