golang.zx2c4.com/wireguard/windows@v0.5.4-0.20230123132234-dcc0eb72a04b/tunnel/winipcfg/netsh.go (about)

     1  /* SPDX-License-Identifier: MIT
     2   *
     3   * Copyright (C) 2019-2022 WireGuard LLC. All Rights Reserved.
     4   */
     5  
     6  package winipcfg
     7  
     8  import (
     9  	"bytes"
    10  	"errors"
    11  	"fmt"
    12  	"io"
    13  	"net/netip"
    14  	"os/exec"
    15  	"path/filepath"
    16  	"strings"
    17  	"syscall"
    18  
    19  	"golang.org/x/sys/windows"
    20  	"golang.org/x/sys/windows/registry"
    21  )
    22  
    23  func runNetsh(cmds []string) error {
    24  	system32, err := windows.GetSystemDirectory()
    25  	if err != nil {
    26  		return err
    27  	}
    28  	cmd := exec.Command(filepath.Join(system32, "netsh.exe"))
    29  	cmd.SysProcAttr = &syscall.SysProcAttr{HideWindow: true}
    30  
    31  	stdin, err := cmd.StdinPipe()
    32  	if err != nil {
    33  		return fmt.Errorf("runNetsh stdin pipe - %w", err)
    34  	}
    35  	go func() {
    36  		defer stdin.Close()
    37  		io.WriteString(stdin, strings.Join(append(cmds, "exit\r\n"), "\r\n"))
    38  	}()
    39  	output, err := cmd.CombinedOutput()
    40  	// Horrible kludges, sorry.
    41  	cleaned := bytes.ReplaceAll(output, []byte{'\r', '\n'}, []byte{'\n'})
    42  	cleaned = bytes.ReplaceAll(cleaned, []byte("netsh>"), []byte{})
    43  	cleaned = bytes.ReplaceAll(cleaned, []byte("There are no Domain Name Servers (DNS) configured on this computer."), []byte{})
    44  	cleaned = bytes.TrimSpace(cleaned)
    45  	if len(cleaned) != 0 && err == nil {
    46  		return fmt.Errorf("netsh: %#q", string(cleaned))
    47  	} else if err != nil {
    48  		return fmt.Errorf("netsh: %v: %#q", err, string(cleaned))
    49  	}
    50  	return nil
    51  }
    52  
    53  const (
    54  	netshCmdTemplateFlush4 = "interface ipv4 set dnsservers name=%d source=static address=none validate=no register=both"
    55  	netshCmdTemplateFlush6 = "interface ipv6 set dnsservers name=%d source=static address=none validate=no register=both"
    56  	netshCmdTemplateAdd4   = "interface ipv4 add dnsservers name=%d address=%s validate=no"
    57  	netshCmdTemplateAdd6   = "interface ipv6 add dnsservers name=%d address=%s validate=no"
    58  )
    59  
    60  func (luid LUID) fallbackSetDNSForFamily(family AddressFamily, dnses []netip.Addr) error {
    61  	var templateFlush string
    62  	if family == windows.AF_INET {
    63  		templateFlush = netshCmdTemplateFlush4
    64  	} else if family == windows.AF_INET6 {
    65  		templateFlush = netshCmdTemplateFlush6
    66  	}
    67  
    68  	cmds := make([]string, 0, 1+len(dnses))
    69  	ipif, err := luid.IPInterface(family)
    70  	if err != nil {
    71  		return err
    72  	}
    73  	cmds = append(cmds, fmt.Sprintf(templateFlush, ipif.InterfaceIndex))
    74  	for i := 0; i < len(dnses); i++ {
    75  		if dnses[i].Is4() && family == windows.AF_INET {
    76  			cmds = append(cmds, fmt.Sprintf(netshCmdTemplateAdd4, ipif.InterfaceIndex, dnses[i].String()))
    77  		} else if dnses[i].Is6() && family == windows.AF_INET6 {
    78  			cmds = append(cmds, fmt.Sprintf(netshCmdTemplateAdd6, ipif.InterfaceIndex, dnses[i].String()))
    79  		}
    80  	}
    81  	return runNetsh(cmds)
    82  }
    83  
    84  func (luid LUID) fallbackSetDNSDomain(domain string) error {
    85  	guid, err := luid.GUID()
    86  	if err != nil {
    87  		return fmt.Errorf("Error converting luid to guid: %w", err)
    88  	}
    89  	key, err := registry.OpenKey(registry.LOCAL_MACHINE, fmt.Sprintf("SYSTEM\\CurrentControlSet\\Services\\Tcpip\\Parameters\\Adapters\\%v", guid), registry.QUERY_VALUE)
    90  	if err != nil {
    91  		return fmt.Errorf("Error opening adapter-specific TCP/IP network registry key: %w", err)
    92  	}
    93  	paths, _, err := key.GetStringsValue("IpConfig")
    94  	key.Close()
    95  	if err != nil {
    96  		return fmt.Errorf("Error reading IpConfig registry key: %w", err)
    97  	}
    98  	if len(paths) == 0 {
    99  		return errors.New("No TCP/IP interfaces found on adapter")
   100  	}
   101  	key, err = registry.OpenKey(registry.LOCAL_MACHINE, fmt.Sprintf("SYSTEM\\CurrentControlSet\\Services\\%s", paths[0]), registry.SET_VALUE)
   102  	if err != nil {
   103  		return fmt.Errorf("Unable to open TCP/IP network registry key: %w", err)
   104  	}
   105  	err = key.SetStringValue("Domain", domain)
   106  	key.Close()
   107  	return err
   108  }