ManagedRoute uses ioctl to add/remove routes on Linux

Added LinuxNetLink to talk to the rtnetlink socket for adding interfaces, addresses routes. Not yet complete. Can currently monitor changes on the system.
2018-05-24 15:22:11 -07:00 · 2018-05-24 15:22:11 -07:00 · c24d16e62e
commit c24d16e62e
parent c0efba79c7
6 changed files with 783 additions and 19 deletions
--- a/osdep/LinuxNetLink.hpp
+++ b/osdep/LinuxNetLink.hpp
@ -0,0 +1,133 @@
+/*
+ * ZeroTier One - Network Virtualization Everywhere
+ * Copyright (C) 2011-2018  ZeroTier, Inc.  https://www.zerotier.com/
+ *
+ * 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, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * 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, see <http://www.gnu.org/licenses/>.
+ *
+ * --
+ *
+ * You can be released from the requirements of the license by purchasing
+ * a commercial license. Buying such a license is mandatory as soon as you
+ * develop commercial closed-source software that incorporates or links
+ * directly against ZeroTier software without disclosing the source code
+ * of your own application.
+ */
+
+#ifndef ZT_LINUX_NETLINK_HPP
+#define ZT_LINUX_NETLINK_HPP
+
+#include <vector>
+
+#include <bits/sockaddr.h>
+#include <asm/types.h>
+#include <linux/rtnetlink.h>
+#include <sys/socket.h>
+
+
+#include "../node/InetAddress.hpp"
+#include "Thread.hpp"
+
+
+namespace ZeroTier {
+
+struct route_entry {
+    InetAddress target;
+    InetAddress via;
+    const char *iface;
+};
+
+typedef std::vector<route_entry> RouteList;
+
+struct nl_req {
+    struct nlmsghdr nl;
+    struct rtmsg rt;
+    char buf[8192];
+};
+
+/**
+ * Interface with Linux's RTNETLINK 
+ */
+class LinuxNetLink
+{
+private:
+    LinuxNetLink();
+    ~LinuxNetLink();
+
+public:
+    static LinuxNetLink& getInstance()
+    {
+        static LinuxNetLink instance;
+        return instance;
+    }
+
+    LinuxNetLink(LinuxNetLink const&) = delete;
+    void operator=(LinuxNetLink const&) = delete;
+
+    void addRoute(const InetAddress &target, const InetAddress &via, const char *ifaceName);
+    void delRoute(const InetAddress &target, const InetAddress &via, const char *ifaceName);
+    RouteList getIPV4Routes() const;
+    RouteList getIPV6Routes() const;
+
+    void addInterface(const char *iface);
+
+    void addAddress(const InetAddress &addr, const char *iface);
+
+    void threadMain() throw();
+private:
+    void _processMessage();
+    void _routeAdded();
+    void _routeDeleted();
+    void _linkAdded();
+    void _linkDeleted();
+    void _ipAddressAdded();
+    void _ipAddressDeleted();
+
+
+    void _requestIPv4Routes();
+    void _requestIPv6Routes();
+
+
+    Thread _t;
+    bool _running;
+    RouteList _routes_ipv4;
+    RouteList _routes_ipv6;
+
+    // socket communication vars;
+    int _fd;
+    struct sockaddr_nl _la;
+    struct sockaddr_nl _pa;
+    struct msghdr _msg;
+    struct iovec _iov;
+    int _rtn;
+    char _buf[8192];
+
+    // RTNETLINK message pointers & lengths
+    // used for processing messages
+    struct nlmsghdr *_nlp;
+    int _nll;
+
+    struct rtmsg *_rtp;
+    int _rtl;
+    struct rtattr *_rtap;
+    
+    struct ifinfomsg *_ifip;
+    int _ifil;
+
+    struct ifaddrmsg *_ifap;
+    int _ifal;
+};    
+
+}
+
+#endif // ZT_LINUX_NETLINK_HPPS