Skip to content

Commit 0bf5436

Browse files
committed
Fix some minor points according to the comments.
1 parent ac73400 commit 0bf5436

File tree

2 files changed

+34
-41
lines changed

2 files changed

+34
-41
lines changed

build.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,13 +25,13 @@ fn create_util_map() -> HashMap<&'static str, &'static str> {
2525
hashmap.insert("tar", "lsb");
2626

2727
hashmap.insert("ping", "networking");
28+
hashmap.insert("nc", "networking");
2829

2930
hashmap.insert("cat", "posix");
3031
hashmap.insert("chmod", "posix");
3132
hashmap.insert("head", "posix");
3233
hashmap.insert("sh", "posix");
3334
hashmap.insert("sleep", "posix");
34-
hashmap.insert("nc", "networking");
3535

3636
hashmap.insert("init", "sysinit");
3737

src/networking/nc/mod.rs

Lines changed: 33 additions & 40 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
11
extern crate mio;
22
extern crate clap;
3-
extern crate libc;
43
extern crate socket2;
54

65
use std;
@@ -10,7 +9,7 @@ use clap::{Arg, ArgMatches};
109
use mio::{Events, Event, Poll, Ready, PollOpt, Token};
1110
use libc::{AF_UNSPEC, AF_INET, AF_INET6, AF_UNIX};
1211
use std::io;
13-
use std::net::{SocketAddr};
12+
use std::net::SocketAddr;
1413
use mio::unix::EventedFd;
1514
use std::io::{Read,Write, ErrorKind};
1615
use mio::unix::UnixReady;
@@ -81,10 +80,10 @@ fn debug_info(msg: &str) {
8180

8281
impl NcOptions {
8382
pub fn parse(matches: ArgMatches, msg: &str) -> Result<NcOptions, MesaError> {
84-
let mut portlist = vec!();
83+
let mut portlist = vec![];
8584
let lflag = matches.is_present("l");
8685
let mut host = String::from("127.0.0.1");
87-
let uport:String;
86+
let uport: String;
8887
let mut interval = None;
8988
let mut timeout = None;
9089
let s_addr = match matches.value_of("s") {
@@ -98,11 +97,7 @@ impl NcOptions {
9897
let kflag = matches.is_present("k");
9998

10099
/* Cruft to make sure options are clean, and used properly. */
101-
let positionals: Vec<&str> = if matches.is_present("positionals") {
102-
matches.values_of("positionals").unwrap().collect()
103-
} else {
104-
vec!()
105-
};
100+
let positionals:Vec<&str> = matches.values_of("positionals").unwrap().collect();
106101

107102
let family = if matches.is_present("U") {
108103
AF_UNIX
@@ -127,8 +122,6 @@ impl NcOptions {
127122
} else if positionals.len() >= 2 {
128123
host = String::from(positionals[0]);
129124
uport = String::from(positionals[1]);
130-
} else {
131-
return mesaerr_result(msg);
132125
}
133126

134127
if lflag && s_addr.is_some() {
@@ -345,7 +338,7 @@ impl <'a> NcCore<'a> {
345338
} else if self.stdinbuf_full() {
346339
self.disable_stdin()?;
347340
}
348-
}
341+
}
349342

350343
// if net writable and buf not empty, try to write to net
351344
// error, stop watching for netout
@@ -444,81 +437,81 @@ impl <'a> NcCore<'a> {
444437
self.netinbufpos >= BUFSIZE
445438
}
446439

447-
fn remove_stdin(&mut self) -> std::io::Result<()> {
440+
fn remove_stdin(&mut self) -> io::Result<()> {
448441
remove_item(&mut self.open_ends, NcCore::STDIN_POLL);
449442
self.poll.deregister(&self.event_stdin)
450443
}
451444

452-
fn remove_stdout(&mut self) -> std::io::Result<()> {
445+
fn remove_stdout(&mut self) -> io::Result<()> {
453446
debug_info("remove_stdout");
454447
remove_item(&mut self.open_ends, NcCore::STDOUT_POLL);
455448
self.poll.deregister(&self.event_stdout)
456449
}
457450

458-
fn remove_netin(&mut self) -> std::io::Result<()> {
451+
fn remove_netin(&mut self) -> io::Result<()> {
459452
remove_item(&mut self.open_ends, NcCore::NETIN_POLL);
460453
self.net_interest.remove(Ready::readable());
461454
self.reregister_net()
462455
}
463456

464-
fn remove_netout(&mut self) -> std::io::Result<()> {
457+
fn remove_netout(&mut self) -> io::Result<()> {
465458
remove_item(&mut self.open_ends, NcCore::NETOUT_POLL);
466459
self.net_interest.remove(Ready::writable());
467460
return self.reregister_net();
468461
}
469462

470-
fn reregister_net(&mut self) -> std::io::Result<()> {
463+
fn reregister_net(&mut self) -> io::Result<()> {
471464
self.poll.reregister(&self.event_net, NcCore::TK_NET, self.net_interest,
472465
PollOpt::empty())
473466
}
474467

475-
fn enable_netin(&mut self) -> std::io::Result<()>{
468+
fn enable_netin(&mut self) -> io::Result<()>{
476469
self.net_interest |= Ready::readable();
477470
self.reregister_net()
478471
}
479472

480-
fn disable_netin(&mut self) -> std::io::Result<()>{
473+
fn disable_netin(&mut self) -> io::Result<()>{
481474
self.net_interest.remove(Ready::readable());
482-
self.reregister_net()
475+
self.reregister_net()
483476
}
484477

485-
fn enable_netout(&mut self) -> std::io::Result<()> {
478+
fn enable_netout(&mut self) -> io::Result<()> {
486479
self.net_interest |= Ready::writable();
487-
self.reregister_net()
480+
self.reregister_net()
488481
}
489482

490-
fn disable_netout(&mut self) -> std::io::Result<()> {
483+
fn disable_netout(&mut self) -> io::Result<()> {
491484
self.net_interest.remove(Ready::writable());
492-
self.reregister_net()
485+
self.reregister_net()
493486
}
494487

495-
fn enable_stdin(&mut self) -> std::io::Result<()> {
488+
fn enable_stdin(&mut self) -> io::Result<()> {
496489
self.poll.reregister(&self.event_stdin, NcCore::TK_STDIN, Ready::readable(),
497490
PollOpt::empty())
498491
}
499492

500-
fn disable_stdin(&mut self) -> std::io::Result<()> {
493+
fn disable_stdin(&mut self) -> io::Result<()> {
501494
self.poll.reregister(&self.event_stdin, NcCore::TK_STDIN, Ready::empty(),
502495
PollOpt::empty())
503496
}
504497

505-
fn enable_stdout(&mut self) -> std::io::Result<()> {
498+
fn enable_stdout(&mut self) -> io::Result<()> {
506499
self.poll.reregister(&self.event_stdout, NcCore::TK_STDOUT, Ready::writable(),
507500
PollOpt::empty())
508501
}
509502

510-
fn disable_stdout(&mut self) -> std::io::Result<()> {
503+
fn disable_stdout(&mut self) -> io::Result<()> {
511504
self.poll.reregister(&self.event_stdout, NcCore::TK_STDOUT, Ready::empty(),
512505
PollOpt::empty())
513506
}
514507

515-
fn remove_net(&mut self) -> std::io::Result<()> {
508+
fn remove_net(&mut self) -> io::Result<()> {
516509
remove_item(&mut self.open_ends, NcCore::NETIN_POLL);
517510
remove_item(&mut self.open_ends, NcCore::NETOUT_POLL);
518511
self.poll.deregister(&self.event_net)
519512
}
520513

521-
fn read_stdin(&mut self) -> std::io::Result<()> {
514+
fn read_stdin(&mut self) -> io::Result<()> {
522515
let mut remove = false;
523516
match io::stdin().read(&mut self.stdinbuf[self.stdinbufpos..]) {
524517
Ok(len) => {
@@ -538,7 +531,7 @@ impl <'a> NcCore<'a> {
538531
Ok(())
539532
}
540533

541-
fn write_netout(&mut self) -> std::io::Result<()> {
534+
fn write_netout(&mut self) -> io::Result<()> {
542535
let mut remove = false;
543536
match self.sock.write(&mut self.stdinbuf[0..self.stdinbufpos]) {
544537
Ok(len) => {
@@ -567,7 +560,7 @@ impl <'a> NcCore<'a> {
567560
Ok(())
568561
}
569562

570-
fn read_netin(&mut self) -> std::io::Result<()> {
563+
fn read_netin(&mut self) -> io::Result<()> {
571564
let mut remove = false;
572565
match self.sock.read(&mut self.netinbuf[self.netinbufpos..]) {
573566
Ok(len) => {
@@ -587,7 +580,7 @@ impl <'a> NcCore<'a> {
587580
Ok(())
588581
}
589582

590-
fn write_stdout(&mut self) -> std::io::Result<()> {
583+
fn write_stdout(&mut self) -> io::Result<()> {
591584
let mut remove = false;
592585
match io::stdout().write(&mut self.netinbuf[0..self.netinbufpos]) {
593586
Ok(len) => {
@@ -607,14 +600,14 @@ impl <'a> NcCore<'a> {
607600
_ => remove = true
608601
}
609602
},
610-
}
603+
}
611604
if remove {
612605
return self.remove_stdout();
613606
}
614-
Ok(())
607+
Ok(())
615608
}
616609

617-
fn handle_error_event(&mut self, event: &Event) -> std::io::Result<()> {
610+
fn handle_error_event(&mut self, event: &Event) -> io::Result<()> {
618611
match event.token() {
619612
NcCore::TK_STDIN => self.remove_stdin(),
620613
NcCore::TK_STDOUT => self.remove_stdout(),
@@ -623,7 +616,7 @@ impl <'a> NcCore<'a> {
623616
}
624617
}
625618

626-
fn handle_hup_event(&mut self, event: &Event) -> std::io::Result<()> {
619+
fn handle_hup_event(&mut self, event: &Event) -> io::Result<()> {
627620
if !self.stdin_gone() && event.token() == NcCore::TK_STDIN &&
628621
!event.readiness().is_readable() {
629622
self.remove_stdin()?
@@ -667,7 +660,7 @@ impl <'a> NcCore<'a> {
667660
debug_info(&format!("new_ready_end {:?}", new_ready_end));
668661
}
669662
}
670-
*last_ready_end = new_ready_end;
663+
*last_ready_end = new_ready_end;
671664
}
672665
}
673666

@@ -910,9 +903,9 @@ where
910903
.takes_value(true))
911904
.arg(Arg::with_name("s")
912905
.short("s")
913-
.value_name("source_ip_address")
906+
.value_name("source_ip_address")
914907
.takes_value(true)
915-
.help("Specifies the IP of the interface which is used to send the packets. It is an error to use this option in conjunction with the -l option."))
908+
.help("Specifies the IP of the interface which is used to send the packets. It is an error to use this option in conjunction with the -l option."))
916909
.arg(Arg::with_name("d")
917910
.short("d")
918911
.help("Do not attempt to read from stdin."))

0 commit comments

Comments
 (0)