3 files changed, 85 insertions, 7 deletions

diff --git a/src/imap/command/authenticated.rs b/src/imap/command/authenticated.rs
index 6c2e43b..5cf8cf9 100644
--- a/src/imap/command/authenticated.rs
+++ b/src/imap/command/authenticated.rs
@@ -121,6 +121,29 @@ impl<'a> AuthenticatedContext<'a> {
             ));
         }
 
+        let wildcard = String::try_from(mailbox_wildcard.clone())?;
+        if wildcard.is_empty() {
+            if is_lsub {
+                return Ok((
+                    Response::ok("LSUB complete")?.with_body(vec![Data::Lsub {
+                        items: vec![],
+                        delimiter: Some(MAILBOX_HIERARCHY_DELIMITER),
+                        mailbox: "".try_into().unwrap(),
+                    }]),
+                    flow::Transition::None,
+                ));
+            } else {
+                return Ok((
+                    Response::ok("LIST complete")?.with_body(vec![Data::List {
+                        items: vec![],
+                        delimiter: Some(MAILBOX_HIERARCHY_DELIMITER),
+                        mailbox: "".try_into().unwrap(),
+                    }]),
+                    flow::Transition::None,
+                ));
+            }
+        }
+
         let mailboxes = self.user.list_mailboxes().await?;
         let mut vmailboxes = BTreeMap::new();
         for mb in mailboxes.iter() {
@@ -135,12 +158,9 @@ impl<'a> AuthenticatedContext<'a> {
             vmailboxes.insert(mb, true);
         }
 
-        let wildcard = String::try_from(mailbox_wildcard.clone())?;
-        let wildcard_pat = globset::Glob::new(&wildcard)?.compile_matcher();
-
         let mut ret = vec![];
         for (mb, is_real) in vmailboxes.iter() {
-            if wildcard_pat.is_match(mb) {
+            if matches_wildcard(&wildcard, &mb) {
                 let mailbox = mb
                     .to_string()
                     .try_into()
@@ -378,3 +398,49 @@ impl<'a> AuthenticatedContext<'a> {
         Ok((mb, uidvalidity, uid))
     }
 }
+
+fn matches_wildcard(wildcard: &str, name: &str) -> bool {
+    let wildcard = wildcard.chars().collect::<Vec<char>>();
+    let name = name.chars().collect::<Vec<char>>();
+
+    let mut matches = vec![vec![false; wildcard.len() + 1]; name.len() + 1];
+
+    for i in 0..=name.len() {
+        for j in 0..=wildcard.len() {
+            matches[i][j] = (i == 0 && j == 0)
+                || (j > 0
+                    && matches[i][j - 1]
+                    && (wildcard[j - 1] == '%' || wildcard[j - 1] == '*'))
+                || (i > 0
+                    && j > 0
+                    && matches[i - 1][j - 1]
+                    && wildcard[j - 1] == name[i - 1]
+                    && wildcard[j - 1] != '%'
+                    && wildcard[j - 1] != '*')
+                || (i > 0
+                    && j > 0
+                    && matches[i - 1][j]
+                    && (wildcard[j - 1] == '*'
+                        || (wildcard[j - 1] == '%' && name[i - 1] != MAILBOX_HIERARCHY_DELIMITER)));
+        }
+    }
+
+    matches[name.len()][wildcard.len()]
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_wildcard_matches() {
+        assert!(matches_wildcard("INBOX", "INBOX"));
+        assert!(matches_wildcard("*", "INBOX"));
+        assert!(matches_wildcard("%", "INBOX"));
+        assert!(!matches_wildcard("%", "Test.Azerty"));
+        assert!(!matches_wildcard("INBOX.*", "INBOX"));
+        assert!(matches_wildcard("Sent.*", "Sent.A"));
+        assert!(matches_wildcard("Sent.*", "Sent.A.B"));
+        assert!(!matches_wildcard("Sent.%", "Sent.A.B"));
+    }
+}
diff --git a/src/imap/flow.rs b/src/imap/flow.rs
index 303b498..0adf1f0 100644
--- a/src/imap/flow.rs
+++ b/src/imap/flow.rs
@@ -41,8 +41,14 @@ impl State {
         match (self, tr) {
             (s, Transition::None) => Ok(s),
             (State::NotAuthenticated, Transition::Authenticate(u)) => Ok(State::Authenticated(u)),
-            (State::Authenticated(u), Transition::Select(m)) => Ok(State::Selected(u, m)),
-            (State::Authenticated(u), Transition::Examine(m)) => Ok(State::Examined(u, m)),
+            (
+                State::Authenticated(u) | State::Selected(u, _) | State::Examined(u, _),
+                Transition::Select(m),
+            ) => Ok(State::Selected(u, m)),
+            (
+                State::Authenticated(u) | State::Selected(u, _) | State::Examined(u, _),
+                Transition::Examine(m),
+            ) => Ok(State::Examined(u, m)),
             (State::Selected(u, _), Transition::Unselect) => Ok(State::Authenticated(u)),
             (State::Examined(u, _), Transition::Unselect) => Ok(State::Authenticated(u)),
             (_, Transition::Logout) => Ok(State::Logout),
diff --git a/src/imap/session.rs b/src/imap/session.rs
index 622a3f6..15141d3 100644
--- a/src/imap/session.rs
+++ b/src/imap/session.rs
@@ -140,7 +140,13 @@ impl Instance {
             let res = match ctrl {
                 Ok((res, tr)) => {
                     //@FIXME remove unwrap
-                    self.state = self.state.apply(tr).unwrap();
+                    self.state = match self.state.apply(tr) {
+                        Ok(new_state) => new_state,
+                        Err(e) => {
+                            tracing::error!("Invalid transition: {}, exiting", e);
+                            break;
+                        }
+                    };
 
                     //@FIXME enrich here the command with some global status