rust/hedgewars-checker/src/main.rs
author unc0rr
Sat, 10 Jul 2021 12:03:50 +0200
changeset 15813 ad79e5c0885c
parent 15812 8c39a11f7756
child 15887 fe519de9c270
permissions -rw-r--r--
Begin attempt to convert checker into async using tokio
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
15811
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
     1
use anyhow::{bail, Result};
13952
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
     2
use argparse::{ArgumentParser, Store};
15811
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
     3
use hedgewars_network_protocol::{
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
     4
    messages::HwProtocolMessage as ClientMessage, messages::HwServerMessage::*, parser,
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
     5
};
13952
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
     6
use ini::Ini;
15811
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
     7
use log::{debug, info, warn};
13952
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
     8
use netbuf::Buf;
15813
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
     9
use std::{io::Write, str::FromStr};
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
    10
use tokio::{io, io::AsyncWriteExt, net::TcpStream, process::Command};
13953
e98e2fc556a7 Create replay file, run engine, result from it and send it to the server. Not implemented: filtering of engine output.
unc0rr
parents: 13952
diff changeset
    11
15813
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
    12
async fn check(executable: &str, data_prefix: &str, buffer: &[String]) -> Result<Vec<String>> {
13953
e98e2fc556a7 Create replay file, run engine, result from it and send it to the server. Not implemented: filtering of engine output.
unc0rr
parents: 13952
diff changeset
    13
    let mut replay = tempfile::NamedTempFile::new()?;
e98e2fc556a7 Create replay file, run engine, result from it and send it to the server. Not implemented: filtering of engine output.
unc0rr
parents: 13952
diff changeset
    14
15811
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
    15
    for line in buffer.into_iter() {
13955
a857cd1cc3f0 Finish hedgewars-checker implementation
unc0rr
parents: 13953
diff changeset
    16
        replay.write(&base64::decode(line)?)?;
13953
e98e2fc556a7 Create replay file, run engine, result from it and send it to the server. Not implemented: filtering of engine output.
unc0rr
parents: 13952
diff changeset
    17
    }
e98e2fc556a7 Create replay file, run engine, result from it and send it to the server. Not implemented: filtering of engine output.
unc0rr
parents: 13952
diff changeset
    18
e98e2fc556a7 Create replay file, run engine, result from it and send it to the server. Not implemented: filtering of engine output.
unc0rr
parents: 13952
diff changeset
    19
    let temp_file_path = replay.path();
e98e2fc556a7 Create replay file, run engine, result from it and send it to the server. Not implemented: filtering of engine output.
unc0rr
parents: 13952
diff changeset
    20
e98e2fc556a7 Create replay file, run engine, result from it and send it to the server. Not implemented: filtering of engine output.
unc0rr
parents: 13952
diff changeset
    21
    let mut home_dir = dirs::home_dir().unwrap();
e98e2fc556a7 Create replay file, run engine, result from it and send it to the server. Not implemented: filtering of engine output.
unc0rr
parents: 13952
diff changeset
    22
    home_dir.push(".hedgewars");
e98e2fc556a7 Create replay file, run engine, result from it and send it to the server. Not implemented: filtering of engine output.
unc0rr
parents: 13952
diff changeset
    23
13955
a857cd1cc3f0 Finish hedgewars-checker implementation
unc0rr
parents: 13953
diff changeset
    24
    debug!("Checking replay in {}", temp_file_path.to_string_lossy());
a857cd1cc3f0 Finish hedgewars-checker implementation
unc0rr
parents: 13953
diff changeset
    25
13953
e98e2fc556a7 Create replay file, run engine, result from it and send it to the server. Not implemented: filtering of engine output.
unc0rr
parents: 13952
diff changeset
    26
    let output = Command::new(executable)
e98e2fc556a7 Create replay file, run engine, result from it and send it to the server. Not implemented: filtering of engine output.
unc0rr
parents: 13952
diff changeset
    27
        .arg("--user-prefix")
e98e2fc556a7 Create replay file, run engine, result from it and send it to the server. Not implemented: filtering of engine output.
unc0rr
parents: 13952
diff changeset
    28
        .arg(&home_dir)
e98e2fc556a7 Create replay file, run engine, result from it and send it to the server. Not implemented: filtering of engine output.
unc0rr
parents: 13952
diff changeset
    29
        .arg("--prefix")
e98e2fc556a7 Create replay file, run engine, result from it and send it to the server. Not implemented: filtering of engine output.
unc0rr
parents: 13952
diff changeset
    30
        .arg(data_prefix)
e98e2fc556a7 Create replay file, run engine, result from it and send it to the server. Not implemented: filtering of engine output.
unc0rr
parents: 13952
diff changeset
    31
        .arg("--nomusic")
e98e2fc556a7 Create replay file, run engine, result from it and send it to the server. Not implemented: filtering of engine output.
unc0rr
parents: 13952
diff changeset
    32
        .arg("--nosound")
e98e2fc556a7 Create replay file, run engine, result from it and send it to the server. Not implemented: filtering of engine output.
unc0rr
parents: 13952
diff changeset
    33
        .arg("--stats-only")
e98e2fc556a7 Create replay file, run engine, result from it and send it to the server. Not implemented: filtering of engine output.
unc0rr
parents: 13952
diff changeset
    34
        .arg(temp_file_path)
15813
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
    35
        //.spawn()?
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
    36
        //.wait_with_output()
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
    37
        .output()
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
    38
        .await?;
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
    39
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
    40
    debug!("Engine finished!");
13953
e98e2fc556a7 Create replay file, run engine, result from it and send it to the server. Not implemented: filtering of engine output.
unc0rr
parents: 13952
diff changeset
    41
e98e2fc556a7 Create replay file, run engine, result from it and send it to the server. Not implemented: filtering of engine output.
unc0rr
parents: 13952
diff changeset
    42
    let mut result = Vec::new();
13955
a857cd1cc3f0 Finish hedgewars-checker implementation
unc0rr
parents: 13953
diff changeset
    43
a857cd1cc3f0 Finish hedgewars-checker implementation
unc0rr
parents: 13953
diff changeset
    44
    let mut engine_lines = output
a857cd1cc3f0 Finish hedgewars-checker implementation
unc0rr
parents: 13953
diff changeset
    45
        .stderr
a857cd1cc3f0 Finish hedgewars-checker implementation
unc0rr
parents: 13953
diff changeset
    46
        .split(|b| *b == '\n' as u8)
a857cd1cc3f0 Finish hedgewars-checker implementation
unc0rr
parents: 13953
diff changeset
    47
        .skip_while(|l| *l != b"WINNERS" && *l != b"DRAW");
a857cd1cc3f0 Finish hedgewars-checker implementation
unc0rr
parents: 13953
diff changeset
    48
15813
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
    49
    debug!("Engine lines: {:?}", &engine_lines);
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
    50
13955
a857cd1cc3f0 Finish hedgewars-checker implementation
unc0rr
parents: 13953
diff changeset
    51
    loop {
a857cd1cc3f0 Finish hedgewars-checker implementation
unc0rr
parents: 13953
diff changeset
    52
        match engine_lines.next() {
15811
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
    53
            Some(b"DRAW") => result.push("DRAW".to_owned()),
13955
a857cd1cc3f0 Finish hedgewars-checker implementation
unc0rr
parents: 13953
diff changeset
    54
            Some(b"WINNERS") => {
15811
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
    55
                result.push("WINNERS".to_owned());
13955
a857cd1cc3f0 Finish hedgewars-checker implementation
unc0rr
parents: 13953
diff changeset
    56
                let winners = engine_lines.next().unwrap();
a857cd1cc3f0 Finish hedgewars-checker implementation
unc0rr
parents: 13953
diff changeset
    57
                let winners_num = u32::from_str(&String::from_utf8(winners.to_vec())?)?;
15811
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
    58
                result.push(String::from_utf8(winners.to_vec())?);
13955
a857cd1cc3f0 Finish hedgewars-checker implementation
unc0rr
parents: 13953
diff changeset
    59
a857cd1cc3f0 Finish hedgewars-checker implementation
unc0rr
parents: 13953
diff changeset
    60
                for _i in 0..winners_num {
15811
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
    61
                    result.push(String::from_utf8(engine_lines.next().unwrap().to_vec())?);
13955
a857cd1cc3f0 Finish hedgewars-checker implementation
unc0rr
parents: 13953
diff changeset
    62
                }
a857cd1cc3f0 Finish hedgewars-checker implementation
unc0rr
parents: 13953
diff changeset
    63
            }
a857cd1cc3f0 Finish hedgewars-checker implementation
unc0rr
parents: 13953
diff changeset
    64
            Some(b"GHOST_POINTS") => {
15811
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
    65
                result.push("GHOST_POINTS".to_owned());
13955
a857cd1cc3f0 Finish hedgewars-checker implementation
unc0rr
parents: 13953
diff changeset
    66
                let points = engine_lines.next().unwrap();
a857cd1cc3f0 Finish hedgewars-checker implementation
unc0rr
parents: 13953
diff changeset
    67
                let points_num = u32::from_str(&String::from_utf8(points.to_vec())?)? * 2;
15811
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
    68
                result.push(String::from_utf8(points.to_vec())?);
13955
a857cd1cc3f0 Finish hedgewars-checker implementation
unc0rr
parents: 13953
diff changeset
    69
a857cd1cc3f0 Finish hedgewars-checker implementation
unc0rr
parents: 13953
diff changeset
    70
                for _i in 0..points_num {
15811
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
    71
                    result.push(String::from_utf8(engine_lines.next().unwrap().to_vec())?);
13955
a857cd1cc3f0 Finish hedgewars-checker implementation
unc0rr
parents: 13953
diff changeset
    72
                }
a857cd1cc3f0 Finish hedgewars-checker implementation
unc0rr
parents: 13953
diff changeset
    73
            }
a857cd1cc3f0 Finish hedgewars-checker implementation
unc0rr
parents: 13953
diff changeset
    74
            Some(b"ACHIEVEMENT") => {
15811
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
    75
                result.push("ACHIEVEMENT".to_owned());
13955
a857cd1cc3f0 Finish hedgewars-checker implementation
unc0rr
parents: 13953
diff changeset
    76
                for _i in 0..4 {
15811
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
    77
                    result.push(String::from_utf8(engine_lines.next().unwrap().to_vec())?);
13955
a857cd1cc3f0 Finish hedgewars-checker implementation
unc0rr
parents: 13953
diff changeset
    78
                }
a857cd1cc3f0 Finish hedgewars-checker implementation
unc0rr
parents: 13953
diff changeset
    79
            }
a857cd1cc3f0 Finish hedgewars-checker implementation
unc0rr
parents: 13953
diff changeset
    80
            _ => break,
a857cd1cc3f0 Finish hedgewars-checker implementation
unc0rr
parents: 13953
diff changeset
    81
        }
13953
e98e2fc556a7 Create replay file, run engine, result from it and send it to the server. Not implemented: filtering of engine output.
unc0rr
parents: 13952
diff changeset
    82
    }
e98e2fc556a7 Create replay file, run engine, result from it and send it to the server. Not implemented: filtering of engine output.
unc0rr
parents: 13952
diff changeset
    83
15813
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
    84
    println!("Engine lines: {:?}", &result);
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
    85
13955
a857cd1cc3f0 Finish hedgewars-checker implementation
unc0rr
parents: 13953
diff changeset
    86
    if result.len() > 0 {
a857cd1cc3f0 Finish hedgewars-checker implementation
unc0rr
parents: 13953
diff changeset
    87
        Ok(result)
a857cd1cc3f0 Finish hedgewars-checker implementation
unc0rr
parents: 13953
diff changeset
    88
    } else {
15811
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
    89
        bail!("no data from engine")
13955
a857cd1cc3f0 Finish hedgewars-checker implementation
unc0rr
parents: 13953
diff changeset
    90
    }
13953
e98e2fc556a7 Create replay file, run engine, result from it and send it to the server. Not implemented: filtering of engine output.
unc0rr
parents: 13952
diff changeset
    91
}
e98e2fc556a7 Create replay file, run engine, result from it and send it to the server. Not implemented: filtering of engine output.
unc0rr
parents: 13952
diff changeset
    92
15813
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
    93
async fn connect_and_run(
13952
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
    94
    username: &str,
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
    95
    password: &str,
15811
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
    96
    protocol_number: u16,
13952
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
    97
    executable: &str,
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
    98
    data_prefix: &str,
15811
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
    99
) -> Result<()> {
13952
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   100
    info!("Connecting...");
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   101
15813
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   102
    let mut stream = TcpStream::connect("hedgewars.org:46631").await?;
13952
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   103
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   104
    let mut buf = Buf::new();
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   105
15813
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   106
    let mut replay_lines: Option<Vec<String>> = None;
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   107
13952
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   108
    loop {
15813
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   109
        let r = if let Some(ref lines) = replay_lines {
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   110
            let r = tokio::select! {
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   111
                _ = stream.readable() => None,
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   112
                r = check(executable, data_prefix, &lines) => Some(r)
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   113
            };
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   114
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   115
            r
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   116
        } else {
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   117
            stream.readable().await?;
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   118
            None
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   119
        };
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   120
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   121
        println!("Loop: {:?}", &r);
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   122
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   123
        if let Some(execution_result) = r {
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   124
            replay_lines = None;
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   125
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   126
            match execution_result {
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   127
                Ok(result) => {
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   128
                    info!("Checked");
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   129
                    debug!("Check result: [{:?}]", result);
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   130
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   131
                    stream
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   132
                        .write(
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   133
                            ClientMessage::CheckedOk(result)
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   134
                                .to_raw_protocol()
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   135
                                .as_bytes(),
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   136
                        )
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   137
                        .await?;
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   138
                    stream
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   139
                        .write(ClientMessage::CheckerReady.to_raw_protocol().as_bytes())
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   140
                        .await?;
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   141
                }
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   142
                Err(e) => {
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   143
                    info!("Check failed: {:?}", e);
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   144
                    stream
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   145
                        .write(
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   146
                            ClientMessage::CheckedFail("error".to_owned())
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   147
                                .to_raw_protocol()
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   148
                                .as_bytes(),
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   149
                        )
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   150
                        .await?;
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   151
                    stream
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   152
                        .write(ClientMessage::CheckerReady.to_raw_protocol().as_bytes())
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   153
                        .await?;
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   154
                }
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   155
            }
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   156
        } else {
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   157
            let mut msg = [0; 4096];
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   158
            // Try to read data, this may still fail with `WouldBlock`
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   159
            // if the readiness event is a false positive.
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   160
            match stream.try_read(&mut msg) {
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   161
                Ok(n) => {
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   162
                    //println!("{:?}", &msg);
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   163
                    buf.write_all(&msg[0..n])?;
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   164
                }
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   165
                Err(ref e) if e.kind() == io::ErrorKind::WouldBlock => {}
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   166
                Err(e) => {
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   167
                    return Err(e.into());
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   168
                }
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   169
            }
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   170
        }
13952
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   171
15811
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
   172
        while let Ok((tail, msg)) = parser::server_message(buf.as_ref()) {
15812
8c39a11f7756 eat warnings
alfadur
parents: 15811
diff changeset
   173
            let tail_len = tail.len();
8c39a11f7756 eat warnings
alfadur
parents: 15811
diff changeset
   174
            buf.consume(buf.len() - tail_len);
13953
e98e2fc556a7 Create replay file, run engine, result from it and send it to the server. Not implemented: filtering of engine output.
unc0rr
parents: 13952
diff changeset
   175
15813
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   176
            println!("Message from server: {:?}", &msg);
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   177
15811
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
   178
            match msg {
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
   179
                Connected(_, _) => {
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
   180
                    info!("Connected");
15813
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   181
                    stream
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   182
                        .write(
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   183
                            ClientMessage::Checker(
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   184
                                protocol_number,
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   185
                                username.to_owned(),
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   186
                                password.to_owned(),
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   187
                            )
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   188
                            .to_raw_protocol()
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   189
                            .as_bytes(),
15811
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
   190
                        )
15813
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   191
                        .await?;
15811
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
   192
                }
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
   193
                Ping => {
15813
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   194
                    stream
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   195
                        .write(ClientMessage::Pong.to_raw_protocol().as_bytes())
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   196
                        .await?;
15811
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
   197
                }
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
   198
                LogonPassed => {
15813
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   199
                    stream
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   200
                        .write(ClientMessage::CheckerReady.to_raw_protocol().as_bytes())
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   201
                        .await?;
15811
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
   202
                }
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
   203
                Replay(lines) => {
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
   204
                    info!("Got a replay");
15813
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   205
                    replay_lines = Some(lines);
13955
a857cd1cc3f0 Finish hedgewars-checker implementation
unc0rr
parents: 13953
diff changeset
   206
                }
15811
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
   207
                Bye(message) => {
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
   208
                    warn!("Received BYE: {}", message);
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
   209
                    return Ok(());
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
   210
                }
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
   211
                ChatMsg { nick, msg } => {
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
   212
                    info!("Chat [{}]: {}", nick, msg);
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
   213
                }
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
   214
                RoomAdd(fields) => {
15812
8c39a11f7756 eat warnings
alfadur
parents: 15811
diff changeset
   215
                    let l = fields.into_iter();
15811
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
   216
                    info!("Room added: {}", l.skip(1).next().unwrap());
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
   217
                }
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
   218
                RoomUpdated(name, fields) => {
15812
8c39a11f7756 eat warnings
alfadur
parents: 15811
diff changeset
   219
                    let l = fields.into_iter();
15811
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
   220
                    let new_name = l.skip(1).next().unwrap();
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
   221
15812
8c39a11f7756 eat warnings
alfadur
parents: 15811
diff changeset
   222
                    if name != new_name {
15811
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
   223
                        info!("Room renamed: {}", new_name);
13957
a4877a16564d Also report chat and added rooms in log
unc0rr
parents: 13955
diff changeset
   224
                    }
a4877a16564d Also report chat and added rooms in log
unc0rr
parents: 13955
diff changeset
   225
                }
15811
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
   226
                RoomRemove(_) => {
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
   227
                    // ignore
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
   228
                }
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
   229
                Error(message) => {
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
   230
                    warn!("Received ERROR: {}", message);
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
   231
                    return Ok(());
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
   232
                }
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
   233
                something => {
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
   234
                    warn!("Unexpected protocol command: {:?}", something)
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
   235
                }
13952
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   236
            }
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   237
        }
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   238
    }
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   239
}
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   240
15813
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   241
async fn get_protocol_number(executable: &str) -> Result<u16> {
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   242
    let output = Command::new(executable).arg("--protocol").output().await?;
13952
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   243
15811
a855f32ab3ca - Update hedgewars-network-protocol library with messages needed for checker
unc0rr
parents: 14396
diff changeset
   244
    Ok(u16::from_str(&String::from_utf8(output.stdout).unwrap().trim()).unwrap_or(55))
13952
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   245
}
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   246
15813
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   247
#[tokio::main]
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   248
async fn main() -> Result<()> {
13952
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   249
    stderrlog::new()
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   250
        .verbosity(3)
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   251
        .timestamp(stderrlog::Timestamp::Second)
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   252
        .module(module_path!())
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   253
        .init()
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   254
        .unwrap();
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   255
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   256
    let mut frontend_settings = dirs::home_dir().unwrap();
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   257
    frontend_settings.push(".hedgewars/settings.ini");
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   258
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   259
    let i = Ini::load_from_file(frontend_settings.to_str().unwrap()).unwrap();
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   260
    let username = i.get_from(Some("net"), "nick").unwrap();
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   261
    let password = i.get_from(Some("net"), "passwordhash").unwrap();
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   262
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   263
    let mut exe = "/usr/local/bin/hwengine".to_string();
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   264
    let mut prefix = "/usr/local/share/hedgewars/Data".to_string();
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   265
    {
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   266
        let mut ap = ArgumentParser::new();
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   267
        ap.set_description("Game replay checker for hedgewars.");
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   268
        ap.refer(&mut exe)
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   269
            .add_option(&["--exe"], Store, "Path to hwengine executable");
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   270
        ap.refer(&mut prefix)
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   271
            .add_option(&["--prefix"], Store, "Path main Data dir");
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   272
        ap.parse_args_or_exit();
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   273
    }
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   274
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   275
    info!("Executable: {}", exe);
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   276
    info!("Data dir: {}", prefix);
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   277
15813
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   278
    let protocol_number = get_protocol_number(&exe.as_str()).await.unwrap_or_default();
13952
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   279
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   280
    info!("Using protocol number {}", protocol_number);
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   281
15813
ad79e5c0885c Begin attempt to convert checker into async using tokio
unc0rr
parents: 15812
diff changeset
   282
    connect_and_run(&username, &password, protocol_number, &exe, &prefix).await
13952
d12ca66054aa Start checker implementation in rust
unc0rr
parents:
diff changeset
   283
}