/* Copyright (C) 2019 MariaDB Corporation 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; version 2 of the License. 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 "CopyTask.h" #include #include "SMLogging.h" #include "messageFormat.h" using namespace std; namespace storagemanager { CopyTask::CopyTask(int sock, uint len) : PosixTask(sock, len) { } CopyTask::~CopyTask() { } #define check_error(msg, ret) \ if (success < 0) \ { \ handleError(msg, errno); \ return ret; \ } bool CopyTask::run() { int success; SMLogging* logger = SMLogging::get(); uint8_t buf[2048] = {0}; if (getLength() > 2047) { handleError("CopyTask read", ENAMETOOLONG); return true; } success = read(buf, getLength()); check_error("CopyTask read", false); auto* cmd = (copy_cmd*)buf; string filename1(cmd->file1.filename, cmd->file1.flen); // need to copy this in case it's not null terminated auto* filename2 = (f_name*)&buf[sizeof(copy_cmd) + cmd->file1.flen]; #ifdef SM_TRACE logger->log(LOG_DEBUG, "copy %s to %s.", filename1.c_str(), filename2->filename); #endif int err; try { err = ioc->copyFile(filename1.c_str(), filename2->filename); } catch (exception& e) { logger->log(LOG_ERR, "CopyTask: caught %s", e.what()); err = -1; errno = EIO; } if (err) { handleError("CopyTask copy", errno); return true; } auto* resp = (sm_response*)buf; resp->returnCode = 0; return write(*resp, 0); } } // namespace storagemanager