Add netlink socket test

Author: gllghr

Cargo.lock

@@ -23,6 +23,9 @@ path = "src/bin/ptree.rs"
 [[bin]]
 name = "epoll_example"
 path = "src/bin/testing/epoll.rs"
+[[bin]]
+name = "netlink_example"
+path = "src/bin/testing/netlink.rs"
 
 [profile.release]
 lto = true
@@ -34,6 +37,7 @@ panic = "abort" # Save size. We don't need unwinding anyway.
 [dependencies]
 getopts = "0.2.15"
 nix = "0.12.0"
+libc = "0.2.48"
 
 [package.metadata.deb]
 assets = [

Cargo.toml

@@ -0,0 +1,47 @@
+//
+//   Copyright 2019 Delphix
+//
+//   Licensed under the Apache License, Version 2.0 (the "License");
+//   you may not use this file except in compliance with the License.
+//   You may obtain a copy of the License at
+//
+//       http://www.apache.org/licenses/LICENSE-2.0
+//
+//   Unless required by applicable law or agreed to in writing, software
+//   distributed under the License is distributed on an "AS IS" BASIS,
+//   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+//   See the License for the specific language governing permissions and
+//   limitations under the License.
+//
+
+use nix::sys::socket::{socket, AddressFamily, SockAddr, SockType, SockFlag, bind};
+use nix::errno::Errno;
+
+use std::fs::File;
+use std::os::raw::c_int;
+
+extern crate nix;
+
+const NETLINK_ROUTE: c_int = 0;
+
+fn main() {
+    // Use libc crate directly instead of nix because nix's 'socket' method doesn't have ability to
+    // specify netlink socket protocol.
+    let fd = unsafe {
+        libc::socket(AddressFamily::Netlink as c_int,
+        SockType::Datagram as c_int,
+        NETLINK_ROUTE)
+    };
+
+    let fd = Errno::result(fd).unwrap();
+
+    bind(fd, &SockAddr::new_netlink(0, 0));
+
+    // Signal parent process (the test process) that this process is ready to be observed by the
+    // ptool being tested.
+    File::create("/tmp/ptools-test-ready").unwrap();
+
+    // Wait for the parent finish running the ptool and then kill us.
+    loop {}
+}
+

src/bin/testing/netlink.rs

@@ -0,0 +1,46 @@
+//
+//   Copyright 2019 Delphix
+//
+//   Licensed under the Apache License, Version 2.0 (the "License");
+//   you may not use this file except in compliance with the License.
+//   You may obtain a copy of the License at
+//
+//       http://www.apache.org/licenses/LICENSE-2.0
+//
+//   Unless required by applicable law or agreed to in writing, software
+//   distributed under the License is distributed on an "AS IS" BASIS,
+//   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+//   See the License for the specific language governing permissions and
+//   limitations under the License.
+//
+
+mod common;
+
+#[test]
+fn netlink_basic() {
+    let stdout = common::run_ptool("pfiles", "netlink_example");
+    let lines = stdout.lines().collect::<Vec<&str>>();
+
+    //
+    // We expect something along the lines of
+    // ...
+    //     3: S_IFSOCK mode:777 dev:0,9 ino:725134 uid:65433 gid:50 size:0
+    //        O_RDWR
+    //          SOCK_DGRAM
+    //          sockname: AF_NETLINK
+    //
+    let pattern = "3: S_IFSOCK";
+    let split_lines = lines
+        .split(|l| l.trim().starts_with(pattern))
+        .collect::<Vec<_>>();
+
+    if split_lines.len() != 2 {
+        panic!("String '{}' not found in command output:\n\n{}\n\n", pattern, stdout);
+    }
+    let fd_info = split_lines[1];
+
+    let pattern = "sockname: AF_NETLINK";
+    if fd_info[2].trim() != pattern {
+        panic!("String '{}' not found in command output:\n\n{}\n\n", pattern, fd_info.join("\n"));
+    }
+}