Signed-off-by: Evan Lezar <elezar@nvidia.com>
| 1 | 1 |
new file mode 100644 |
| ... | ... |
@@ -0,0 +1,38 @@ |
| 0 |
+package daemon // import "github.com/docker/docker/daemon" |
|
| 1 |
+ |
|
| 2 |
+import ( |
|
| 3 |
+ "github.com/docker/docker/api/types/container" |
|
| 4 |
+ "github.com/docker/docker/pkg/capabilities" |
|
| 5 |
+ specs "github.com/opencontainers/runtime-spec/specs-go" |
|
| 6 |
+) |
|
| 7 |
+ |
|
| 8 |
+var deviceDrivers = map[string]*deviceDriver{}
|
|
| 9 |
+ |
|
| 10 |
+type deviceDriver struct {
|
|
| 11 |
+ capset capabilities.Set |
|
| 12 |
+ updateSpec func(*specs.Spec, *deviceInstance) error |
|
| 13 |
+} |
|
| 14 |
+ |
|
| 15 |
+type deviceInstance struct {
|
|
| 16 |
+ req container.DeviceRequest |
|
| 17 |
+ selectedCaps []string |
|
| 18 |
+} |
|
| 19 |
+ |
|
| 20 |
+func registerDeviceDriver(name string, d *deviceDriver) {
|
|
| 21 |
+ deviceDrivers[name] = d |
|
| 22 |
+} |
|
| 23 |
+ |
|
| 24 |
+func (daemon *Daemon) handleDevice(req container.DeviceRequest, spec *specs.Spec) error {
|
|
| 25 |
+ if req.Driver == "" {
|
|
| 26 |
+ for _, dd := range deviceDrivers {
|
|
| 27 |
+ if selected := dd.capset.Match(req.Capabilities); selected != nil {
|
|
| 28 |
+ return dd.updateSpec(spec, &deviceInstance{req: req, selectedCaps: selected})
|
|
| 29 |
+ } |
|
| 30 |
+ } |
|
| 31 |
+ } else if dd := deviceDrivers[req.Driver]; dd != nil {
|
|
| 32 |
+ if selected := dd.capset.Match(req.Capabilities); selected != nil {
|
|
| 33 |
+ return dd.updateSpec(spec, &deviceInstance{req: req, selectedCaps: selected})
|
|
| 34 |
+ } |
|
| 35 |
+ } |
|
| 36 |
+ return incompatibleDeviceRequest{req.Driver, req.Capabilities}
|
|
| 37 |
+} |
| 0 | 38 |
deleted file mode 100644 |
| ... | ... |
@@ -1,38 +0,0 @@ |
| 1 |
-package daemon // import "github.com/docker/docker/daemon" |
|
| 2 |
- |
|
| 3 |
-import ( |
|
| 4 |
- "github.com/docker/docker/api/types/container" |
|
| 5 |
- "github.com/docker/docker/pkg/capabilities" |
|
| 6 |
- specs "github.com/opencontainers/runtime-spec/specs-go" |
|
| 7 |
-) |
|
| 8 |
- |
|
| 9 |
-var deviceDrivers = map[string]*deviceDriver{}
|
|
| 10 |
- |
|
| 11 |
-type deviceDriver struct {
|
|
| 12 |
- capset capabilities.Set |
|
| 13 |
- updateSpec func(*specs.Spec, *deviceInstance) error |
|
| 14 |
-} |
|
| 15 |
- |
|
| 16 |
-type deviceInstance struct {
|
|
| 17 |
- req container.DeviceRequest |
|
| 18 |
- selectedCaps []string |
|
| 19 |
-} |
|
| 20 |
- |
|
| 21 |
-func registerDeviceDriver(name string, d *deviceDriver) {
|
|
| 22 |
- deviceDrivers[name] = d |
|
| 23 |
-} |
|
| 24 |
- |
|
| 25 |
-func (daemon *Daemon) handleDevice(req container.DeviceRequest, spec *specs.Spec) error {
|
|
| 26 |
- if req.Driver == "" {
|
|
| 27 |
- for _, dd := range deviceDrivers {
|
|
| 28 |
- if selected := dd.capset.Match(req.Capabilities); selected != nil {
|
|
| 29 |
- return dd.updateSpec(spec, &deviceInstance{req: req, selectedCaps: selected})
|
|
| 30 |
- } |
|
| 31 |
- } |
|
| 32 |
- } else if dd := deviceDrivers[req.Driver]; dd != nil {
|
|
| 33 |
- if selected := dd.capset.Match(req.Capabilities); selected != nil {
|
|
| 34 |
- return dd.updateSpec(spec, &deviceInstance{req: req, selectedCaps: selected})
|
|
| 35 |
- } |
|
| 36 |
- } |
|
| 37 |
- return incompatibleDeviceRequest{req.Driver, req.Capabilities}
|
|
| 38 |
-} |