1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
use std::io;
use std::task::Poll;
cfg_if::cfg_if! {
if #[cfg(unix)] {
pub(crate) use std::os::unix::io::AsRawFd as AsRawFdOrSocket;
} else if #[cfg(windows)] {
pub(crate) use std::os::windows::io::AsRawSocket as AsRawFdOrSocket;
} else {
compile_error!("async-imap-lite does not support this target OS");
}
}
pub(crate) use async_io::Async;
use async_stream_packed::SyncableWithWakerAsyncStream;
pub(crate) use async_stream_packed::{TlsClientUpgrader, UpgradableAsyncStream, UpgraderExtRefer};
use futures_lite::future;
pub(crate) use futures_lite::{AsyncRead, AsyncWrite};
use crate::imap::{Connection, Error, Result};
use crate::util::optimistic;
pub struct AsyncConnection<AS, ASTU>
where
AS: AsRawFdOrSocket,
Async<AS>: AsyncRead + AsyncWrite,
ASTU: TlsClientUpgrader<Async<AS>> + Unpin,
ASTU::Output: AsyncRead + AsyncWrite + Unpin,
{
connection: Connection<SyncableWithWakerAsyncStream<UpgradableAsyncStream<Async<AS>, ASTU>>>,
}
impl<AS, ASTU> AsyncConnection<AS, ASTU>
where
AS: AsRawFdOrSocket,
Async<AS>: AsyncRead + AsyncWrite + Unpin,
ASTU: TlsClientUpgrader<Async<AS>> + UpgraderExtRefer<Async<AS>> + Unpin,
ASTU::Output: AsyncRead + AsyncWrite + Unpin,
{
pub async fn new(
stream: UpgradableAsyncStream<Async<AS>, ASTU>,
debug: bool,
greeting_read: bool,
) -> Self {
let mut stream = Some(stream);
let connection = future::poll_fn(|cx| {
let stream =
SyncableWithWakerAsyncStream::new(stream.take().expect("never"), cx.waker());
let connection = Connection::new(stream, debug, greeting_read);
Poll::Ready(connection)
})
.await;
Self { connection }
}
pub async fn read_greeting(&mut self) -> Result<Vec<u8>> {
assert!(
!self.connection.greeting_read,
"Greeting can only be read once"
);
let mut v = Vec::new();
self.readline(&mut v).await?;
self.connection.greeting_read = true;
Ok(v)
}
pub async fn run_command_and_check_ok<S: AsRef<str>>(&mut self, command: S) -> Result<()> {
self.run_command_and_read_response(command)
.await
.map(|_| ())
}
pub async fn run_command_and_read_response<S: AsRef<str>>(
&mut self,
untagged_command: S,
) -> Result<Vec<u8>> {
self.run_command(untagged_command.as_ref()).await?;
let mut v = Vec::new();
self.read_response_onto(&mut v).await?;
Ok(v)
}
async fn readline(&mut self, into: &mut Vec<u8>) -> Result<usize> {
loop {
match self.connection.readline(into) {
Err(Error::Io(err)) if err.kind() == io::ErrorKind::WouldBlock => {}
ret => break ret,
}
optimistic(
self.connection
.get_mut()
.get_mut()
.get_mut()
.get_mut()
.readable(),
)
.await?;
}
}
async fn run_command(&mut self, untagged_command: &str) -> Result<()> {
loop {
match self.connection.run_command(untagged_command) {
Err(Error::Io(err)) if err.kind() == io::ErrorKind::WouldBlock => {}
ret => break ret,
}
optimistic(
self.connection
.get_mut()
.get_mut()
.get_mut()
.get_mut()
.writable(),
)
.await?;
}
}
async fn read_response_onto(&mut self, data: &mut Vec<u8>) -> Result<()> {
loop {
match self.connection.read_response_onto(data) {
Err(Error::Io(err)) if err.kind() == io::ErrorKind::WouldBlock => {}
ret => break ret,
}
optimistic(
self.connection
.get_mut()
.get_mut()
.get_mut()
.get_mut()
.readable(),
)
.await?;
}
}
}