Class: OsCtld::Daemon

Inherits:
Object
  • Object
show all
Includes:
OsCtl::Lib::Utils::Log, OsCtl::Lib::Utils::System, Assets::Definition
Defined in:
lib/osctld/daemon.rb

Defined Under Namespace

Classes: ClientHandler

Constant Summary collapse

SOCKET = 
File.join(RunState::RUNDIR, 'osctld.sock')
@@instance = 
nil

Instance Attribute Summary collapse

Class Method Summary collapse

Instance Method Summary collapse

Methods included from Assets::Definition

#define_assets

Instance Attribute Details

#config ⇒ Config (readonly)

Returns:



69
70
71
 
# File 'lib/osctld/daemon.rb', line 69

def config
  @config
end

#initialized ⇒ Boolean (readonly)

Returns:

  • (Boolean) 


75
76
77
 
# File 'lib/osctld/daemon.rb', line 75

def initialized
  @initialized
end

#started_at ⇒ Time (readonly)

Returns:

  • (Time) 


72
73
74
 
# File 'lib/osctld/daemon.rb', line 72

def started_at
  @started_at
end

Class Method Details

.create(config) ⇒ Object

Parameters:

  • config (String) —

    path to config file



57
58
59
60
61
 
# File 'lib/osctld/daemon.rb', line 57

def create(config)
  raise 'Daemon already instantiated' if @instance

  @@instance = new(config)
end

.get ⇒ Object 



63
64
65
 
# File 'lib/osctld/daemon.rb', line 63

def get
  @@instance
end

Instance Method Details

#abort_shutdown ⇒ Object 



235
236
237
238
239
240
241
242
243
 
# File 'lib/osctld/daemon.rb', line 235

def abort_shutdown
  @abort_shutdown = true

  begin
    File.unlink(RunState::SHUTDOWN_MARKER)
  rescue Errno::ENOENT
    # ignore
  end
end

#abort_shutdown? ⇒ Boolean

Returns:

  • (Boolean) 


257
258
259
 
# File 'lib/osctld/daemon.rb', line 257

def abort_shutdown?
  @abort_shutdown
end

#assets ⇒ Object 



177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
 
# File 'lib/osctld/daemon.rb', line 177

def assets
  define_assets do |add|
    RunState.assets(add)

    add.socket(
      SOCKET,
      desc: 'Management socket',
      user: 0,
      group: 0,
      mode: 0o600
    )

    Devices::V2::BpfProgramCache.assets(add)
  end
end

#begin_shutdown ⇒ Object 



230
231
232
233
 
# File 'lib/osctld/daemon.rb', line 230

def begin_shutdown
  @abort_shutdown = false
  File.new(RunState::SHUTDOWN_MARKER, 'w', 0o000).close
end

#confirm_shutdown ⇒ Object 



245
246
247
248
249
250
251
 
# File 'lib/osctld/daemon.rb', line 245

def confirm_shutdown
  unless File.exist?(RunState::SHUTDOWN_MARKER)
    File.new(RunState::SHUTDOWN_MARKER, 'w', 0o100).close
  end

  File.chmod(0o100, RunState::SHUTDOWN_MARKER)
end

#join_server ⇒ Object 



205
206
207
 
# File 'lib/osctld/daemon.rb', line 205

def join_server
  @server_thread.join
end

#log_type ⇒ Object 



261
262
263
 
# File 'lib/osctld/daemon.rb', line 261

def log_type
  'daemon'
end

#serve ⇒ Object 



193
194
195
196
197
198
199
200
201
202
203
 
# File 'lib/osctld/daemon.rb', line 193

def serve
  @server_thread = Thread.new do
    log(:info, "Listening on control socket at #{SOCKET}")

    socket = UNIXServer.new(SOCKET)
    File.chmod(0o600, SOCKET)

    @server = Generic::Server.new(socket, Daemon::ClientHandler)
    @server.start
  end
end

#setup ⇒ Object 



111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
 
# File 'lib/osctld/daemon.rb', line 111

def setup
  # Setup /run/osctl
  RunState.create

  # Open start config
  start_cfg = RunState.open_start_config

  SystemLimits.instance

  CpuScheduler.setup

  # Increase allowed number of open files
  PrLimits.set(Process.pid, PrLimits::NOFILE, 131_072, 131_072)

  # Setup shared AppArmor files
  if AppArmor.enabled?
    log(:info, 'AppArmor enabled')
    AppArmor.setup
  else
    log(:info, 'AppArmor disabled')
  end

  # Setup BPF FS
  BpfFs.setup

  # User-control supervisor
  UserControl::Supervisor.instance

  # Send/Receive hooks and server
  SendReceive.setup

  # Setup network interfaces
  NetInterface.setup

  # Start accepting client commands
  serve

  # Load data pools
  if KernelParams.import_pools?
    autostart = KernelParams.autostart_cts?

    Commands::Pool::Import.run(all: true, autostart:)

    unless autostart
      log(:info, 'Container autostart disabled by kernel parameter')
    end
  else
    log(:info, 'Pool autoimport disabled by kernel parameter')
  end

  # Resume shutdown
  if shutdown?
    log(:info, 'Resuming shutdown')
    Commands::Self::Shutdown.run
  end

  # Close start config
  start_cfg.close

  # All components are up
  @initialized = true

  # Wait for the server to finish
  join_server
end

#shutdown? ⇒ Boolean

Returns:

  • (Boolean) 


253
254
255
 
# File 'lib/osctld/daemon.rb', line 253

def shutdown?
  File.exist?(RunState::SHUTDOWN_MARKER)
end

#stop ⇒ Object 



209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
 
# File 'lib/osctld/daemon.rb', line 209

def stop
  @stopping = true
  log(:info, 'Stopping daemon')
  Eventd.shutdown
  @server.stop if @server
  FileUtils.rm_f(SOCKET)
  UserControl.stop
  SendReceive.stop
  DB::Pools.get.each(&:stop)
  CpuScheduler.shutdown
  ThreadReaper.stop
  Monitor::Master.stop
  LockRegistry.stop
  log(:info, 'Daemon stopped successfully')
  exit(false)
end

#stopping? ⇒ Boolean

Returns:

  • (Boolean) 


226
227
228
 
# File 'lib/osctld/daemon.rb', line 226

def stopping?
  @stopping
end