YGGverse/nexy
1
0
Fork 0
mirror of https://github.com/YGGverse/nexy.git synced 2026-03-31 17:25:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

stop storage reader on chunk sending error

Browse source
This commit is contained in:
yggverse 2025-06-29 12:44:44 +03:00
parent 2003cefb67
commit 9a40d2a7a5
2 changed files with 75 additions and 47 deletions
Download patch file Download diff file Expand all files Collapse all files

92
src/server/connection.rs
View file

@ -37,36 +37,80 @@ impl Connection {
Ok(q) => {
if self.session.is_debug {
println!(
"[{}] < [{}] request `{q}`...",
"[{}] < [{}] incoming request: `{q}`",
self.address.server, self.address.client
)
}
if let Some(ref i) = self.session.request {
i.add(&self.address.client, &q)
if let Some(ref r) = self.session.request {
r.add(&self.address.client, &q)
}
self.session
if self
.session
.clone()
.public
.request(&q, |r| t += self.response(r)); // chunk loop
.request(&q, |r| match self.response(r) {
Ok(sent) => {
t += sent;
if self.session.is_debug {
println!(
"[{}] > [{}] sent {sent} ({t} total) bytes response.",
self.address.server, self.address.client
)
};
true
}
Err(e) => {
eprintln!(
"[{}] > [{}] `{q}`: error sending response: `{e}`",
self.address.server, self.address.client
);
false
}
})
{
self.session
.access_log
.clf(&self.address.client, Some(&q), 0, t);
self.shutdown()
} else {
self.session
.access_log
.clf(&self.address.client, Some(&q), 1, t);
}
Err(e) => {
t += self.response(Response::InternalServerError(
}
Err(e) => match self.response(Response::InternalServerError(
"",
format!(
"[{}] < [{}] failed to handle incoming request: `{e}`",
self.address.server, self.address.client
),
));
)) {
Ok(sent) => {
t += sent;
if self.session.is_debug {
println!(
"[{}] > [{}] sent {sent} ({t} total) bytes response.",
self.address.server, self.address.client
)
};
self.session
.access_log
.clf(&self.address.client, None, 2, t);
self.shutdown()
}
Err(e) => {
eprintln!(
"[{}] > [{}] handle request error: `{e}`",
self.address.server, self.address.client
);
self.session
.access_log
.clf(&self.address.client, None, 1, t);
}
}
self.shutdown()
}
},
}
}
fn request(&mut self) -> Result<String> {
let mut b = [0; 1024]; // @TODO unspecified len?
@ -74,7 +118,7 @@ impl Connection {
Ok(urlencoding::decode(std::str::from_utf8(&b[..n])?.trim())?.to_string())
}
fn response(&mut self, response: Response) -> usize {
fn response(&mut self, response: Response) -> Result<usize> {
let bytes = match response {
Response::File(b) => b,
Response::Directory(q, ref s, is_root) => {
@ -114,29 +158,9 @@ impl Connection {
self.session.template.not_found()
}
};
match self.stream.write_all(bytes) {
Ok(()) => {
if self.session.is_debug {
println!(
"[{}] > [{}] sent {} bytes response.",
self.address.server,
self.address.client,
bytes.len()
)
}
if let Err(e) = self.stream.flush() {
eprintln!(
"[{}] ! [{}] failed to flush the stream: `{e}`",
self.address.server, self.address.client,
)
}
}
Err(e) => eprintln!(
"[{}] ! [{}] failed to response: `{e}`",
self.address.server, self.address.client,
),
};
bytes.len()
self.stream.write_all(bytes)?;
self.stream.flush()?;
Ok(bytes.len())
}
fn shutdown(self) {

10
src/session/public.rs
View file

@ -42,7 +42,7 @@ impl Public {
})
}
pub fn request(&self, query: &str, mut callback: impl FnMut(Response)) {
pub fn request(&self, query: &str, mut callback: impl FnMut(Response) -> bool) -> bool {
let p = {
// access restriction zone, change carefully!
let mut p = PathBuf::from(&self.public_dir);
@ -67,8 +67,12 @@ impl Public {
Ok(mut f) => loop {
let mut b = vec![0; self.read_chunk];
match f.read(&mut b) {
Ok(0) => break,
Ok(n) => callback(Response::File(&b[..n])),
Ok(0) => return true,
Ok(n) => {
if !callback(Response::File(&b[..n])) {
return false; // break reader on callback failure
}
}
Err(e) => {
return callback(Response::InternalServerError(
query,