diff options
Diffstat (limited to 'rs/shell/src')
-rw-r--r-- | rs/shell/src/main.rs | 143 |
1 files changed, 0 insertions, 143 deletions
diff --git a/rs/shell/src/main.rs b/rs/shell/src/main.rs deleted file mode 100644 index ef0ef48..0000000 --- a/rs/shell/src/main.rs +++ /dev/null | |||
@@ -1,143 +0,0 @@ | |||
1 | use std::{os::unix::process::CommandExt, process::ExitCode}; | ||
2 | |||
3 | fn parse_sq(s: &str) -> Option<(String, &str)> { | ||
4 | #[derive(PartialEq, Eq)] | ||
5 | enum SqState { | ||
6 | Quoted, | ||
7 | Unquoted { may_escape: bool }, | ||
8 | UnquotedEscaped, | ||
9 | } | ||
10 | |||
11 | let mut result = String::new(); | ||
12 | let mut state = SqState::Unquoted { may_escape: false }; | ||
13 | let mut remaining = ""; | ||
14 | for (i, c) in s.char_indices() { | ||
15 | match state { | ||
16 | SqState::Unquoted { may_escape: false } => { | ||
17 | if c != '\'' { | ||
18 | return None; | ||
19 | } | ||
20 | state = SqState::Quoted | ||
21 | } | ||
22 | SqState::Quoted => { | ||
23 | if c == '\'' { | ||
24 | state = SqState::Unquoted { may_escape: true }; | ||
25 | continue; | ||
26 | } | ||
27 | result.push(c); | ||
28 | } | ||
29 | SqState::Unquoted { may_escape: true } => { | ||
30 | if is_posix_space(c) { | ||
31 | remaining = &s[i..]; | ||
32 | break; | ||
33 | } | ||
34 | if c != '\\' { | ||
35 | return None; | ||
36 | } | ||
37 | state = SqState::UnquotedEscaped; | ||
38 | } | ||
39 | SqState::UnquotedEscaped => { | ||
40 | if c != '\\' && c != '!' { | ||
41 | return None; | ||
42 | } | ||
43 | result.push(c); | ||
44 | state = SqState::Unquoted { may_escape: false }; | ||
45 | } | ||
46 | } | ||
47 | } | ||
48 | |||
49 | if state != (SqState::Unquoted { may_escape: true }) { | ||
50 | return None; | ||
51 | } | ||
52 | Some((result, remaining)) | ||
53 | } | ||
54 | |||
55 | fn parse_cmd(mut cmd: &str) -> Option<Vec<String>> { | ||
56 | let mut args = Vec::<String>::new(); | ||
57 | |||
58 | cmd = cmd.trim_matches(is_posix_space); | ||
59 | while cmd != "" { | ||
60 | if cmd.starts_with('\'') { | ||
61 | let (arg, remaining) = parse_sq(cmd)?; | ||
62 | args.push(arg); | ||
63 | cmd = remaining.trim_start_matches(is_posix_space); | ||
64 | } else if let Some((arg, remaining)) = cmd.split_once(is_posix_space) { | ||
65 | args.push(arg.to_owned()); | ||
66 | cmd = remaining.trim_start_matches(is_posix_space); | ||
67 | } else { | ||
68 | args.push(cmd.to_owned()); | ||
69 | cmd = ""; | ||
70 | } | ||
71 | } | ||
72 | |||
73 | Some(args) | ||
74 | } | ||
75 | |||
76 | fn is_posix_space(c: char) -> bool { | ||
77 | // Form feed: 0x0c | ||
78 | // Vertical tab: 0x0b | ||
79 | c == ' ' || c == '\x0c' || c == '\n' || c == '\r' || c == '\t' || c == '\x0b' | ||
80 | } | ||
81 | |||
82 | fn main() -> ExitCode { | ||
83 | let bad_usage = ExitCode::from(2); | ||
84 | |||
85 | let mut args = std::env::args().skip(1); | ||
86 | if args.next() != Some("-c".to_string()) { | ||
87 | eprintln!("Expected usage: shell -c <argument>"); | ||
88 | return bad_usage; | ||
89 | } | ||
90 | let Some(cmd) = args.next() else { | ||
91 | eprintln!("Missing argument for argument '-c'"); | ||
92 | return bad_usage; | ||
93 | }; | ||
94 | if args.next() != None { | ||
95 | eprintln!("Too many arguments passed"); | ||
96 | return bad_usage; | ||
97 | } | ||
98 | |||
99 | let Some(mut cmd) = parse_cmd(&cmd) else { | ||
100 | eprintln!("Bad command"); | ||
101 | return bad_usage; | ||
102 | }; | ||
103 | |||
104 | let Some(mut program) = cmd.drain(0..1).next() else { | ||
105 | eprintln!("Bad command"); | ||
106 | return bad_usage; | ||
107 | }; | ||
108 | if program == "git" { | ||
109 | let Some(subcommand) = cmd.drain(0..1).next() else { | ||
110 | eprintln!("Bad command"); | ||
111 | return bad_usage; | ||
112 | }; | ||
113 | program.push('-'); | ||
114 | program.push_str(&subcommand); | ||
115 | } | ||
116 | |||
117 | let mut args = Vec::new(); | ||
118 | |||
119 | let git_cmds = ["git-receive-pack", "git-upload-archive", "git-upload-pack"]; | ||
120 | if git_cmds.contains(&program.as_str()) { | ||
121 | if cmd.len() != 1 { | ||
122 | eprintln!("Bad command"); | ||
123 | return bad_usage; | ||
124 | } | ||
125 | let repository = cmd[0].trim_start_matches('/'); | ||
126 | args.push(repository); | ||
127 | } else if program == "git-lfs-authenticate" { | ||
128 | if cmd.len() != 2 { | ||
129 | eprintln!("Bad command"); | ||
130 | return bad_usage; | ||
131 | } | ||
132 | let repository = cmd[0].trim_start_matches('/'); | ||
133 | args.push(repository); | ||
134 | args.push(&cmd[1]); | ||
135 | } else { | ||
136 | eprintln!("Unknown command"); | ||
137 | return bad_usage; | ||
138 | } | ||
139 | |||
140 | let e = std::process::Command::new(program).args(args).exec(); | ||
141 | eprintln!("Error: {e}"); | ||
142 | ExitCode::FAILURE | ||
143 | } | ||