summaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/main.rs121
1 files changed, 64 insertions, 57 deletions
diff --git a/src/main.rs b/src/main.rs
index 6894e50..22a49bc 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,41 +1,41 @@
-/*
- * Copyright (C) 2015 Sebastian Dröge <sebastian@centricular.com>
- *
- * This library is free software; you can redistribute it and/or
- * modify it under the terms of the GNU Library General Public
- * License as published by the Free Software Foundation; either
- * version 2 of the License, or (at your option) any later version.
- *
- * This library 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
- * Library General Public License for more details.
- *
- * You should have received a copy of the GNU Library General Public
- * License along with this library; if not, write to the
- * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
- * Boston, MA 02110-1301, USA.
- */
+// Copyright (C) 2015 Sebastian Dröge <sebastian@centricular.com>
+//
+// This library is free software; you can redistribute it and/or
+// modify it under the terms of the GNU Library General Public
+// License as published by the Free Software Foundation; either
+// version 2 of the License, or (at your option) any later version.
+//
+// This library 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
+// Library General Public License for more details.
+//
+// You should have received a copy of the GNU Library General Public
+// License along with this library; if not, write to the
+// Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
+// Boston, MA 02110-1301, USA.
+//
extern crate mio;
use mio::*;
use mio::udp::*;
-use std::net::{SocketAddr};
-use std::ops::{Range};
+use std::net::SocketAddr;
+use std::ops::Range;
-const SOCKET_EVENT : Token = Token(0);
+const SOCKET_EVENT: Token = Token(0);
const SOCKET_GENERAL: Token = Token(1);
struct PtpReflectorHandler {
- event_socket : UdpSocket,
- event_addr : SocketAddr,
- general_socket : UdpSocket,
- general_addr : SocketAddr,
+ event_socket: UdpSocket,
+ event_addr: SocketAddr,
+ general_socket: UdpSocket,
+ general_addr: SocketAddr,
}
-fn slice_map_range_in_place<T,F>(s: &mut [T], range: Range<usize>, mut f: F)
- where F: FnMut(&T) -> T {
+fn slice_map_range_in_place<T, F>(s: &mut [T], range: Range<usize>, mut f: F)
+ where F: FnMut(&T) -> T
+{
let n = s.len();
if range.end > n {
@@ -53,23 +53,31 @@ impl PtpReflectorHandler {
let event_bind_addr = "0.0.0.0:319".parse().unwrap();
let event_socket = try!(UdpSocket::bound(&event_bind_addr).map_err(|e| e.to_string()));
- try!(event_socket.join_multicast(&multicast_group).map_err(|e|e.to_string()));
- try!(event_loop.register(&event_socket, SOCKET_EVENT, EventSet::readable(), PollOpt::level()).map_err(|e| e.to_string()));
+ try!(event_socket.join_multicast(&multicast_group).map_err(|e| e.to_string()));
+ try!(event_loop.register(&event_socket,
+ SOCKET_EVENT,
+ EventSet::readable(),
+ PollOpt::level())
+ .map_err(|e| e.to_string()));
// FIXME: How can we create this from the multicast group?
- //let event_addr = SocketAddr::new(multicast_group, 319);
+ // let event_addr = SocketAddr::new(multicast_group, 319);
let event_addr = "224.0.1.129:319".parse().unwrap();
let general_bind_addr = "0.0.0.0:320".parse().unwrap();
let general_socket = try!(UdpSocket::bound(&general_bind_addr).map_err(|e| e.to_string()));
- try!(general_socket.join_multicast(&multicast_group).map_err(|e|e.to_string()));
- try!(event_loop.register(&general_socket, SOCKET_GENERAL, EventSet::readable(), PollOpt::level()).map_err(|e| e.to_string()));
+ try!(general_socket.join_multicast(&multicast_group).map_err(|e| e.to_string()));
+ try!(event_loop.register(&general_socket,
+ SOCKET_GENERAL,
+ EventSet::readable(),
+ PollOpt::level())
+ .map_err(|e| e.to_string()));
// FIXME: How can we create this from the multicast group?
- //let general_addr = SocketAddr::new(multicast_group, 320);
+ // let general_addr = SocketAddr::new(multicast_group, 320);
let general_addr = "224.0.1.129:320".parse().unwrap();
- Ok (PtpReflectorHandler {
+ Ok(PtpReflectorHandler {
event_socket: event_socket,
event_addr: event_addr,
general_socket: general_socket,
@@ -88,20 +96,20 @@ impl Handler for PtpReflectorHandler {
}
let (socket, addr) = match token {
- SOCKET_EVENT => (&self.event_socket, &self.event_addr),
+ SOCKET_EVENT => (&self.event_socket, &self.event_addr),
SOCKET_GENERAL => (&self.general_socket, &self.general_addr),
- _ => panic!("unexpected token")
+ _ => panic!("unexpected token"),
};
let msg = &mut [0; 1024];
let length = match socket.recv_from(msg) {
- Err(e) => {
+ Err(e) => {
print!("Error while receiving message: {}\n", e.to_string());
return;
- },
- Ok(Some((l,_))) => l,
- Ok(None) => {
+ }
+ Ok(Some((l, _))) => l,
+ Ok(None) => {
print!("Error while receiving message\n");
return;
}
@@ -122,25 +130,25 @@ impl Handler for PtpReflectorHandler {
slice_map_range_in_place(msg, 20..28, |&x| x ^ 0xff);
match msg_type {
- 0x0 | 0x08 | 0xb => true,
+ 0x0 | 0x08 | 0xb => true,
0x9 if msg.len() >= 54 => {
slice_map_range_in_place(msg, 44..52, |&x| x ^ 0xff);
true
- },
- _ => false
+ }
+ _ => false,
}
- },
+ }
1 => {
msg[4] = 0;
slice_map_range_in_place(msg, 20..28, |&x| x ^ 0xff);
match msg_type {
0x1 => true,
- _ => false
+ _ => false,
}
- },
- _ => false
+ }
+ _ => false,
};
if forward {
@@ -148,8 +156,8 @@ impl Handler for PtpReflectorHandler {
Err(e) => {
print!("Error while sending message: {}\n", e.to_string());
return;
- },
- _ => ()
+ }
+ _ => (),
};
print!("Sent message of size {}\n", length);
@@ -159,16 +167,15 @@ impl Handler for PtpReflectorHandler {
fn main() {
let mut event_loop = EventLoop::new().unwrap();
- let res =
- PtpReflectorHandler::new(&mut event_loop)
- .map_err(|e| e.to_string())
- .and_then(|mut handler| {
- event_loop.run(&mut handler)
- .map_err(|e| e.to_string())
+ let res = PtpReflectorHandler::new(&mut event_loop)
+ .map_err(|e| e.to_string())
+ .and_then(|mut handler| {
+ event_loop.run(&mut handler)
+ .map_err(|e| e.to_string())
});
match res {
- Ok(_) => (),
- Err(s) => panic!(s)
+ Ok(_) => (),
+ Err(s) => panic!(s),
};
}
generated by cgit v0.11.2-2-gd1dd at 2024-11-15 11:31:35 (GMT)