#include <mysql.h>
#include <stdio.h>
#include <stdlib.h>
#include <iostream>
#include <fstream>
#include <string>
#include <sstream>
#include <sys/stat.h>
int main(int argc, char* argv[]) {
// Check if the database name was passed on the cmd-line and if not output usage instructions
// and terminate.
if(argv[1] == NULL) {
std::cout << "Usage :\n";
std::cout << "./docbackup <database_name>\n\n";
std::cout << "(Where '<database_name>' is the name of the MySQL database to connect to.)\n";
return 1;
}
MYSQL *conn;
MYSQL_RES *res;
MYSQL_ROW row;
// Usernames and password changed to protect the guilty
// If you need somewhere to start to make this more useful, I'm sure you can
// work out how to pass the server, username and password on the cmd-line rather than
// being evil and hardcoding it. :)
const char *server = "localhost";
const char *user = "someuser";
const char *password = "somepass";
char *database = argv[1];
std::ofstream myfile;
std::ofstream logfile;
conn = mysql_init(NULL);
// Connect to the database
if(!mysql_real_connect(conn, server, user, password, database, 0, NULL, 0)) {
fprintf(stderr, "%s\n", mysql_error(conn));
return 1;
}
// Send SQL query
if(mysql_query(conn, "select uploads.content,uploads.name,LENGTH(uploads.content),clients.legalname,clients.tradingname from uploads LEFT JOIN clients ON uploads.client_id = clients.client_id where 1")) {
fprintf(stderr, "%s\n", mysql_error(conn));
return 1;
}
res = mysql_use_result(conn);
// Create a directory for this database to export files into
mkdir(argv[1],S_IRWXU | S_IRWXG | S_IROTH | S_IXOTH);
std::stringstream f;
f << argv[1] << "/" << "files.csv";
std::string logfilename = f.str();
// Output results
logfile.open(logfilename.c_str(), std::ios::out);
printf("files:\n");
while((row = mysql_fetch_row(res)) != NULL) {
std::string lname;
std::string tname;
if(row[3] != NULL)
lname = row[3];
if(row[4] != NULL)
tname = row[4];
std::stringstream s;
s << "\"" << row[1] << "\"," << "\"" << lname << "\",\"" << tname << "\"" << std::endl;
std::string log = s.str();
std::stringstream of;
of << argv[1] << "/" << row[1];
std::string outfilename = of.str();
myfile.open(outfilename.c_str(), std::ios::binary | std::ios::out );
int size = strtol(row[2],NULL,0);
myfile.write(row[0], size);
myfile.close();
printf("%s (%d)\n", row[1], size);
logfile.write(log.c_str(), log.size());
}
logfile.close();
// close connection
mysql_free_result(res);
mysql_close(conn);
return 1;
}
