unix_aux.hxx

Go to the documentation of this file.

#pragma once
 
// C++
#include <vector>
 
// Cosmos
#include <cosmos/dso_export.h>
#include <cosmos/fs/types.hxx>
#include <cosmos/net/message_header.hxx>
#include <cosmos/net/types.hxx>
#include <cosmos/proc/process.hxx>
 
namespace cosmos {
 
 
struct COSMOS_API UnixCredentials :
                protected ::ucred {
 
        UnixCredentials() :
                UnixCredentials{ProcessID::INVALID, UserID::INVALID, GroupID::INVALID} {
        }
 
        UnixCredentials(const ProcessID p_pid, const UserID p_uid, const GroupID p_gid) {
                pid = cosmos::to_integral(p_pid);
                uid = cosmos::to_integral(p_uid);
                gid = cosmos::to_integral(p_gid);
        }
 
 
        void setCurrentCreds();
 
        auto processID() { return ProcessID{pid}; }
        auto userID() { return UserID{uid}; }
        auto groupID() { return GroupID{gid}; }
};
 
 
class COSMOS_API UnixRightsMessage :
                public AncillaryMessage<OptLevel::SOCKET, UnixMessage> {
public: // types
 
        using FileNumVector = std::vector<FileNum>;
 
public: // data
 
        constexpr static size_t MAX_FDS = 253; /* SCM_MAX_FD, only found in kernel headers */
 
public: // functions
 
        ~UnixRightsMessage() {
                closeUnclaimed();
        }
 
 
        void deserialize(const ReceiveMessageHeader::ControlMessage &msg);
 
 
        SendMessageHeader::ControlMessage serialize() const;
 
        void addFD(cosmos::FileNum fd) {
                m_fds.push_back(fd);
        }
 
        void clearFDs() {
                closeUnclaimed();
                m_fds.clear();
        }
 
        void takeFDs(FileNumVector &fds) {
                if (!m_unclaimed_fds) {
                        fds.clear();
                        return;
                }
                fds = std::move(m_fds);
                m_fds = FileNumVector{};
                m_unclaimed_fds = false;
        }
 
        size_t numFDs() const {
                return m_unclaimed_fds ? m_fds.size() : 0;
        }
 
protected: // functions
 
        void closeUnclaimed();
 
protected: // data
 
        FileNumVector m_fds;
        bool m_unclaimed_fds = false; 
};
 
 
class COSMOS_API UnixCredentialsMessage :
                public AncillaryMessage<OptLevel::SOCKET, UnixMessage> {
public: // functions
 
        void deserialize(const ReceiveMessageHeader::ControlMessage &msg);
 
        SendMessageHeader::ControlMessage serialize() const;
 
        void setCreds(const UnixCredentials &creds) {
                m_creds = creds;
        }
 
        const UnixCredentials& creds() const {
                return m_creds;
        }
 
protected: // data
 
        UnixCredentials m_creds;
};
 
} // end ns