98 lines
3.3 KiB
Ruby
98 lines
3.3 KiB
Ruby
|
#! /usr/bin/env ruby
|
||
|
|
# coding: utf-8
|
||
|
|
|
||
|
|
# Copyright © 2016 Quentin "Sardem FF7" Glidic
|
||
|
|
#
|
||
|
|
# Permission is hereby granted, free of charge, to any person obtaining a copy
|
||
|
|
# of this software and associated documentation files (the "Software"), to deal
|
||
|
|
# in the Software without restriction, including without limitation the rights
|
||
|
|
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||
|
|
# copies of the Software, and to permit persons to whom the Software is
|
||
|
|
# furnished to do so, subject to the following conditions:
|
||
|
|
#
|
||
|
|
# The above copyright notice and this permission notice shall be included in
|
||
|
|
# all copies or substantial portions of the Software.
|
||
|
|
#
|
||
|
|
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||
|
|
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||
|
|
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||
|
|
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||
|
|
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||
|
|
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||
|
|
# THE SOFTWARE.
|
||
|
|
|
||
|
|
rofi=%w( rofi -dmenu -input /dev/null -password )
|
||
|
|
|
||
|
|
|
||
|
|
def assuan_send(t)
|
||
|
|
$stdout.puts t
|
||
|
|
$stdout.flush
|
||
|
|
end
|
||
|
|
|
||
|
|
assuan_send "OK Please go ahead"
|
||
|
|
while ( line = $stdin.readline )
|
||
|
|
ok = true
|
||
|
|
case ( line )
|
||
|
|
when /^OPTION (.+)$/
|
||
|
|
#OPTION grab
|
||
|
|
#OPTION ttyname=/dev/pts/1
|
||
|
|
#OPTION ttytype=tmux-256color
|
||
|
|
#OPTION lc-messages=C
|
||
|
|
#OPTION allow-external-password-cache
|
||
|
|
#OPTION default-ok=_OK
|
||
|
|
#OPTION default-cancel=_Cancel
|
||
|
|
#OPTION default-yes=_Yes
|
||
|
|
#OPTION default-no=_No
|
||
|
|
#OPTION default-prompt=PIN:
|
||
|
|
#OPTION default-pwmngr=_Save in password manager
|
||
|
|
#OPTION default-cf-visi=Do you really want to make your passphrase visible on the screen?
|
||
|
|
#OPTION default-tt-visi=Make passphrase visible
|
||
|
|
#OPTION default-tt-hide=Hide passphrase
|
||
|
|
#OPTION touch-file=/run/user/1000/gnupg/S.gpg-agent
|
||
|
|
when /^GETINFO (.+)$/
|
||
|
|
info = $1
|
||
|
|
case ( info )
|
||
|
|
when 'pid'
|
||
|
|
assuan_send "D #{Process.pid}"
|
||
|
|
end
|
||
|
|
when /^SETKEYINFO (.+)$/
|
||
|
|
#SETKEYINFO s/FINGERPRINT
|
||
|
|
when /^SETDESC (.+)$/
|
||
|
|
print "hi\n"
|
||
|
|
#SETDESC Please enter the passphrase for the ssh key%0A ke:yf:in:ge:rp:ri:nt
|
||
|
|
rofi << '-mesg' << $1.gsub('<', '<').gsub(/%([0-9A-Fa-f]{2})/) { $1.to_i(16).chr }
|
||
|
|
when /^SETPROMPT (.+)$/
|
||
|
|
#SETPROMPT Passphrase:
|
||
|
|
rofi << '-p' << $1
|
||
|
|
when /^GETPIN$/
|
||
|
|
pass = nil
|
||
|
|
IO.popen(%w( systemctl --user show-environment )) do |io|
|
||
|
|
until ( io.eof? )
|
||
|
|
var = io.readline
|
||
|
|
var.match(/^([A-Za-z_]+)=(.+)$/)
|
||
|
|
var, val = $~[1..2]
|
||
|
|
ENV[var] = val
|
||
|
|
end
|
||
|
|
end
|
||
|
|
err_in, err_out = IO.pipe
|
||
|
|
IO.popen([*rofi, :err => err_out]) do |io|
|
||
|
|
pass = io.read
|
||
|
|
end
|
||
|
|
status = $?
|
||
|
|
err_out.close
|
||
|
|
case ( status.exitstatus )
|
||
|
|
when 0
|
||
|
|
assuan_send "D #{pass}" unless ( pass.empty? )
|
||
|
|
when 1
|
||
|
|
assuan_send "ERR 83886179 Operation cancelled <rofi>"
|
||
|
|
ok = false
|
||
|
|
end
|
||
|
|
when /^BYE(?= |$)/
|
||
|
|
exit(0)
|
||
|
|
else
|
||
|
|
assuan_send "BYE"
|
||
|
|
exit(1)
|
||
|
|
end
|
||
|
|
assuan_send "OK" if ( ok )
|
||
|
|
end
|