version 1.22.0
source tree changed detached PIConsole and PIScreen* in "pip_console" library
This commit is contained in:
335
lib/main/io_utils/pifiletransfer.cpp
Normal file
335
lib/main/io_utils/pifiletransfer.cpp
Normal file
@@ -0,0 +1,335 @@
|
||||
/*
|
||||
PIP - Platform Independent Primitives
|
||||
Class for send and receive files and directories via PIBaseTransfer
|
||||
Andrey Bychkov work.a.b@yandex.ru
|
||||
|
||||
This program is free software: you can redistribute it and/or modify
|
||||
it under the terms of the GNU Lesser General Public License as published by
|
||||
the Free Software Foundation, either version 3 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 Lesser General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU Lesser General Public License
|
||||
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
#include "pifiletransfer.h"
|
||||
|
||||
const char PIFileTransfer::sign[] = {'P', 'F', 'T'};
|
||||
|
||||
PIFileTransfer::PIFileTransfer() {
|
||||
for (uint i = 0; i < sizeof(sign); i++)
|
||||
pftheader.sig[i] = sign[i];
|
||||
pftheader.version = __PIFILETRANSFER_VERSION;
|
||||
pftheader.session_id = 0;
|
||||
pftheader.step = pft_None;
|
||||
dir = PIDir::current();
|
||||
started_ = scanning = false;
|
||||
bytes_file_all = bytes_file_cur = 0;
|
||||
CONNECT(void, this, receiveStarted, this, receive_started);
|
||||
CONNECT1(void, bool, this, sendFinished, this, send_finished);
|
||||
CONNECT1(void, bool, this, receiveFinished, this, receive_finished);
|
||||
}
|
||||
|
||||
|
||||
PIFileTransfer::~PIFileTransfer() {
|
||||
stop();
|
||||
work_file.close();
|
||||
}
|
||||
|
||||
|
||||
bool PIFileTransfer::send(const PIFile& file) {
|
||||
return send(file.path());
|
||||
}
|
||||
|
||||
|
||||
bool PIFileTransfer::send(const PIString & file) {
|
||||
return send(PIVector<PIFile::FileInfo>() << PIFile::fileInfo(file));
|
||||
}
|
||||
|
||||
|
||||
bool PIFileTransfer::send(const PIStringList& files) {
|
||||
PIVector<PIFile::FileInfo> fil;
|
||||
piForeachC(PIString &file, files) fil << PIFile::fileInfo(file);
|
||||
return send(fil);
|
||||
}
|
||||
|
||||
|
||||
bool PIFileTransfer::send(PIVector<PIFile::FileInfo> entries) {
|
||||
started_ = true;
|
||||
PIVector<PFTFileInfo> allEntries;
|
||||
for (int i = 0; i < entries.size_s(); i++) {
|
||||
allEntries << PFTFileInfo(entries[i]);
|
||||
allEntries.back().dest_path = entries[i].name();
|
||||
if (entries[i].isDir()) {
|
||||
cur_file_string = "scanning ... ";
|
||||
scan_dir.setDir(entries[i].path);
|
||||
scanning = true;
|
||||
PIVector<PIFile::FileInfo> fls = scan_dir.allEntries();
|
||||
scanning = false;
|
||||
scan_dir.up();
|
||||
//piCout << "files rel to" << d.absolutePath();
|
||||
for (uint j = 0; j < fls.size(); j++) {
|
||||
allEntries << PFTFileInfo(fls[j]);
|
||||
allEntries.back().dest_path = scan_dir.relative(fls[j].path);
|
||||
//piCout << " abs path" << fls[j].path;
|
||||
//piCout << " rel path" << allEntries.back().dest_path;
|
||||
}
|
||||
}
|
||||
}
|
||||
return sendFiles(allEntries);
|
||||
}
|
||||
|
||||
|
||||
bool PIFileTransfer::sendFiles(const PIVector<PFTFileInfo> &files) {
|
||||
files_ = files;
|
||||
PIStringList names;
|
||||
//piCoutObj << "prepare to send" << files_.size() << "files";
|
||||
for (int i = 0; i < files_.size_s(); i++) {
|
||||
if (names.contains(files_[i].path)) {
|
||||
files_.remove(i);
|
||||
--i;
|
||||
}
|
||||
else names << files_[i].path;
|
||||
if (files_[i].isDir())
|
||||
files_[i].size = 0;
|
||||
//piCout << "prepare" << i << files_[i].path << files_[i].dest_path << files_[i].name();
|
||||
}
|
||||
randomize();
|
||||
step_mutex.lock();
|
||||
pftheader.session_id = randomi();
|
||||
sendFilesStarted();
|
||||
cur_file_string = "build session";
|
||||
desc.clear();
|
||||
desc << files_;
|
||||
pftheader.step = pft_Description;
|
||||
buildSession(PIVector<Part>() << Part(0, desc.size()));
|
||||
cur_file_string = "";
|
||||
step_mutex.unlock();
|
||||
if (!send_process()) {
|
||||
sendFilesFinished(false);
|
||||
started_ = false;
|
||||
return false;
|
||||
}
|
||||
step_mutex.lock();
|
||||
pftheader.step = pft_Data;
|
||||
PIVector<Part> pts;
|
||||
for (int i = 0; i < files_.size_s(); i++) {
|
||||
pts << Part(i + 1, files_[i].size);
|
||||
}
|
||||
buildSession(pts);
|
||||
step_mutex.unlock();
|
||||
bool ok = send_process();
|
||||
sendFilesFinished(ok);
|
||||
started_ = false;
|
||||
return ok;
|
||||
}
|
||||
|
||||
|
||||
void PIFileTransfer::processFile(int id, ullong start, PIByteArray & data) {
|
||||
//piCout << "processFile" << id << files_.size();
|
||||
PFTFileInfo fi = files_[id-1];
|
||||
bytes_file_all = fi.size;
|
||||
bytes_file_cur = start;
|
||||
cur_file_string = fi.dest_path;
|
||||
PIString path = dir.absolutePath() + dir.separator + fi.dest_path;
|
||||
//piCout << "receive" << path << fi.size << data.size();
|
||||
if (fi.isDir()) {
|
||||
//piCoutObj << "make dir" << fi.entry.path;
|
||||
if (!PIDir::make(path)) {
|
||||
cur_file_string = "Error: While create directory \"" + path + "\"";
|
||||
piCoutObj << cur_file_string;
|
||||
stopReceive();
|
||||
return;
|
||||
}
|
||||
}
|
||||
if (fi.isFile()) {
|
||||
if (work_file.path() != path || !work_file.isOpened()) {
|
||||
//piCout << "file close";
|
||||
work_file.close();
|
||||
//piCout << "new file" << path << work_file.path() << work_file.isOpened();
|
||||
if (!work_file.open(path, PIIODevice::ReadWrite)) {
|
||||
cur_file_string = "Error: While open file \"" + path + "\"";
|
||||
piCoutObj << cur_file_string;
|
||||
stopReceive();
|
||||
return;
|
||||
}
|
||||
PIFile::applyFileInfo(path, fi);
|
||||
if (work_file.size() > fi.size) {
|
||||
//piCoutObj << "error size" << work_file.size() << fi.size;
|
||||
work_file.clear();
|
||||
work_file.resize(fi.size);
|
||||
//piCoutObj << "correct size" << work_file.size() << fi.size;
|
||||
}
|
||||
}
|
||||
//piCoutObj << "write file" << path << work_file.path() << work_file.size() << fi.entry.size << work_file.pos() << fi.fstart << fi.fsize;
|
||||
if (work_file.size() < (llong)start) {
|
||||
//piCoutObj << "error pos size" << work_file.pos() << fi.fstart;
|
||||
work_file.resize(start);
|
||||
//piCoutObj << "correct pos size" << work_file.pos() << fi.fstart;
|
||||
}
|
||||
if (work_file.size() > fi.size) {
|
||||
piCoutObj << "****** error size" << work_file.size() << fi.size;
|
||||
work_file.clear();
|
||||
work_file.resize(fi.size);
|
||||
piCoutObj << "****** correct size" << work_file.size() << fi.size;
|
||||
}
|
||||
//if (fi.fstart != work_file.pos()) piCoutObj << "error pos" << work_file.pos() << fi.fstart;
|
||||
work_file.seek(start);
|
||||
int rs = work_file.write(data.data(), data.size());
|
||||
if (rs != data.size_s()) {
|
||||
cur_file_string = "Error: While write file \"" + fi.path + "\" (must " + PIString::fromNumber(data.size()) + ", but write " + PIString::fromNumber(rs) + ")";
|
||||
piCoutObj << cur_file_string;
|
||||
stopReceive();
|
||||
return;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
PIByteArray PIFileTransfer::buildPacket(Part p) {
|
||||
//piCoutObj << "Step" << pftheader.step;
|
||||
//piCoutObj << "session id" << pftheader.session_id;
|
||||
PIByteArray ba;
|
||||
switch(pftheader.step) {
|
||||
case pft_None:
|
||||
stopSend();
|
||||
return PIByteArray();
|
||||
case pft_Description:
|
||||
ba.resize(p.size);
|
||||
memcpy(ba.data(), desc.data(p.start), p.size);
|
||||
return ba;
|
||||
case pft_Data:
|
||||
//piCout << "send data" << p.id << files_.size();
|
||||
PIFile::FileInfo fi = files_[p.id-1];
|
||||
if (fi.isFile()) {
|
||||
//piCout << "send file" << fi.name() << fi.size;
|
||||
PIString path = fi.path;
|
||||
if (work_file.path() != path || !work_file.isOpened()) {
|
||||
if (!work_file.open(path, PIIODevice::ReadOnly)) {
|
||||
break_ = true;
|
||||
cur_file_string = "Error: While open file \"" + fi.path + "\"";
|
||||
piCoutObj << cur_file_string;
|
||||
stopSend();
|
||||
return PIByteArray();
|
||||
}
|
||||
}
|
||||
work_file.seek(p.start);
|
||||
ba.resize(p.size);
|
||||
int rs = work_file.read(ba.data(), ba.size());
|
||||
if ((llong)rs != (llong)p.size) {
|
||||
break_ = true;
|
||||
cur_file_string = "Error: While read file \"" + fi.path + "\" (must " + PIString::fromNumber(p.size) + ", but read " + PIString::fromNumber(rs) + ")";
|
||||
piCoutObj << cur_file_string;
|
||||
stopSend();
|
||||
return PIByteArray();
|
||||
}
|
||||
}
|
||||
cur_file_string = fi.path;
|
||||
bytes_file_all = fi.size;
|
||||
bytes_file_cur = p.start;
|
||||
return ba;
|
||||
}
|
||||
return ba;
|
||||
}
|
||||
|
||||
|
||||
void PIFileTransfer::receivePart(PIBaseTransfer::Part fi, PIByteArray ba, PIByteArray pheader) {
|
||||
PFTHeader h;
|
||||
// piCout << pheader.size() << sizeof(PFTHeader);
|
||||
pheader >> h;
|
||||
// piCout << h.session_id;
|
||||
PIMutexLocker ml(step_mutex);
|
||||
StepType st = (StepType)h.step;
|
||||
pftheader.step = st;
|
||||
if (!h.check_sig()) {
|
||||
cur_file_string = "Error: Check signature: File Transfer incompatable or invalid version!";
|
||||
piCoutObj << cur_file_string;
|
||||
stopReceive();
|
||||
return;
|
||||
}
|
||||
switch(st) {
|
||||
case pft_None:
|
||||
break;
|
||||
case pft_Description:
|
||||
pftheader.session_id = h.session_id;
|
||||
if (desc.size() < fi.start + fi.size) desc.resize(fi.start + fi.size);
|
||||
memcpy(desc.data(fi.start), ba.data(), ba.size_s());
|
||||
break;
|
||||
case pft_Data:
|
||||
if (h.session_id == pftheader.session_id)
|
||||
processFile(fi.id, fi.start, ba);
|
||||
else stopReceive();
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
PIString PIFileTransfer::curFile() const {
|
||||
PIString s = cur_file_string;
|
||||
if (scanning)
|
||||
return s + scan_dir.scanDir();
|
||||
return s;
|
||||
}
|
||||
|
||||
|
||||
PIByteArray PIFileTransfer::customHeader() {
|
||||
PIByteArray ba;
|
||||
ba << pftheader;
|
||||
return ba;
|
||||
}
|
||||
|
||||
|
||||
void PIFileTransfer::receive_started() {
|
||||
if (pftheader.step == pft_None) {
|
||||
files_.clear();
|
||||
// piCoutObj << "start receive"
|
||||
started_ = true;
|
||||
receiveFilesStarted();
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
void PIFileTransfer::receive_finished(bool ok) {
|
||||
step_mutex.lock();
|
||||
StepType st = (StepType)pftheader.step;
|
||||
step_mutex.unlock();
|
||||
if (st == pft_Description) {
|
||||
bool user_ok = true;
|
||||
if (ok) {
|
||||
// piCoutObj << desc.size() << PICoutManipulators::Hex << desc;
|
||||
desc >> files_;
|
||||
// piCoutObj << files_;
|
||||
PIStringList files;
|
||||
piForeachC(PFTFileInfo &fi, files_) files << fi.dest_path;
|
||||
pause();
|
||||
receiveFilesRequest(files, bytesAll(), &user_ok);
|
||||
}
|
||||
if (!ok || !user_ok) {
|
||||
pftheader.session_id--;
|
||||
piCoutObj << "receive aborted";
|
||||
receiveFilesFinished(false);
|
||||
started_ = false;
|
||||
} else resume();
|
||||
}
|
||||
if (st == pft_Data) {
|
||||
//piCoutObj << "file close";
|
||||
work_file.close();
|
||||
receiveFilesFinished(ok);
|
||||
started_ = false;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
void PIFileTransfer::send_finished(bool ok) {
|
||||
// piCoutObj << "file close";
|
||||
if (pftheader.step == pft_Data) {
|
||||
work_file.close();
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user