new file and progress

src/main.rs

@@ -1,7 +1,7 @@
-use std::{collections::HashMap, net::SocketAddr, sync::Arc, time::Duration};
-
-use russh::{keys::ssh_key::{rand_core::OsRng, PublicKey}, server::{Auth, Handler as SshHandler, Msg, Server, Session}, Channel, ChannelId, Error};
-use russh_sftp::{protocol::{File, FileAttributes, Handle, Name, Status, StatusCode}, server::Handler as SftpHandler};
+mod sftp;
+use std::{net::SocketAddr, sync::Arc, time::Duration};
+use russh::{keys::ssh_key::{rand_core::OsRng, PublicKey}, server::{Auth, Handler as SshHandler, Msg, Server, Session}, Channel, ChannelId};
+use sftp::SftpSession;
 
 
 struct SftpServer;
@@ -20,7 +20,7 @@ struct SshSession {
 }
 
 impl SshHandler for SshSession {
-    type Error = Error;
+    type Error = russh::Error;
 
     async fn auth_publickey_offered(
         &mut self,
@@ -67,8 +67,8 @@ impl SshHandler for SshSession {
     ) -> Result<(), Self::Error> {
         if name == "sftp" {
             session.channel_success(channel_id)?;
-            let root_dir = format!("/srv/sftp/{}", self.user.take().unwrap());
-            let sftp_handler = SftpSession { cwd: root_dir.clone(), root_dir, handle_map: HashMap::new() };
+            let jail_dir = format!("/srv/sftp/{}", self.user.take().unwrap());
+            let sftp_handler = SftpSession::new(jail_dir);
             russh_sftp::server::run(self.channel.take().ok_or(Self::Error::WrongChannel)?.into_stream(), sftp_handler).await;
         }
         else {
@@ -78,80 +78,6 @@ impl SshHandler for SshSession {
     }
 }
 
-struct SftpSession {
-    root_dir: String,
-    cwd: String,
-    handle_map: HashMap<String, bool>
-}
-
-impl SftpHandler for SftpSession {
-    type Error = StatusCode;
-
-    fn unimplemented(&self) -> Self::Error {
-        Self::Error::OpUnsupported
-    }
-
-    async fn realpath(
-        &mut self,
-        id: u32,
-        path: String,
-    ) -> Result<Name, Self::Error> {
-        let paths = path.split('/');
-        for path_part in paths {
-            match path_part {
-                ".." => {
-                    if self.cwd != self.root_dir {
-                        if let Some(pos) = self.cwd.rfind('/') {
-                            self.cwd.truncate(pos);
-                        }
-                    }
-                },
-                "." => {},
-                _ => self.cwd.push_str(&format!("/{}", path_part))
-            }
-        }
-
-        Ok(Name { id, files: vec![File::dummy(&self.cwd)] })
-    }
-
-    async fn opendir(
-        &mut self,
-        id: u32,
-        path: String,
-    ) -> Result<Handle, Self::Error> {
-        self.handle_map.insert(path.clone(), false);
-        Ok(Handle { id, handle: path })
-    }
-
-    async fn readdir(
-        &mut self,
-        id: u32,
-        handle: String,
-    ) -> Result<Name, Self::Error> {
-        if !self.handle_map.get(&handle).unwrap() {
-            *self.handle_map.get_mut(&handle).unwrap() = true;
-            return Ok(Name { id, files: vec![File::new("test", FileAttributes::default())] })
-        }
-        Err(StatusCode::Eof)
-    }
-
-    async fn close(
-        &mut self,
-        id: u32,
-        handle: String,
-    ) -> Result<Status, Self::Error> {
-        self.handle_map.remove(&handle);
-        Ok(Status {
-            id,
-            status_code: StatusCode::Ok,
-            error_message: "Ok".to_string(),
-            language_tag: "en-US".to_string(),
-        })
-    }
-
-}
-
-
 
 #[tokio::main]
 async fn main() {