124 lines
3.2 KiB
Rust
124 lines
3.2 KiB
Rust
// Copyright 2015-2017 Parity Technologies (UK) Ltd.
|
|
// This file is part of Parity.
|
|
|
|
// Parity 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.
|
|
|
|
// Parity 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 Parity. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
extern crate hyper;
|
|
extern crate multihash;
|
|
extern crate cid;
|
|
|
|
extern crate rlp;
|
|
extern crate ethcore;
|
|
extern crate ethcore_util as util;
|
|
|
|
mod error;
|
|
mod handler;
|
|
|
|
use std::sync::Arc;
|
|
use error::ServerError;
|
|
use handler::{IpfsHandler, Out};
|
|
use hyper::server::{Listening, Handler, Request, Response};
|
|
use hyper::net::HttpStream;
|
|
use hyper::header::{ContentLength, ContentType};
|
|
use hyper::{Next, Encoder, Decoder, Method, RequestUri, StatusCode};
|
|
use ethcore::client::BlockChainClient;
|
|
|
|
impl Handler<HttpStream> for IpfsHandler {
|
|
fn on_request(&mut self, req: Request<HttpStream>) -> Next {
|
|
if *req.method() != Method::Get {
|
|
return Next::write();
|
|
}
|
|
|
|
let (path, query) = match *req.uri() {
|
|
RequestUri::AbsolutePath { ref path, ref query } => (path, query.as_ref().map(AsRef::as_ref)),
|
|
_ => return Next::write(),
|
|
};
|
|
|
|
self.route(path, query)
|
|
}
|
|
|
|
fn on_request_readable(&mut self, _decoder: &mut Decoder<HttpStream>) -> Next {
|
|
Next::write()
|
|
}
|
|
|
|
fn on_response(&mut self, res: &mut Response) -> Next {
|
|
use Out::*;
|
|
|
|
match *self.out() {
|
|
OctetStream(ref bytes) => {
|
|
let headers = res.headers_mut();
|
|
|
|
headers.set(ContentLength(bytes.len() as u64));
|
|
headers.set(ContentType("application/octet-stream".parse()
|
|
.expect("Static content type; qed")));
|
|
|
|
Next::write()
|
|
},
|
|
NotFound(reason) => {
|
|
res.set_status(StatusCode::NotFound);
|
|
|
|
res.headers_mut().set(ContentLength(reason.len() as u64));
|
|
res.headers_mut().set(ContentType("text/plain".parse()
|
|
.expect("Static content type; qed")));
|
|
|
|
Next::write()
|
|
},
|
|
Bad(reason) => {
|
|
res.set_status(StatusCode::BadRequest);
|
|
|
|
res.headers_mut().set(ContentLength(reason.len() as u64));
|
|
res.headers_mut().set(ContentType("text/plain".parse()
|
|
.expect("Static content type; qed")));
|
|
|
|
Next::write()
|
|
}
|
|
}
|
|
}
|
|
|
|
fn on_response_writable(&mut self, transport: &mut Encoder<HttpStream>) -> Next {
|
|
use Out::*;
|
|
|
|
match *self.out() {
|
|
OctetStream(ref bytes) => {
|
|
// Nothing to do here
|
|
let _ = transport.write(&bytes);
|
|
|
|
Next::end()
|
|
},
|
|
NotFound(reason) | Bad(reason) => {
|
|
// Nothing to do here
|
|
let _ = transport.write(reason.as_bytes());
|
|
|
|
Next::end()
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
pub fn start_server(client: Arc<BlockChainClient>) -> Result<Listening, ServerError> {
|
|
let addr = "0.0.0.0:5001".parse().expect("can't fail on static input; qed");
|
|
|
|
hyper::Server::http(&addr)?
|
|
.handle(move |_| IpfsHandler::new(client.clone()))
|
|
.map(|(listening, srv)| {
|
|
|
|
::std::thread::spawn(move || {
|
|
srv.run();
|
|
});
|
|
|
|
listening
|
|
})
|
|
.map_err(Into::into)
|
|
}
|