mirror of
				https://gitee.com/gitea/gitea
				synced 2025-11-04 16:40:24 +08:00 
			
		
		
		
	
		
			
				
	
	
		
			356 lines
		
	
	
		
			8.0 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			356 lines
		
	
	
		
			8.0 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
// Copyright 2019 The Go Authors. All rights reserved.
 | 
						|
// Use of this source code is governed by a BSD-style
 | 
						|
// license that can be found in the LICENSE file.
 | 
						|
 | 
						|
// +build ignore
 | 
						|
 | 
						|
// Parse the header files for OpenBSD and generate a Go usable sysctl MIB.
 | 
						|
//
 | 
						|
// Build a MIB with each entry being an array containing the level, type and
 | 
						|
// a hash that will contain additional entries if the current entry is a node.
 | 
						|
// We then walk this MIB and create a flattened sysctl name to OID hash.
 | 
						|
 | 
						|
package main
 | 
						|
 | 
						|
import (
 | 
						|
	"bufio"
 | 
						|
	"fmt"
 | 
						|
	"os"
 | 
						|
	"path/filepath"
 | 
						|
	"regexp"
 | 
						|
	"sort"
 | 
						|
	"strings"
 | 
						|
)
 | 
						|
 | 
						|
var (
 | 
						|
	goos, goarch string
 | 
						|
)
 | 
						|
 | 
						|
// cmdLine returns this programs's commandline arguments.
 | 
						|
func cmdLine() string {
 | 
						|
	return "go run mksysctl_openbsd.go " + strings.Join(os.Args[1:], " ")
 | 
						|
}
 | 
						|
 | 
						|
// buildTags returns build tags.
 | 
						|
func buildTags() string {
 | 
						|
	return fmt.Sprintf("%s,%s", goarch, goos)
 | 
						|
}
 | 
						|
 | 
						|
// reMatch performs regular expression match and stores the substring slice to value pointed by m.
 | 
						|
func reMatch(re *regexp.Regexp, str string, m *[]string) bool {
 | 
						|
	*m = re.FindStringSubmatch(str)
 | 
						|
	if *m != nil {
 | 
						|
		return true
 | 
						|
	}
 | 
						|
	return false
 | 
						|
}
 | 
						|
 | 
						|
type nodeElement struct {
 | 
						|
	n  int
 | 
						|
	t  string
 | 
						|
	pE *map[string]nodeElement
 | 
						|
}
 | 
						|
 | 
						|
var (
 | 
						|
	debugEnabled bool
 | 
						|
	mib          map[string]nodeElement
 | 
						|
	node         *map[string]nodeElement
 | 
						|
	nodeMap      map[string]string
 | 
						|
	sysCtl       []string
 | 
						|
)
 | 
						|
 | 
						|
var (
 | 
						|
	ctlNames1RE = regexp.MustCompile(`^#define\s+(CTL_NAMES)\s+{`)
 | 
						|
	ctlNames2RE = regexp.MustCompile(`^#define\s+(CTL_(.*)_NAMES)\s+{`)
 | 
						|
	ctlNames3RE = regexp.MustCompile(`^#define\s+((.*)CTL_NAMES)\s+{`)
 | 
						|
	netInetRE   = regexp.MustCompile(`^netinet/`)
 | 
						|
	netInet6RE  = regexp.MustCompile(`^netinet6/`)
 | 
						|
	netRE       = regexp.MustCompile(`^net/`)
 | 
						|
	bracesRE    = regexp.MustCompile(`{.*}`)
 | 
						|
	ctlTypeRE   = regexp.MustCompile(`{\s+"(\w+)",\s+(CTLTYPE_[A-Z]+)\s+}`)
 | 
						|
	fsNetKernRE = regexp.MustCompile(`^(fs|net|kern)_`)
 | 
						|
)
 | 
						|
 | 
						|
func debug(s string) {
 | 
						|
	if debugEnabled {
 | 
						|
		fmt.Fprintln(os.Stderr, s)
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
// Walk the MIB and build a sysctl name to OID mapping.
 | 
						|
func buildSysctl(pNode *map[string]nodeElement, name string, oid []int) {
 | 
						|
	lNode := pNode // local copy of pointer to node
 | 
						|
	var keys []string
 | 
						|
	for k := range *lNode {
 | 
						|
		keys = append(keys, k)
 | 
						|
	}
 | 
						|
	sort.Strings(keys)
 | 
						|
 | 
						|
	for _, key := range keys {
 | 
						|
		nodename := name
 | 
						|
		if name != "" {
 | 
						|
			nodename += "."
 | 
						|
		}
 | 
						|
		nodename += key
 | 
						|
 | 
						|
		nodeoid := append(oid, (*pNode)[key].n)
 | 
						|
 | 
						|
		if (*pNode)[key].t == `CTLTYPE_NODE` {
 | 
						|
			if _, ok := nodeMap[nodename]; ok {
 | 
						|
				lNode = &mib
 | 
						|
				ctlName := nodeMap[nodename]
 | 
						|
				for _, part := range strings.Split(ctlName, ".") {
 | 
						|
					lNode = ((*lNode)[part]).pE
 | 
						|
				}
 | 
						|
			} else {
 | 
						|
				lNode = (*pNode)[key].pE
 | 
						|
			}
 | 
						|
			buildSysctl(lNode, nodename, nodeoid)
 | 
						|
		} else if (*pNode)[key].t != "" {
 | 
						|
			oidStr := []string{}
 | 
						|
			for j := range nodeoid {
 | 
						|
				oidStr = append(oidStr, fmt.Sprintf("%d", nodeoid[j]))
 | 
						|
			}
 | 
						|
			text := "\t{ \"" + nodename + "\", []_C_int{ " + strings.Join(oidStr, ", ") + " } }, \n"
 | 
						|
			sysCtl = append(sysCtl, text)
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
func main() {
 | 
						|
	// Get the OS (using GOOS_TARGET if it exist)
 | 
						|
	goos = os.Getenv("GOOS_TARGET")
 | 
						|
	if goos == "" {
 | 
						|
		goos = os.Getenv("GOOS")
 | 
						|
	}
 | 
						|
	// Get the architecture (using GOARCH_TARGET if it exists)
 | 
						|
	goarch = os.Getenv("GOARCH_TARGET")
 | 
						|
	if goarch == "" {
 | 
						|
		goarch = os.Getenv("GOARCH")
 | 
						|
	}
 | 
						|
	// Check if GOOS and GOARCH environment variables are defined
 | 
						|
	if goarch == "" || goos == "" {
 | 
						|
		fmt.Fprintf(os.Stderr, "GOARCH or GOOS not defined in environment\n")
 | 
						|
		os.Exit(1)
 | 
						|
	}
 | 
						|
 | 
						|
	mib = make(map[string]nodeElement)
 | 
						|
	headers := [...]string{
 | 
						|
		`sys/sysctl.h`,
 | 
						|
		`sys/socket.h`,
 | 
						|
		`sys/tty.h`,
 | 
						|
		`sys/malloc.h`,
 | 
						|
		`sys/mount.h`,
 | 
						|
		`sys/namei.h`,
 | 
						|
		`sys/sem.h`,
 | 
						|
		`sys/shm.h`,
 | 
						|
		`sys/vmmeter.h`,
 | 
						|
		`uvm/uvmexp.h`,
 | 
						|
		`uvm/uvm_param.h`,
 | 
						|
		`uvm/uvm_swap_encrypt.h`,
 | 
						|
		`ddb/db_var.h`,
 | 
						|
		`net/if.h`,
 | 
						|
		`net/if_pfsync.h`,
 | 
						|
		`net/pipex.h`,
 | 
						|
		`netinet/in.h`,
 | 
						|
		`netinet/icmp_var.h`,
 | 
						|
		`netinet/igmp_var.h`,
 | 
						|
		`netinet/ip_ah.h`,
 | 
						|
		`netinet/ip_carp.h`,
 | 
						|
		`netinet/ip_divert.h`,
 | 
						|
		`netinet/ip_esp.h`,
 | 
						|
		`netinet/ip_ether.h`,
 | 
						|
		`netinet/ip_gre.h`,
 | 
						|
		`netinet/ip_ipcomp.h`,
 | 
						|
		`netinet/ip_ipip.h`,
 | 
						|
		`netinet/pim_var.h`,
 | 
						|
		`netinet/tcp_var.h`,
 | 
						|
		`netinet/udp_var.h`,
 | 
						|
		`netinet6/in6.h`,
 | 
						|
		`netinet6/ip6_divert.h`,
 | 
						|
		`netinet6/pim6_var.h`,
 | 
						|
		`netinet/icmp6.h`,
 | 
						|
		`netmpls/mpls.h`,
 | 
						|
	}
 | 
						|
 | 
						|
	ctls := [...]string{
 | 
						|
		`kern`,
 | 
						|
		`vm`,
 | 
						|
		`fs`,
 | 
						|
		`net`,
 | 
						|
		//debug			/* Special handling required */
 | 
						|
		`hw`,
 | 
						|
		//machdep		/* Arch specific */
 | 
						|
		`user`,
 | 
						|
		`ddb`,
 | 
						|
		//vfs			/* Special handling required */
 | 
						|
		`fs.posix`,
 | 
						|
		`kern.forkstat`,
 | 
						|
		`kern.intrcnt`,
 | 
						|
		`kern.malloc`,
 | 
						|
		`kern.nchstats`,
 | 
						|
		`kern.seminfo`,
 | 
						|
		`kern.shminfo`,
 | 
						|
		`kern.timecounter`,
 | 
						|
		`kern.tty`,
 | 
						|
		`kern.watchdog`,
 | 
						|
		`net.bpf`,
 | 
						|
		`net.ifq`,
 | 
						|
		`net.inet`,
 | 
						|
		`net.inet.ah`,
 | 
						|
		`net.inet.carp`,
 | 
						|
		`net.inet.divert`,
 | 
						|
		`net.inet.esp`,
 | 
						|
		`net.inet.etherip`,
 | 
						|
		`net.inet.gre`,
 | 
						|
		`net.inet.icmp`,
 | 
						|
		`net.inet.igmp`,
 | 
						|
		`net.inet.ip`,
 | 
						|
		`net.inet.ip.ifq`,
 | 
						|
		`net.inet.ipcomp`,
 | 
						|
		`net.inet.ipip`,
 | 
						|
		`net.inet.mobileip`,
 | 
						|
		`net.inet.pfsync`,
 | 
						|
		`net.inet.pim`,
 | 
						|
		`net.inet.tcp`,
 | 
						|
		`net.inet.udp`,
 | 
						|
		`net.inet6`,
 | 
						|
		`net.inet6.divert`,
 | 
						|
		`net.inet6.ip6`,
 | 
						|
		`net.inet6.icmp6`,
 | 
						|
		`net.inet6.pim6`,
 | 
						|
		`net.inet6.tcp6`,
 | 
						|
		`net.inet6.udp6`,
 | 
						|
		`net.mpls`,
 | 
						|
		`net.mpls.ifq`,
 | 
						|
		`net.key`,
 | 
						|
		`net.pflow`,
 | 
						|
		`net.pfsync`,
 | 
						|
		`net.pipex`,
 | 
						|
		`net.rt`,
 | 
						|
		`vm.swapencrypt`,
 | 
						|
		//vfsgenctl		/* Special handling required */
 | 
						|
	}
 | 
						|
 | 
						|
	// Node name "fixups"
 | 
						|
	ctlMap := map[string]string{
 | 
						|
		"ipproto":             "net.inet",
 | 
						|
		"net.inet.ipproto":    "net.inet",
 | 
						|
		"net.inet6.ipv6proto": "net.inet6",
 | 
						|
		"net.inet6.ipv6":      "net.inet6.ip6",
 | 
						|
		"net.inet.icmpv6":     "net.inet6.icmp6",
 | 
						|
		"net.inet6.divert6":   "net.inet6.divert",
 | 
						|
		"net.inet6.tcp6":      "net.inet.tcp",
 | 
						|
		"net.inet6.udp6":      "net.inet.udp",
 | 
						|
		"mpls":                "net.mpls",
 | 
						|
		"swpenc":              "vm.swapencrypt",
 | 
						|
	}
 | 
						|
 | 
						|
	// Node mappings
 | 
						|
	nodeMap = map[string]string{
 | 
						|
		"net.inet.ip.ifq": "net.ifq",
 | 
						|
		"net.inet.pfsync": "net.pfsync",
 | 
						|
		"net.mpls.ifq":    "net.ifq",
 | 
						|
	}
 | 
						|
 | 
						|
	mCtls := make(map[string]bool)
 | 
						|
	for _, ctl := range ctls {
 | 
						|
		mCtls[ctl] = true
 | 
						|
	}
 | 
						|
 | 
						|
	for _, header := range headers {
 | 
						|
		debug("Processing " + header)
 | 
						|
		file, err := os.Open(filepath.Join("/usr/include", header))
 | 
						|
		if err != nil {
 | 
						|
			fmt.Fprintf(os.Stderr, "%v\n", err)
 | 
						|
			os.Exit(1)
 | 
						|
		}
 | 
						|
		s := bufio.NewScanner(file)
 | 
						|
		for s.Scan() {
 | 
						|
			var sub []string
 | 
						|
			if reMatch(ctlNames1RE, s.Text(), &sub) ||
 | 
						|
				reMatch(ctlNames2RE, s.Text(), &sub) ||
 | 
						|
				reMatch(ctlNames3RE, s.Text(), &sub) {
 | 
						|
				if sub[1] == `CTL_NAMES` {
 | 
						|
					// Top level.
 | 
						|
					node = &mib
 | 
						|
				} else {
 | 
						|
					// Node.
 | 
						|
					nodename := strings.ToLower(sub[2])
 | 
						|
					ctlName := ""
 | 
						|
					if reMatch(netInetRE, header, &sub) {
 | 
						|
						ctlName = "net.inet." + nodename
 | 
						|
					} else if reMatch(netInet6RE, header, &sub) {
 | 
						|
						ctlName = "net.inet6." + nodename
 | 
						|
					} else if reMatch(netRE, header, &sub) {
 | 
						|
						ctlName = "net." + nodename
 | 
						|
					} else {
 | 
						|
						ctlName = nodename
 | 
						|
						ctlName = fsNetKernRE.ReplaceAllString(ctlName, `$1.`)
 | 
						|
					}
 | 
						|
 | 
						|
					if val, ok := ctlMap[ctlName]; ok {
 | 
						|
						ctlName = val
 | 
						|
					}
 | 
						|
					if _, ok := mCtls[ctlName]; !ok {
 | 
						|
						debug("Ignoring " + ctlName + "...")
 | 
						|
						continue
 | 
						|
					}
 | 
						|
 | 
						|
					// Walk down from the top of the MIB.
 | 
						|
					node = &mib
 | 
						|
					for _, part := range strings.Split(ctlName, ".") {
 | 
						|
						if _, ok := (*node)[part]; !ok {
 | 
						|
							debug("Missing node " + part)
 | 
						|
							(*node)[part] = nodeElement{n: 0, t: "", pE: &map[string]nodeElement{}}
 | 
						|
						}
 | 
						|
						node = (*node)[part].pE
 | 
						|
					}
 | 
						|
				}
 | 
						|
 | 
						|
				// Populate current node with entries.
 | 
						|
				i := -1
 | 
						|
				for !strings.HasPrefix(s.Text(), "}") {
 | 
						|
					s.Scan()
 | 
						|
					if reMatch(bracesRE, s.Text(), &sub) {
 | 
						|
						i++
 | 
						|
					}
 | 
						|
					if !reMatch(ctlTypeRE, s.Text(), &sub) {
 | 
						|
						continue
 | 
						|
					}
 | 
						|
					(*node)[sub[1]] = nodeElement{n: i, t: sub[2], pE: &map[string]nodeElement{}}
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
		err = s.Err()
 | 
						|
		if err != nil {
 | 
						|
			fmt.Fprintf(os.Stderr, "%v\n", err)
 | 
						|
			os.Exit(1)
 | 
						|
		}
 | 
						|
		file.Close()
 | 
						|
	}
 | 
						|
	buildSysctl(&mib, "", []int{})
 | 
						|
 | 
						|
	sort.Strings(sysCtl)
 | 
						|
	text := strings.Join(sysCtl, "")
 | 
						|
 | 
						|
	fmt.Printf(srcTemplate, cmdLine(), buildTags(), text)
 | 
						|
}
 | 
						|
 | 
						|
const srcTemplate = `// %s
 | 
						|
// Code generated by the command above; DO NOT EDIT.
 | 
						|
 | 
						|
// +build %s
 | 
						|
 | 
						|
package unix
 | 
						|
 | 
						|
type mibentry struct {
 | 
						|
	ctlname string
 | 
						|
	ctloid []_C_int
 | 
						|
}
 | 
						|
 | 
						|
var sysctlMib = []mibentry {
 | 
						|
%s
 | 
						|
}
 | 
						|
`
 |