Skip to content

Commit

Permalink
fix(puppet-agent): Fixes puppetlabs#9208 - puppet agent windows daemo…
Browse files Browse the repository at this point in the history
…n startup inconsistency
  • Loading branch information
Christian Franco committed May 7, 2024
1 parent ca922ca commit d3f0e20
Showing 1 changed file with 17 additions and 33 deletions.
50 changes: 17 additions & 33 deletions lib/puppet/util/windows/daemon.rb
Original file line number Diff line number Diff line change
Expand Up @@ -187,25 +187,6 @@ class Daemon
SetTheServiceStatus.call(SERVICE_STOPPED, NO_ERROR, 0, 0)
end

ThreadProc = FFI::Function.new(:ulong, [:pointer]) do |lpParameter|
ste = FFI::MemoryPointer.new(SERVICE_TABLE_ENTRYW, 2)

s = SERVICE_TABLE_ENTRYW.new(ste[0])
s[:lpServiceName] = FFI::MemoryPointer.from_string('')
s[:lpServiceProc] = lpParameter

s = SERVICE_TABLE_ENTRYW.new(ste[1])
s[:lpServiceName] = nil
s[:lpServiceProc] = nil

# No service to step, no service handle, no ruby exceptions, just terminate the thread..
unless StartServiceCtrlDispatcherW(ste)
return 1
end

return 0
end

# This is a shortcut for Daemon.new + Daemon#mainloop.
#
def self.mainloop
Expand Down Expand Up @@ -255,26 +236,29 @@ def mainloop
raise SystemCallError.new('CreateEvent', FFI.errno)
end

hThread = CreateThread(nil, 0, ThreadProc, Service_Main, 0, nil)
hThread = Thread.new(Service_Main) do |lp_proc|
ste = FFI::MemoryPointer.new(SERVICE_TABLE_ENTRYW, 2)

if hThread == 0
raise SystemCallError.new('CreateThread', FFI.errno)
end
s = SERVICE_TABLE_ENTRYW.new(ste[0])
s[:lpServiceName] = FFI::MemoryPointer.from_string("")
s[:lpServiceProc] = lp_proc

events = FFI::MemoryPointer.new(:pointer, 2)
events.put_pointer(0, FFI::Pointer.new(hThread))
events.put_pointer(FFI::Pointer.size, FFI::Pointer.new(@@hStartEvent))
s = SERVICE_TABLE_ENTRYW.new(ste[1])
s[:lpServiceName] = nil
s[:lpServiceProc] = nil

while (index = WaitForMultipleObjects(2, events, 0, 1000)) == WAIT_TIMEOUT
# When returning 'false', there is no service to stop, no service handle,
# no ruby exceptions, just terminate the thread.
StartServiceCtrlDispatcherW(ste)
end

if index == WAIT_FAILED
raise SystemCallError.new('WaitForMultipleObjects', FFI.errno)
while (index = WaitForSingleObject(@@hStartEvent, 1000)) == WAIT_TIMEOUT
# The thread exited, so the show is off.
raise "Service_Main thread exited abnormally" unless hThread.alive?
end

# The thread exited, so the show is off.
if index == WAIT_OBJECT_0
raise "Service_Main thread exited abnormally"
if index == WAIT_FAILED
raise SystemCallError.new("WaitForSingleObject", FFI.errno)
end

thr = Thread.new do
Expand Down Expand Up @@ -354,4 +338,4 @@ def running?
[SERVICE_RUNNING, SERVICE_PAUSED, 0].include?(@@dwServiceState)
end
end
end
end

0 comments on commit d3f0e20

Please sign in to comment.