Get the process IDs that run the command in the remote machine.
(ip, port, cmd_regex)
| 153 | |
| 154 | |
| 155 | def get_remote_pids(ip, port, cmd_regex): |
| 156 | """Get the process IDs that run the command in the remote machine.""" |
| 157 | pids = [] |
| 158 | curr_pid = os.getpid() |
| 159 | # Here we want to get the python processes. We may get some ssh processes, so we should filter them out. |
| 160 | ps_cmd = ( |
| 161 | "ssh -o StrictHostKeyChecking=no -p " |
| 162 | + str(port) |
| 163 | + " " |
| 164 | + ip |
| 165 | + " 'ps -aux | grep python | grep -v StrictHostKeyChecking'" |
| 166 | ) |
| 167 | res = subprocess.run(ps_cmd, shell=True, stdout=subprocess.PIPE) |
| 168 | for p in res.stdout.decode("utf-8").split("\n"): |
| 169 | l = p.split() |
| 170 | if len(l) < 2: |
| 171 | continue |
| 172 | # We only get the processes that run the specified command. |
| 173 | res = re.search(cmd_regex, p) |
| 174 | if res is not None and int(l[1]) != curr_pid: |
| 175 | pids.append(l[1]) |
| 176 | |
| 177 | pid_str = ",".join([str(pid) for pid in pids]) |
| 178 | ps_cmd = ( |
| 179 | "ssh -o StrictHostKeyChecking=no -p " |
| 180 | + str(port) |
| 181 | + " " |
| 182 | + ip |
| 183 | + " 'pgrep -P {}'".format(pid_str) |
| 184 | ) |
| 185 | res = subprocess.run(ps_cmd, shell=True, stdout=subprocess.PIPE) |
| 186 | pids1 = res.stdout.decode("utf-8").split("\n") |
| 187 | all_pids = [] |
| 188 | for pid in set(pids + pids1): |
| 189 | if pid == "" or int(pid) == curr_pid: |
| 190 | continue |
| 191 | all_pids.append(int(pid)) |
| 192 | all_pids.sort() |
| 193 | return all_pids |
| 194 | |
| 195 | |
| 196 | def get_all_remote_pids(hosts, ssh_port, udf_command): |