76 lines
2.3 KiB
Ruby
76 lines
2.3 KiB
Ruby
|
require 'ffi'
|
||
|
|
|
||
|
|
module PTY
|
||
|
|
private
|
||
|
|
module LibC
|
||
|
|
extend FFI::Library
|
||
|
|
ffi_lib FFI::Library::LIBC
|
||
|
|
attach_function :forkpty, [ :buffer_out, :buffer_out, :buffer_in, :buffer_in ], :int
|
||
|
|
attach_function :openpty, [ :buffer_out, :buffer_out, :buffer_out, :buffer_in, :buffer_in ], :int
|
||
|
|
attach_function :login_tty, [ :int ], :int
|
||
|
|
attach_function :close, [ :int ], :int
|
||
|
|
attach_function :strerror, [ :int ], :string
|
||
|
|
attach_function :fork, [], :int
|
||
|
|
attach_function :execv, [ :string, :buffer_in ], :int
|
||
|
|
attach_function :execvp, [ :string, :buffer_in ], :int
|
||
|
|
attach_function :dup2, [ :int, :int ], :int
|
||
|
|
attach_function :dup, [ :int ], :int
|
||
|
|
end
|
||
|
|
Buffer = FFI::Buffer
|
||
|
|
def self.build_args(args)
|
||
|
|
cmd = args.shift
|
||
|
|
cmd_args = args.map do |arg|
|
||
|
|
MemoryPointer.from_string(arg)
|
||
|
|
end
|
||
|
|
exec_args = MemoryPointer.new(:pointer, 1 + cmd_args.length + 1)
|
||
|
|
exec_cmd = MemoryPointer.from_string(cmd)
|
||
|
|
exec_args[0].put_pointer(0, exec_cmd)
|
||
|
|
cmd_args.each_with_index do |arg, i|
|
||
|
|
exec_args[i + 1].put_pointer(0, arg)
|
||
|
|
end
|
||
|
|
[ cmd, exec_args ]
|
||
|
|
end
|
||
|
|
public
|
||
|
|
def self.getpty(*args)
|
||
|
|
mfdp = Buffer.new :int
|
||
|
|
name = Buffer.new 1024
|
||
|
|
#
|
||
|
|
# All the execv setup is done in the parent, since doing anything other than
|
||
|
|
# execv in the child after fork is really flakey
|
||
|
|
#
|
||
|
|
exec_cmd, exec_args = build_args(args)
|
||
|
|
pid = LibC.forkpty(mfdp, name, nil, nil)
|
||
|
|
raise "forkpty failed: #{LibC.strerror(FFI.errno)}" if pid < 0
|
||
|
|
if pid == 0
|
||
|
|
LibC.execvp(exec_cmd, exec_args)
|
||
|
|
exit 1
|
||
|
|
end
|
||
|
|
masterfd = mfdp.get_int(0)
|
||
|
|
rfp = FFI::IO.for_fd(masterfd, "r")
|
||
|
|
wfp = FFI::IO.for_fd(LibC.dup(masterfd), "w")
|
||
|
|
if block_given?
|
||
|
|
yield rfp, wfp, pid
|
||
|
|
rfp.close unless rfp.closed?
|
||
|
|
wfp.close unless wfp.closed?
|
||
|
|
else
|
||
|
|
[ rfp, wfp, pid ]
|
||
|
|
end
|
||
|
|
end
|
||
|
|
def self.spawn(*args, &block)
|
||
|
|
self.getpty("/bin/sh", "-c", args[0], &block)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
module LibC
|
||
|
|
extend FFI::Library
|
||
|
|
attach_function :close, [ :int ], :int
|
||
|
|
attach_function :write, [ :int, :buffer_in, :ulong ], :long
|
||
|
|
attach_function :read, [ :int, :buffer_out, :ulong ], :long
|
||
|
|
end
|
||
|
|
PTY.getpty("/bin/ls", "-alR", "/") { |rfd, wfd, pid|
|
||
|
|
#PTY.spawn("ls -laR /") { |rfd, wfd, pid|
|
||
|
|
puts "child pid=#{pid}"
|
||
|
|
while !rfd.eof? && (buf = rfd.gets)
|
||
|
|
puts "child: '#{buf.strip}'"
|
||
|
|
end
|
||
|
|
}
|