Class: Distem::Node::Container

198
199
200
201

# File 'lib/distem/node/container.rb', line 198

def destroy
  stop() if !@stopped
  remove()
end

203
204
205

# File 'lib/distem/node/container.rb', line 203

def freeze
  LXCWrapper::Command.freeze(@vnode.name)
end

212
213
214
215

# File 'lib/distem/node/container.rb', line 212

def reconfigure
    @cpuforge.apply
    @networkforges.each_value { |netforge| netforge.apply }
end

187
188
189
190
191
192
193
194
195

# File 'lib/distem/node/container.rb', line 187

def remove
  LXCWrapper::Command.destroy(@vnode.name,true)
  if !@vnode.filesystem.shared && @vnode.filesystem.cow
    # The subvolume deletion is performed automatically by LXC in the Jessie version.
    if File.exist?(@vnode.filesystem.path)
      Lib::Shell.run("btrfs subvolume delete #{@vnode.filesystem.path}")
    end
  end
end

225
226
227
228
229
230
231

# File 'lib/distem/node/container.rb', line 225

def set_global_arptable(data, file)
  rootfspath = @vnode.filesystem.shared ? @vnode.filesystem.sharedpath : @vnode.filesystem.path
  path = File.join(rootfspath, file)
  File.open(path,'w') {|f|
    f.write(data + "\n")
  }
end

217
218
219
220
221
222
223

# File 'lib/distem/node/container.rb', line 217

def set_global_etchosts(data)
  rootfspath = @vnode.filesystem.shared ? @vnode.filesystem.sharedpath : @vnode.filesystem.path
  etcpath = File.join(rootfspath,'etc')
  File.open(File.join(etcpath,'hosts'),'w') {|f|
    f.write(data + "\n")
  }
end

65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125

# File 'lib/distem/node/container.rb', line 65

def setup()
  rootfspath = nil
  if @vnode.filesystem.shared
    rootfspath = @vnode.filesystem.sharedpath
  else
    rootfspath = @vnode.filesystem.path
  end
  sshpath = File.join(rootfspath,'root','.ssh')

  # Creating SSH directory
  unless File.exist?(sshpath)
    Lib::Shell.run("mkdir -p #{sshpath}")
  end

  # Copying every private keys if not already existing
  Daemon::Admin.ssh_keys_priv.each do |keyfile|
    keypath=File.join(sshpath,"#{SSH_KEY_PREFIX}#{File.basename(keyfile)}")
    Lib::Shell.run("cp #{keyfile} #{keypath}") unless File.exist?(keypath)
  end
  File.open(File.join(sshpath,SSH_KEY_FILENAME),'w') do |f|
    f.puts @vnode.sshkey['private']
  end if @vnode.sshkey and @vnode.sshkey['private']

  # Copying every public keys if not already existing
  Daemon::Admin.ssh_keys_pub.each do |keyfile|
    keypath=File.join(sshpath,"#{SSH_KEY_PREFIX}#{File.basename(keyfile)}")
    Lib::Shell.run("cp #{keyfile} #{keypath}") unless File.exist?(keypath)
  end
  File.open(File.join(sshpath,"#{SSH_KEY_FILENAME}.pub"),'w') do |f|
    f.puts @vnode.sshkey['public']
  end if @vnode.sshkey and @vnode.sshkey['public']

  # Copying authorized_keys file of the host
  hostauthfile = File.join(Daemon::Admin::PATH_SSH,'authorized_keys')
  authfile = File.join(sshpath,'authorized_keys')
  if File.exist?(authfile)
    authkeys = IO.readlines(authfile).collect{|v| v.chomp}
    hostauthkeys = IO.readlines(hostauthfile).collect{|v| v.chomp}
    hostauthkeys.each do |key|
      File.open(authfile,'a') { |f| f.puts key } unless \
        authkeys.include?(key)
    end
  else
    Lib::Shell.run("cp -f #{hostauthfile} #{authfile}") if File.exist?(hostauthfile)
  end

  # Adding public keys to SSH authorized_keys file
  pubkeys = Daemon::Admin.ssh_keys_pub.collect{ |v| IO.read(v).chomp }
  pubkeys << @vnode.sshkey['public'] if @vnode.sshkey and @vnode.sshkey['public']
  if File.exist?(authfile)
    authkeys = IO.readlines(authfile).collect{|v| v.chomp} unless authkeys
    pubkeys.each do |key|
      File.open(authfile,'a') { |f| f.puts key } unless \
        authkeys.include?(key)
    end
  else
    pubkeys.each do |key|
      File.open(authfile,'a') { |f| f.puts key }
    end
  end
end

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

# File 'lib/distem/node/container.rb', line 142

def start
  raise @vnode.name if @vnode.status == Resource::Status::RUNNING
  @@contsem.synchronize do
    LXCWrapper::Command.start(@vnode.name)
    @vnode.vifaces.each do |viface|
      Lib::Shell::run("ethtool -K #{Lib::NetTools.get_iface_name(@vnode,viface)} gso off || true")
      Lib::Shell::run("ethtool -K #{Lib::NetTools.get_iface_name(@vnode,viface)} tso off || true")
    end
    @cpuforge.apply
    @networkforges.each_value { |netforge| netforge.apply }
  end
  if @vnode.filesystem.disk_throttling
    # On jessie, this does not return a correct value ...
    #major = File.stat(@vnode.filesystem.disk_throttling['device']).dev_major
    major = `stat --printf %t #{@vnode.filesystem.disk_throttling['device']}`
    cgroup_path = File.join(Distem::Node::Admin::PATH_CGROUP, 'lxc', @vnode.name)
    # In the description, read and write limits are supposed to be specified in bytes
    if @vnode.filesystem.disk_throttling['read_limit']
      limit = @vnode.filesystem.disk_throttling['read_limit'].to_i
      Lib::Shell::run("echo \"#{major}:0 #{limit}\" > #{cgroup_path}/blkio.throttle.read_bps_device")
    end
    if @vnode.filesystem.disk_throttling['write_limit']
      limit = @vnode.filesystem.disk_throttling['write_limit'].to_i
      Lib::Shell::run("echo \"#{major}:0 #{limit}\" > #{cgroup_path}/blkio.throttle.write_bps_device")
    end
  end
  @stopped = false
end

172
173
174
175
176
177
178
179
180
181
182
183
184

# File 'lib/distem/node/container.rb', line 172

def stop
  if @vnode.filesystem.disk_throttling
    cgroup_path = File.join(Distem::Node::Admin::PATH_CGROUP, 'lxc', @vnode.name)
    Lib::Shell::run("echo -n > #{cgroup_path}/blkio.throttle.read_bps_device") if @vnode.filesystem.disk_throttling['read_limit']
    Lib::Shell::run("echo -n > #{cgroup_path}/blkio.throttle.write_bps_device") if @vnode.filesystem.disk_throttling['write_limit']
  end
  @@contsem.synchronize do
    @cpuforge.undo
    @networkforges.each_value { |netforge| netforge.undo }
    LXCWrapper::Command.stop(@vnode.name)
  end
  @stopped = true
end

207
208
209

# File 'lib/distem/node/container.rb', line 207

def unfreeze
  LXCWrapper::Command.unfreeze(@vnode.name)
end