Check PID

Author: sinn3r

modules/exploits/windows/local/payload_inject.rb

@@ -65,14 +65,37 @@ def exploit
 	# Figures out which PID to inject to
 	def get_pid
 		pid = datastore['PID']
-		if pid == 0 or datastore['NEWPROCESS']
+		if pid == 0 or datastore['NEWPROCESS'] or not has_pid?(pid)
 			print_status("Launching notepad.exe...")
 			pid = create_temp_proc
 		end
 
 		return pid
 	end
 
+
+	# Determines if a PID actually exists
+	def has_pid?(pid)
+		procs = []
+		begin
+			procs = client.sys.process.processes
+		rescue Rex::Post::Meterpreter::RequestError
+			print_error("Unable to enumerate processes")
+			return false
+		end
+
+		pids = []
+
+		procs.each do |p|
+			found_pid = p['pid'] 
+			return true if found_pid == pid
+		end
+
+		print_error("PID #{pid.to_s} does not actually exist.")
+
+		return false
+	end
+
 	# Checks the Architeture of a Payload and PID are compatible
 	# Returns true if they are false if they are not
 	def arch_check(pid)