fumiama/water
1
0
Fork 0
mirror of https://github.com/fumiama/water.git synced 2026-06-07 19:40:38 +08:00
Code Activity

Additional Cleanup (DRY)

Browse Source
This commit is contained in:
Matthew Ellison
2017-05-23 10:01:22 -04:00
parent f9df1e79f2
commit 6a94337a59
1 changed files with 23 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

59
syscalls_linux.go
View File

@@ -39,23 +39,8 @@ func newTAP(config Config) (ifce *Interface, err error) {
return nil, err
}
// Set Device Owner
if config.Owner >= 0 {
if err = ioctl(file.Fd(), syscall.TUNSETOWNER, uintptr(config.Owner)); err != nil {
return
}
}
// Set Device Group
if config.Group >= 0 {
if err = ioctl(file.Fd(), syscall.TUNSETGROUP, uintptr(config.Group)); err != nil {
return
}
}
// Set/Clear Persist Device Flag
if err = setPersistence(file.Fd(), config.Persist); err != nil {
return
if err = setDeviceOptions(file.Fd(), config); err != nil {
return nil, err
}
ifce = &Interface{isTAP: true, ReadWriteCloser: file, name: name}
@@ -72,23 +57,8 @@ func newTUN(config Config) (ifce *Interface, err error) {
return nil, err
}
// Set Device Owner
if config.Owner >= 0 {
if err = ioctl(file.Fd(), syscall.TUNSETOWNER, uintptr(config.Owner)); err != nil {
return
}
}
// Set Device Group
if config.Group >= 0 {
if err = ioctl(file.Fd(), syscall.TUNSETGROUP, uintptr(config.Group)); err != nil {
return
}
}
// Set/Clear Persist Device Flag
if err = setPersistence(file.Fd(), config.Persist); err != nil {
return
if err = setDeviceOptions(file.Fd(), config); err != nil {
return nil, err
}
ifce = &Interface{isTAP: false, ReadWriteCloser: file, name: name}
@@ -109,10 +79,27 @@ func createInterface(fd uintptr, ifName string, flags uint16) (createdIFName str
return
}
func setPersistence(fd uintptr, enabled bool) error {
func setDeviceOptions(fd uintptr, config Config) (err error) {
// Set Device Owner
if config.Owner >= 0 {
if err = ioctl(fd, syscall.TUNSETOWNER, uintptr(config.Owner)); err != nil {
return
}
}
// Set Device Group
if config.Group >= 0 {
if err = ioctl(fd, syscall.TUNSETGROUP, uintptr(config.Group)); err != nil {
return
}
}
// Set/Clear Persist Device Flag
value := 0
if enabled {
if config.Persist {
value = 1
}
return ioctl(fd, syscall.TUNSETPERSIST, uintptr(value))
}