Module: Portless::Daemon

Defined in:
lib/portless/daemon.rb

Overview

Starting/stopping the proxy daemon, including the privileged-port dance: for ports < 1024 we re-exec under sudo; the elevated process spawns the detached daemon that binds the socket as root. Falls back to :1355 when sudo is unavailable. Mirrors portless's handleProxy + ensureProxyRunning.

Class Method Summary collapse

Class Method Details

.cleanup_markers ⇒ Object 



165
166
167
 
# File 'lib/portless/daemon.rb', line 165

def cleanup_markers
  [ State.proxy_pid_file, State.proxy_port_file ].each { |f| File.delete(f) if File.exist?(f) }
end

.default_port(tls) ⇒ Object 



159
 
# File 'lib/portless/daemon.rb', line 159

def default_port(tls) = tls ? Constants::HTTPS_PORT : Constants::HTTP_PORT

.discovered_pid ⇒ Object

Fallback when the pid marker is missing: whoever listens on the port the proxy answered from (lsof only sees our own processes unless root).



105
106
107
108
109
 
# File 'lib/portless/daemon.rb', line 105

def discovered_pid
  port = Health.discover_port or return nil

  PortOwner.listeners(port).find { |pid| pid != Process.pid }
end

.ensure_running(tls:) ⇒ Object 



13
14
15
16
17
18
19
 
# File 'lib/portless/daemon.rb', line 13

def ensure_running(tls:)
  port = Health.discover_port
  return refresh_stale(port, tls: tls) if port

  start(tls: tls)
  Health.discover_port
end

.lib_dir ⇒ Object 



169
 
# File 'lib/portless/daemon.rb', line 169

def lib_dir = File.expand_path("..", __dir__)

.monotonic ⇒ Object 



170
 
# File 'lib/portless/daemon.rb', line 170

def monotonic = Process.clock_gettime(Process::CLOCK_MONOTONIC)

.read_pid ⇒ Object 



161
162
163
 
# File 'lib/portless/daemon.rb', line 161

def read_pid
  Integer(File.read(State.proxy_pid_file).strip, exception: false) if File.exist?(State.proxy_pid_file)
end

.refresh_stale(port, tls:) ⇒ Object

The daemon outlives gem updates — after a bundle update the process on :443 may still be running last week's code. Compare the version it stamps on its responses with ours and offer to restart it; the reverse mismatch (a newer proxy) means this project's gem is the stale side.



25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
 
# File 'lib/portless/daemon.rb', line 25

def refresh_stale(port, tls:)
  running = Health.proxy_version(port)
  case version_action(running, VERSION)
  when :restart
    if restart_consented?(running)
      restart(tls: tls, port: port)
      port = Health.discover_port || port
    else
      warn "rb-portless: keeping the v#{running} proxy — run `rb-portless proxy restart` when ready"
    end
  when :update_gem
    warn "rb-portless: the proxy is v#{running} but this project loads rb-portless v#{VERSION} — " \
         "update the gem (`bundle update rb-portless`) so they match"
  end
  port
end

.restart(tls:, port: nil) ⇒ Object 



62
63
64
65
66
67
 
# File 'lib/portless/daemon.rb', line 62

def restart(tls:, port: nil)
  port ||= Health.discover_port
  stop
  wait_until_stopped(port) if port
  start(tls: tls, port: port)
end

.restart_consented?(running) ⇒ Boolean

Returns:

  • (Boolean) 


54
55
56
57
58
59
60
 
# File 'lib/portless/daemon.rb', line 54

def restart_consented?(running)
  warn "rb-portless: the running proxy is v#{running}; this rb-portless is v#{VERSION}"
  return false unless Privilege.interactive?

  $stderr.print "rb-portless: restart the proxy to pick up the update? [Y/n] "
  !$stdin.gets.to_s.strip.downcase.start_with?("n")
end

.spawn_detached(port:, tls:) ⇒ Object 



126
127
128
129
130
131
132
133
134
135
 
# File 'lib/portless/daemon.rb', line 126

def spawn_detached(port:, tls:)
  State.ensure_dir!
  log = File.open(State.proxy_log, "a")
  args = [ RbConfig.ruby, "-I", lib_dir, Privilege.program,
           "proxy", "start", "--foreground", "--port", port.to_s, tls ? "--tls" : "--no-tls" ]
  pid = Process.spawn(*args, out: log, err: log, pgroup: true)
  Process.detach(pid)
  log.close
  wait_until_running(port)
end

.start(tls:, port: nil, foreground: false) ⇒ Object

foreground: become the daemon (binds the port, blocks). Otherwise orchestrate: elevate if needed, then spawn the detached foreground daemon.



71
72
73
74
75
76
77
78
79
80
81
82
 
# File 'lib/portless/daemon.rb', line 71

def start(tls:, port: nil, foreground: false)
  port ||= Integer(ENV["PORTLESS_PORT"], exception: false) || default_port(tls)

  return Proxy.new(port: port, tls: tls).run if foreground
  return if Health.proxy_running?(port)

  if Privilege.needs_sudo?(port) && !Privilege.root?
    start_privileged(port: port, tls: tls)
  else
    spawn_detached(port: port, tls: tls)
  end
end

.start_privileged(port:, tls:) ⇒ Object 



111
112
113
114
115
116
117
118
119
120
121
122
123
124
 
# File 'lib/portless/daemon.rb', line 111

def start_privileged(port:, tls:)
  unless Privilege.interactive?
    warn "rb-portless: can't bind :#{port} without a terminal — using :#{Constants::FALLBACK_PROXY_PORT}"
    return spawn_detached(port: Constants::FALLBACK_PROXY_PORT, tls: tls)
  end

  warn "rb-portless: binding :#{port} needs sudo (it's a privileged port) — " \
       "enter your password to serve #{tls ? 'HTTPS' : 'HTTP'} without a port number"
  ok = Privilege.reexec_with_sudo([ "proxy", "start", "--port", port.to_s, tls ? "--tls" : "--no-tls" ])
  return wait_until_running(port) if ok

  warn "rb-portless: sudo declined — using :#{Constants::FALLBACK_PROXY_PORT}"
  spawn_detached(port: Constants::FALLBACK_PROXY_PORT, tls: tls)
end

.stop ⇒ Object 



84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
 
# File 'lib/portless/daemon.rb', line 84

def stop
  pid = read_pid || discovered_pid
  unless pid
    # A proxy answers but we can't see who owns the port (a root daemon
    # whose marker files were lost) — retry the whole stop under sudo.
    return Privilege.reexec_with_sudo([ "proxy", "stop" ]) if Health.discover_port && !Privilege.root?

    warn "rb-portless: no proxy is running"
    return
  end

  Process.kill("TERM", pid)
rescue Errno::ESRCH
  cleanup_markers
rescue Errno::EPERM
  # Proxy owned by root (privileged bind) — stop it with sudo.
  Privilege.reexec_with_sudo([ "proxy", "stop" ]) unless Privilege.root?
end

.version_action(running, current) ⇒ Object

nil → proxy unreadable or versions equal: leave it alone.



43
44
45
46
47
48
49
50
51
52
 
# File 'lib/portless/daemon.rb', line 43

def version_action(running, current)
  return nil unless running

  case Gem::Version.new(running) <=> Gem::Version.new(current)
  when -1 then :restart
  when 1 then :update_gem
  end
rescue ArgumentError
  nil
end

.wait_until_running(port, timeout: 10) ⇒ Object 



137
138
139
140
141
142
143
144
145
146
 
# File 'lib/portless/daemon.rb', line 137

def wait_until_running(port, timeout: 10)
  deadline = monotonic + timeout
  until Health.proxy_running?(port)
    return false if monotonic > deadline

    sleep 0.2
  end
  State.fix_ownership
  true
end

.wait_until_stopped(port, timeout: 10) ⇒ Object

A restart can't rebind the port until the old daemon has let go of it.



149
150
151
152
153
154
155
156
157
 
# File 'lib/portless/daemon.rb', line 149

def wait_until_stopped(port, timeout: 10)
  deadline = monotonic + timeout
  while Health.proxy_running?(port)
    return false if monotonic > deadline

    sleep 0.2
  end
  true
end