-
Notifications
You must be signed in to change notification settings - Fork 53
/
add_address.rs
66 lines (54 loc) · 1.52 KB
/
add_address.rs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
// SPDX-License-Identifier: MIT
use futures::stream::TryStreamExt;
use std::env;
use ipnetwork::IpNetwork;
use rtnetlink::{new_connection, Error, Handle};
#[tokio::main]
async fn main() -> Result<(), ()> {
let args: Vec<String> = env::args().collect();
if args.len() != 3 {
usage();
return Ok(());
}
let link_name = &args[1];
let ip: IpNetwork = args[2].parse().unwrap_or_else(|_| {
eprintln!("invalid address");
std::process::exit(1);
});
let (connection, handle, _) = new_connection().unwrap();
tokio::spawn(connection);
if let Err(e) = add_address(link_name, ip, handle.clone()).await {
eprintln!("{e}");
}
Ok(())
}
async fn add_address(
link_name: &str,
ip: IpNetwork,
handle: Handle,
) -> Result<(), Error> {
let mut links = handle
.link()
.get()
.match_name(link_name.to_string())
.execute();
if let Some(link) = links.try_next().await? {
handle
.address()
.add(link.header.index, ip.ip(), ip.prefix())
.execute()
.await?
}
Ok(())
}
fn usage() {
eprintln!(
"usage:
cargo run --example add_address -- <link_name> <ip_address>
Note that you need to run this program as root. Instead of running cargo as root,
build the example normally:
cd rtnetlink ; cargo build --example add_address
Then find the binary in the target directory:
cd ../target/debug/example ; sudo ./add_address <link_name> <ip_address>"
);
}