public ProcessTracker(IMessageTransport transport, JobObject jobObject, IProcess hostProcess, ProcessHelper processHelper)
{
this.transport = transport;
this.jobObject = jobObject;
this.hostProcess = hostProcess;
this.processHelper = processHelper;
}
public void FiltersOutMissingProcesses()
{
var processes = new[]
{
Process.Start("cmd.exe"),
Process.Start("cmd.exe"),
};
try
{
var processHelper = new ProcessHelper();
var processIds = processes.Select(p => p.Id).ToList();
processIds.Add(1);
var result = processHelper.GetProcesses(processes.Select(p => p.Id));
Assert.Collection(result,
x => Assert.Equal(processes[0].Id, x.Id),
x => Assert.Equal(processes[1].Id, x.Id)
);
}
finally
{
foreach (var p in processes)
p.Kill();
}
}
public Container(
string id,
string handle,
IContainerUser user,
IContainerDirectory directory,
IContainerPropertyService propertyService,
ILocalTcpPortManager tcpPortManager,
JobObject jobObject,
DiskQuotaControl diskQuotaControl,
IProcessRunner processRunner,
IProcessRunner constrainedProcessRunner,
ProcessHelper processHelper,
Dictionary<string, string> defaultEnvironment,
ContainerHostDependencyHelper dependencyHelper
)
{
this.id = id;
this.handle = handle;
this.user = user;
this.directory = directory;
this.propertyService = propertyService;
this.tcpPortManager = tcpPortManager;
this.jobObject = jobObject;
this.diskQuotaControl = diskQuotaControl;
this.processRunner = processRunner;
this.constrainedProcessRunner = constrainedProcessRunner;
this.processHelper = processHelper;
this.dependencyHelper = dependencyHelper;
this.defaultEnvironment = defaultEnvironment ?? new Dictionary<string, string>(StringComparer.OrdinalIgnoreCase);
this.currentState = ContainerState.Active;
}
public void WhenProcessDoesNotExist_ReturnsNull()
{
var processHelper = new ProcessHelper();
var result = processHelper.GetProcessById(1);
Assert.Null(result);
}
public void WhenProcessIsValid_ReturnsProcess()
{
Process p = Process.Start("cmd.exe");
try
{
var processHelper = new ProcessHelper();
var result = processHelper.GetProcessById(p.Id);
Assert.NotNull(result);
Assert.Equal(p.Id, result.Id);
}
finally
{
p.Kill();
}
}
IContainer RestoreContainerFromPath(string containerPath)
{
var id = Path.GetFileName(containerPath);
var user = ContainerUser.Restore(userManager, id);
var directory = ContainerDirectory.Restore(fileSystem, containerPath);
var jobObjectName = id;
var jobObject = new JobObject(jobObjectName);
var environment = new Dictionary<string, string>();
var processHelper = new ProcessHelper();
var diskQuotaControl = new DiskQuotaControl();
diskQuotaControl.UserNameResolution = UserNameResolutionConstants.dqResolveNone;
diskQuotaControl.Initialize(directory.Volume, true);
var dependencyHelper = new ContainerHostDependencyHelper();
var container = new Container(
id,
id, // TODO: Recover the handle from container metadata
user,
directory,
containerPropertiesService,
tcpPortManager,
jobObject,
diskQuotaControl,
processRunner,
processRunner,
processHelper,
environment,
dependencyHelper);
return container;
}
public IContainer CreateContainer(ContainerSpec containerSpec)
{
Guard.NotNull(containerSpec, "containerSpec");
UndoStack undoStack = new UndoStack();
IContainer container;
try
{
var handle = containerSpec.Handle;
if (String.IsNullOrEmpty(handle))
handle = handleHelper.GenerateHandle();
var id = handleHelper.GenerateId(handle);
var user = ContainerUser.Create(userManager, id);
undoStack.Push(() => user.Delete());
var directory = directoryFactory.Create(fileSystem, containerBasePath, id);
directory.CreateSubdirectories(user);
undoStack.Push(directory.Destroy);
directory.CreateBindMounts(containerSpec.BindMounts, user);
var jobObject = new JobObject(id);
undoStack.Push(() => jobObject.Dispose());
var containerHostClient = containerHostService.StartContainerHost(id, directory, jobObject, user.GetCredential());
undoStack.Push(() => containerHostClient.Shutdown());
var constrainedProcessRunner = new ConstrainedProcessRunner(containerHostClient);
undoStack.Push(() => constrainedProcessRunner.Dispose());
var processHelper = new ProcessHelper();
var dependencyHelper = new ContainerHostDependencyHelper();
var diskQuotaControl = diskQuotaManager.CreateDiskQuotaControl(directory);
container = new Container(
id,
handle,
user,
directory,
containerPropertiesService,
tcpPortManager,
jobObject,
diskQuotaControl,
processRunner,
constrainedProcessRunner,
processHelper,
containerSpec.Environment,
dependencyHelper);
containerPropertiesService.SetProperties(container, containerSpec.Properties);
lock (containers)
{
containers.Add(container);
}
}
catch (Exception e)
{
try
{
undoStack.UndoAll();
throw;
}
catch (AggregateException undoException)
{
throw new AggregateException(new[] { e, undoException });
}
}
return container;
}
public void ReturnsProcesses()
{
var processes = new []
{
Process.Start("cmd.exe"),
Process.Start("cmd.exe"),
};
try
{
var processHelper = new ProcessHelper();
var result = processHelper.GetProcesses(processes.Select(p => p.Id));
Assert.Collection(result,
x => Assert.Equal(processes[0].Id, x.Id),
x => Assert.Equal(processes[1].Id, x.Id)
);
}
finally
{
foreach (var p in processes)
p.Kill();
}
}
