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,58 @@ +/*** + 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 + +class FdReceiver : public QObject +{ + Q_OBJECT + +public: + FdReceiver(QObject *parent = nullptr); + + bool startListening(const QString &path); + void stopListening(); + int fileDescriptor() const; +private: + Q_SLOT void receiveFileDescriptor(); + + QSocketNotifier *m_readNotifier; + int m_socketDes; + int m_fileDes; +}; + +class FdSender +{ +public: + FdSender(); + + void connectToPath(const std::string &path); + 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,191 @@ +/*** + 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 "sharefd.h" + +//borrowed from klocalsocket.cpp +class KSockaddrUn +{ + sockaddr_un addr; + int addrlen; + +public: + KSockaddrUn(const std::string &path) + { + addrlen = sizeof(sockaddr_un); + memset(&addr, 0, addrlen); + addr.sun_family = AF_UNIX; + std::string finalPath = "/tmp/" + path; +#ifdef __linux__ + strcpy(&addr.sun_path[1], finalPath.c_str()); +#else + strcpy(&addr.sun_path, finalPath.c_str()); + unlink(finalPath.c_str()); +#endif + } + + int length() const + { + return addrlen; + } + const sockaddr *address() + { + return reinterpret_cast(&addr); + } +}; + +class KMsgHdr +{ + char io_buf[2]; + char cmsg_buf[CMSG_SPACE(sizeof(int))]; + msghdr msg; + iovec io; + +public: + KMsgHdr() + { + io.iov_base = io_buf; + io.iov_len = 2; + msg.msg_name = NULL; + msg.msg_namelen = 0; + msg.msg_iov = &io; + msg.msg_iovlen = 1; + msg.msg_control = cmsg_buf; + msg.msg_controllen = sizeof(cmsg_buf); + } + + msghdr *message() + { + return &msg; + } + + cmsghdr *cmsgHeader() const + { + return CMSG_FIRSTHDR(&msg); + } +}; + + +// File descriptor reciever +FdReceiver::FdReceiver(QObject *parent) + : QObject(parent) + , m_socketDes(-1) + , m_fileDes(-1) +{ +} + +bool FdReceiver::startListening(const QString &path) +{ + if (path.isEmpty()) + return false; + + m_socketDes = ::socket(AF_LOCAL, SOCK_STREAM | SOCK_NONBLOCK, 0); + if (m_socketDes == -1) + return false; + + KSockaddrUn addr(path.toStdString()); + bool bound = bind(m_socketDes, addr.address(), addr.length()) != -1; + bool listening = listen(m_socketDes, 5) != -1; + + if (!bound || !listening) { + ::close(m_socketDes); + return false; + } + + m_readNotifier = new QSocketNotifier(m_socketDes, QSocketNotifier::Read, this); + connect(m_readNotifier, &QSocketNotifier::activated, this, &FdReceiver::receiveFileDescriptor); + return true; +} + +void FdReceiver::receiveFileDescriptor() +{ + int client = ::accept(m_socketDes, NULL, NULL); + if (client > 0) { + KMsgHdr msg; + if (recvmsg(client, msg.message(), 0) == 2) { + cmsghdr *cmsg = msg.cmsgHeader(); + memcpy(&m_fileDes, (int*)CMSG_DATA(cmsg), sizeof(int)); + m_readNotifier->setEnabled(false); + } + ::close(client); + } + ::close(m_socketDes); +} + +void FdReceiver::stopListening() +{ + if (m_readNotifier->isEnabled()) { + m_readNotifier->setEnabled(false); + ::close(m_socketDes); + } +} + +int FdReceiver::fileDescriptor() const +{ + return m_fileDes; +} + + +// File descriptor sender +FdSender::FdSender() + : m_socketDes(-1) + , m_connected(false) +{ +} + +void FdSender::connectToPath(const std::string &path) +{ + if (path.empty()) + return; + + m_socketDes = ::socket(AF_UNIX, SOCK_STREAM | SOCK_NONBLOCK, 0); + if (m_socketDes == -1) + return; + + KSockaddrUn addr(path); + bool connected = ::connect(m_socketDes, addr.address(), addr.length()) == 0; + + if (!connected) + close(m_socketDes); + + m_connected = true; +} + +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((int*)CMSG_DATA(cmsg), &fd, sizeof(int)); + bool success = sendmsg(m_socketDes, msg.message(), 0) == 2; + close(m_socketDes); + return success; +} + +bool FdSender::isConnected() const +{ + return m_connected; +}