sock.h

Go to the documentation of this file.

/* SPDX-License-Identifier: LGPL-3.0-or-later
   SPDX-FileCopyrightText: Copyright © 2021 Erez Geva <ErezGeva2@gmail.com> */
 
#ifndef __PTPMGMT_SOCK_H
#define __PTPMGMT_SOCK_H
 
#ifdef __cplusplus
#include "buf.h"
#ifdef __PTPMGMT_HAVE_SYS_UN_H
#include <sys/un.h>
#endif
#ifdef __PTPMGMT_HAVE_NETINET_IN_H
#include <netinet/in.h>
#endif
#ifdef __linux__
#include <linux/if_packet.h>
#endif
#include "cfg.h"
#include "ptp.h"
 
__PTPMGMT_NAMESPACE_BEGIN
 
class SockBase
{
  protected:
    int m_fd = -1;
    bool m_isInit = false;
    SockBase() = default;
    bool sendReply(ssize_t cnt, size_t len) const;
    virtual bool sendBase(const void *msg, size_t len) const = 0;
    virtual ssize_t rcvBase(void *buf, size_t bufSize, bool block) const = 0;
    virtual bool initBase() = 0;
    virtual void closeChild() {}
    void closeBase();
  public:
    virtual ~SockBase();
 
    void close();
    bool init();
    bool send(const void *msg, size_t len) const;
    bool send(const Buf &buf, size_t len) const;
    bool sendBuf(const Buf &buf, size_t len) const;
    ssize_t rcv(void *buf, size_t bufSize, bool block = false) const;
    ssize_t rcv(Buf &buf, bool block = false) const;
    ssize_t rcvBuf(Buf &buf, bool block = false) const;
    int getFd() const;
    int fileno() const;
    #ifdef __PTPMGMT_SWIG_THREAD_START
    __PTPMGMT_SWIG_THREAD_START;
    #endif
    bool poll(uint64_t timeout_ms = 0) const;
    bool tpoll(uint64_t &timeout_ms) const; /* poll with timeout update */
    #ifdef __PTPMGMT_SWIG_THREAD_END
    __PTPMGMT_SWIG_THREAD_END;
    #endif
};
 
class SockUnix : public SockBase
{
  private:
    static std::string m_homeDir;
    std::string m_me, m_peer, m_lastFrom;
    sockaddr_un m_peerAddr;
    bool setPeerInternal(const std::string &str, bool useAbstract);
    bool sendAny(const void *msg, size_t len, const sockaddr_un &addr) const;
    static void setUnixAddr(sockaddr_un &addr, const std::string &str);
 
    bool sendBase(const void *msg, size_t len) const override final;
    ssize_t rcvBase(void *buf, size_t bufSize, bool block) const override final;
    bool initBase() override final;
    void closeChild() override final;
 
  public:
    SockUnix();
    const std::string &getPeerAddress() const;
    const char *getPeerAddress_c() const;
    bool isPeerAddressAbstract() const;
    bool setPeerAddress(const std::string &string, bool useAbstract = false);
    bool setPeerAddress(const ConfigFile &cfg, const std::string &section = "");
    const std::string &getSelfAddress() const;
    const char *getSelfAddress_c() const;
    bool isSelfAddressAbstract() const;
    bool setSelfAddress(const std::string &string, bool useAbstract = false);
    bool setDefSelfAddress(const std::string &rootBase = "",
        const std::string &useDef = "");
    static const std::string &getHomeDir();
    static const char *getHomeDir_c();
    bool sendTo(const void *msg, size_t len, const std::string &addrStr,
        bool useAbstract = false) const;
    bool sendTo(const Buf &buf, size_t len, const std::string &addrStr,
        bool useAbstract = false) const;
    ssize_t rcvFrom(void *buf, size_t bufSize, std::string &from,
        bool block = false) const;
    ssize_t rcvFrom(Buf &buf, std::string &from, bool block = false) const;
    ssize_t rcvFrom(void *buf, size_t bufSize, bool block = false);
    ssize_t rcvFrom(Buf &buf, bool block = false);
    ssize_t rcvBufFrom(Buf &buf, bool block = false);
    const std::string &getLastFrom() const;
    const char *getLastFrom_c() const;
    bool isLastFromAbstract() const;
    static bool isAddressAbstract(const std::string &addr);
};
 
class SockBaseIf : public SockBase
{
  protected:
    std::string m_ifName; /* interface to use */
    Binary m_mac;
    int m_ifIndex;
    bool m_have_if = false;
    bool setInt(const IfInfo &ifObj);
    SockBaseIf() = default;
    virtual bool setAllBase(const ConfigFile &cfg, const std::string &section) = 0;
  public:
    bool setIfUsingName(const std::string &ifName);
    bool setIfUsingIndex(int ifIndex);
    bool setIf(const IfInfo &ifObj);
    bool setAll(const IfInfo &ifObj, const ConfigFile &cfg,
        const std::string &section = "");
    bool setAllInit(const IfInfo &ifObj, const ConfigFile &cfg,
        const std::string &section = "");
};
 
class SockIp : public SockBaseIf
{
  protected:
    int m_domain, m_udp_ttl = -1;
    /* First for bind then for send */
    sockaddr *m_addr;
    size_t m_addr_len;
    const char *m_mcast_str; /* string form */
    Binary m_mcast;
    SockIp(int domain, const char *mcast, sockaddr *addr, size_t len);
    virtual bool initIp() = 0;
    bool sendBase(const void *msg, size_t len) const override final;
    ssize_t rcvBase(void *buf, size_t bufSize, bool block) const override final;
    bool initBase() override final;
  public:
    bool setUdpTtl(uint8_t udp_ttl);
    bool setUdpTtl(const ConfigFile &cfg, const std::string &section = "");
};
 
class SockIp4 : public SockIp
{
  private:
    sockaddr_in m_addr4;
 
    bool initIp() override final;
    bool setAllBase(const ConfigFile &cfg,
        const std::string &section) override final;
 
  public:
    SockIp4();
};
 
class SockIp6 : public SockIp
{
  private:
    sockaddr_in6 m_addr6;
    int m_udp6_scope = -1;
 
    bool initIp() override final;
    bool setAllBase(const ConfigFile &cfg,
        const std::string &section) override final;
 
  public:
    SockIp6();
 
    bool setScope(uint8_t udp6_scope);
    bool setScope(const ConfigFile &cfg, const std::string &section = "");
};
 
class SockRaw : public SockBaseIf
{
  private:
    Binary m_ptp_dst_mac;
    int m_socket_priority = -1;
    sockaddr_ll m_addr = {0};
    ethhdr m_hdr;
 
    bool setAllBase(const ConfigFile &cfg,
        const std::string &section) override final;
    bool sendBase(const void *msg, size_t len) const override final;
    ssize_t rcvBase(void *buf, size_t bufSize, bool block) const override final;
    bool initBase() override final;
 
  public:
    SockRaw();
    bool setPtpDstMacStr(const std::string &string);
    bool setPtpDstMac(const Binary &ptp_dst_mac);
    bool setPtpDstMac(const void *ptp_dst_mac, size_t len);
    bool setPtpDstMac(const ConfigFile &cfg, const std::string &section = "");
    bool setSocketPriority(uint8_t socket_priority);
    bool setSocketPriority(const ConfigFile &cfg, const std::string &section = "");
};
 
__PTPMGMT_NAMESPACE_END
#else /* __cplusplus */
#include "c/sock.h"
#endif /* __cplusplus */
 
#endif /* __PTPMGMT_SOCK_H */