Index: src/ioslaves/file/CMakeLists.txt =================================================================== --- src/ioslaves/file/CMakeLists.txt +++ src/ioslaves/file/CMakeLists.txt @@ -8,7 +8,7 @@ if(WIN32) set(kio_file_PART_SRCS file.cpp file_win.cpp ) else() - set(kio_file_PART_SRCS file.cpp file_unix.cpp ) + set(kio_file_PART_SRCS file.cpp file_unix.cpp sharefd.cpp ) endif() find_package(ACL) @@ -23,6 +23,10 @@ add_library(kio_file MODULE ${kio_file_PART_SRCS}) target_link_libraries(kio_file KF5::KIOCore KF5::I18n) +if(UNIX) + target_link_libraries(kio_file Qt5::Network) +endif() + if (HAVE_VOLMGT AND CMAKE_SYSTEM_NAME MATCHES SunOS) target_link_libraries(kio_file -lvolmgt) endif () Index: src/ioslaves/file/sharefd.h =================================================================== --- /dev/null +++ src/ioslaves/file/sharefd.h @@ -0,0 +1,63 @@ +/*** + Copyright (C) 2017 by Chinmoy Ranjan Pradhan + + This library 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 2.1 of the License, or (at your option) version 3, or any + later version accepted by the membership of KDE e.V. (or its + successor approved by the membership of KDE e.V.), which shall + act as a proxy defined in Section 6 of version 3 of the license. + + This library 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 library. If not, see . +***/ + +#ifndef SHAREFD_H +#define SHAREFD_H + +#include +#include + +class QSocketNotifier; + +class FdReceiver : public QObject +{ + Q_OBJECT + +public: + FdReceiver(const QString &path, QObject *parent = nullptr); + ~FdReceiver(); + + bool isValid() const; + int fileDescriptor() const; + +private: + void stopListening(); + Q_SLOT void receiveFileDescriptor(); + + QSocketNotifier *m_readNotifier; + int m_socketDes; + int m_fileDes; +}; + +class FdSender +{ +public: + FdSender(const std::string &path); + ~FdSender(); + + bool sendFileDescriptor(int fd); + bool isConnected() const; + +private: + int m_socketDes; + bool m_connected; +}; + +#endif // SHAREFD_H Index: src/ioslaves/file/sharefd.cpp =================================================================== --- /dev/null +++ src/ioslaves/file/sharefd.cpp @@ -0,0 +1,202 @@ +/*** + Copyright (C) 2017 by Chinmoy Ranjan Pradhan + + This library 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 2.1 of the License, or (at your option) version 3, or any + later version accepted by the membership of KDE e.V. (or its + successor approved by the membership of KDE e.V.), which shall + act as a proxy defined in Section 6 of version 3 of the license. + + This library 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 library. If not, see . +***/ + +#include +#include +#include +#include +#include +#include + +#include + +#include "sharefd.h" + +//borrowed from klocalsocket.cpp +class KSockaddrUn +{ + const sockaddr_un addr; + +public: + KSockaddrUn(const std::string &path) + : addr(make_address(path)) + { + } + + int length() const + { + return sizeof addr; + } + const sockaddr *address() const + { + return reinterpret_cast(&addr); + } + +private: + static sockaddr_un make_address(const std::string& path) + { + sockaddr_un a{ AF_UNIX, {0}}; + std::string finalPath = "/tmp/" + path; +#ifdef __linux__ + ::strcpy(&a.sun_path[1], finalPath.c_str()); +#else + ::strcpy(&a.sun_path, finalPath.c_str()); + ::unlink(finalPath.c_str()); +#endif + return a; + } +}; + +class KMsgHdr +{ + char io_buf[2]; + char cmsg_buf[CMSG_SPACE(sizeof(int))]; + iovec io; + msghdr msg; + +public: + KMsgHdr() + : io_buf{0} + , cmsg_buf{0} + , io{io_buf, sizeof io_buf} + , msg{nullptr, 0, &io, 1, &cmsg_buf, sizeof cmsg_buf, 0} + { + } + + msghdr *message() + { + return &msg; + } + + cmsghdr *cmsgHeader() + { + return CMSG_FIRSTHDR(&msg); + } +}; + + +// File descriptor reciever +FdReceiver::FdReceiver(const QString &path, QObject *parent) + : QObject(parent) + , m_readNotifier(nullptr) + , m_socketDes(-1) + , m_fileDes(-1) +{ + m_socketDes = ::socket(AF_LOCAL, SOCK_STREAM|SOCK_NONBLOCK, 0); + if (m_socketDes == -1) { + std::cerr << "socket error:" << strerror(errno); + return; + } + + const KSockaddrUn addr(path.toStdString()); + if (bind(m_socketDes, addr.address(), addr.length()) < 0 || listen(m_socketDes, 5) < 0) { + std::cerr << "bind/listen error:" << strerror(errno); + ::close(m_socketDes); + m_socketDes = -1; + return; + } + + m_readNotifier = new QSocketNotifier(m_socketDes, QSocketNotifier::Read, this); + connect(m_readNotifier, &QSocketNotifier::activated, this, &FdReceiver::receiveFileDescriptor); +} + +FdReceiver::~FdReceiver() +{ + if (m_socketDes >= 0) { + ::close(m_socketDes); + } +} + +bool FdReceiver::isValid() const +{ + return m_socketDes >= 0; +} + +void FdReceiver::receiveFileDescriptor() +{ + int client = ::accept(m_socketDes, NULL, NULL); + if (client > 0) { + KMsgHdr msg; + if (::recvmsg(client, msg.message(), 0) == 2) { + ::memcpy(&m_fileDes, CMSG_DATA(msg.cmsgHeader()), sizeof m_fileDes); + stopListening(); + } + ::close(client); + } +} + +void FdReceiver::stopListening() +{ + if (m_readNotifier) { + delete m_readNotifier; + m_readNotifier = nullptr; + } +} + +int FdReceiver::fileDescriptor() const +{ + return m_fileDes; +} + + +// File descriptor sender +FdSender::FdSender(const std::string &path) + : m_socketDes(-1) + , m_connected(false) +{ + m_socketDes = ::socket(AF_UNIX, SOCK_STREAM | SOCK_NONBLOCK, 0); + if (m_socketDes == -1) { + std::cerr << "socket error:" << strerror(errno); + return; + } + + KSockaddrUn addr(path); + if (::connect(m_socketDes, addr.address(), addr.length()) != 0) { + std::cerr << "connection error:" << strerror(errno); + return; + } + + m_connected = true; +} + +FdSender::~FdSender() +{ + if (m_socketDes >= 0) { + ::close(m_socketDes); + } +} + +bool FdSender::sendFileDescriptor(int fd) +{ + KMsgHdr msg; + cmsghdr *cmsg = msg.cmsgHeader(); + cmsg->cmsg_len = CMSG_LEN(sizeof(int)); + cmsg->cmsg_type = SCM_RIGHTS; + cmsg->cmsg_level = SOL_SOCKET; + memcpy(CMSG_DATA(cmsg), &fd, sizeof fd); + bool success = sendmsg(m_socketDes, msg.message(), 0) == 2; + ::close(m_socketDes); + return success; +} + +bool FdSender::isConnected() const +{ + return m_connected; +}