taigrr/wtf
1
0
Fork 0
mirror of https://github.com/taigrr/wtf synced 2025-01-18 04:03:14 -08:00
Code Issues Projects Releases Wiki Activity

replace c9s/goprocinfo with shirou/gopsutil to get cross platform cpu info

Browse Source
This commit is contained in:
Nicholas Eden 2018-11-23 22:52:05 -08:00
parent db1917763c
commit 24899daeb4
401 changed files with 204122 additions and 2811 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
Gopkg.lock generated
View File

@ -17,6 +17,14 @@
pruneopts = "UT"
revision = "2aa6f33b730c79971cfc3c742f279195b0abc627"
[[projects]]
digest = "1:e92f5581902c345eb4ceffdcd4a854fb8f73cf436d47d837d1ec98ef1fe0a214"
name = "github.com/StackExchange/wmi"
packages = ["."]
pruneopts = "UT"
revision = "5d049714c4a64225c3c79a7cf7d02f7fb5b96338"
version = "1.0.0"
[[projects]]
branch = "master"
digest = "1:636ac9f696c988f0038afd43592f7a0fff29038588ab1064ba8ba3476bb41091"
@ -81,14 +89,6 @@
pruneopts = "UT"
revision = "5f41b7c9d92de5d74bf32f4486375c7547bc8a3c"
[[projects]]
branch = "master"
digest = "1:080689a7cb710dc8f694005f8e5b3aab861d01e885eaf53fae54431c4635bfe6"
name = "github.com/c9s/goprocinfo"
packages = ["linux"]
pruneopts = "UT"
revision = "0010a05ce49fde7f50669bc7ecda7d41dd6ab824"
[[projects]]
digest = "1:2209584c0f7c9b68c23374e659357ab546e1b70eec2761f03280f69a8fd23d77"
name = "github.com/cenkalti/backoff"
@ -159,6 +159,17 @@
pruneopts = "UT"
revision = "493f3b46b3c20880afc8e04ceeb1c6d5aa3363d7"
[[projects]]
digest = "1:64a5a67c69b70c2420e607a8545d674a23778ed9c3e80607bfd17b77c6c87f6a"
name = "github.com/go-ole/go-ole"
packages = [
".",
"oleutil",
]
pruneopts = "UT"
revision = "a41e3c4b706f6ae8dfbff342b06e40fa4d2d0506"
version = "v1.2.1"
[[projects]]
digest = "1:57fa4c058c21ce25d0b7272518dd746065117abf6cc706158b0d361202024520"
name = "github.com/godbus/dbus"
@ -263,6 +274,18 @@
pruneopts = "UT"
revision = "a7c1880d62d37422830f0a15823b51301ca9f354"
[[projects]]
digest = "1:266b1517f99511a50e97cf782c496eae4226ff94e884ef7e1ed53f423795b310"
name = "github.com/shirou/gopsutil"
packages = [
"cpu",
"internal/common",
"mem",
]
pruneopts = "UT"
revision = "3ec50d2876a36047b2ca39f955ba88fb7a455e92"
version = "v2.18.10"
[[projects]]
branch = "master"
digest = "1:93ff598337cb35c40079a1ba766729c20eb1756e37f5740eac4c8e943f59664d"
@ -336,6 +359,17 @@
pruneopts = "UT"
revision = "9dcd33a902f40452422c2367fefcb95b54f9f8f8"
[[projects]]
branch = "master"
digest = "1:f343f077a5b0bc3a3788b3a04e24dd417e3e25b2acb529c413e212d2c42416ef"
name = "golang.org/x/sys"
packages = [
"unix",
"windows",
]
pruneopts = "UT"
revision = "62eef0e2fa9b2c385f7b2778e763486da6880d37"
[[projects]]
digest = "1:37672ad5821719e2df8509c2edd4ba5ae192463237c73c3a2d24ef8b2bc9e36f"
name = "golang.org/x/text"
@ -402,7 +436,6 @@
"github.com/alecthomas/chroma/styles",
"github.com/andygrunwald/go-gerrit",
"github.com/briandowns/openweathermap",
"github.com/c9s/goprocinfo/linux",
"github.com/darkSasori/todoist",
"github.com/dustin/go-humanize",
"github.com/gdamore/tcell",
@ -412,6 +445,8 @@
"github.com/pkg/profile",
"github.com/radovskyb/watcher",
"github.com/rivo/tview",
"github.com/shirou/gopsutil/cpu",
"github.com/shirou/gopsutil/mem",
"github.com/sticreations/spotigopher/spotigopher",
"github.com/stretchr/testify/assert",
"github.com/xanzy/go-gitlab",

6
Gopkg.toml
View File

@ -100,8 +100,8 @@
[[constraint]]
branch = "master"
name = "github.com/c9s/goprocinfo"
name = "code.cloudfoundry.org/bytefmt"
[[constraint]]
branch = "master"
name = "code.cloudfoundry.org/bytefmt"
name = "github.com/shirou/gopsutil"
version = "2.18.10"

40
resourceusage/widget.go
View File

@ -3,14 +3,15 @@ package resourceusage
import (
"code.cloudfoundry.org/bytefmt"
"fmt"
"github.com/c9s/goprocinfo/linux"
"github.com/rivo/tview"
"github.com/senorprogrammer/wtf/wtf"
"github.com/shirou/gopsutil/cpu"
"github.com/shirou/gopsutil/mem"
)
var started = false
var ok = true
var prevStats []linux.CPUStat
var prevStats []cpu.TimesStat
// Widget define wtf widget to register widget later
type Widget struct {
@ -20,7 +21,7 @@ type Widget struct {
// NewWidget Make new instance of widget
func NewWidget(app *tview.Application) *Widget {
widget := Widget{
BarGraph: wtf.NewBarGraph(app, "Resource Usage", "resourceusage", false),
BarGraph: wtf.NewBarGraph(app, "Resource Usage", "resourceusage", false),
}
widget.View.SetWrap(true)
@ -34,14 +35,14 @@ func NewWidget(app *tview.Application) *Widget {
// MakeGraph - Load the dead drop stats
func MakeGraph(widget *Widget) {
cpuStat, err := linux.ReadStat("/proc/stat")
cpuStats, err := cpu.Times(true)
if err != nil {
return
}
var stats = make([]wtf.Bar, len(cpuStat.CPUStats)+2)
var stats = make([]wtf.Bar, len(cpuStats)+2)
for i, stat := range cpuStat.CPUStats {
for i, stat := range cpuStats {
prevStat := stat
if len(prevStats) > i {
prevStat = prevStats[i]
@ -50,11 +51,11 @@ func MakeGraph(widget *Widget) {
}
// based on htop algorithm described here: https://stackoverflow.com/a/23376195/1516085
prevIdle := prevStat.Idle + prevStat.IOWait
idle := stat.Idle + stat.IOWait
prevIdle := prevStat.Idle + prevStat.Iowait
idle := stat.Idle + stat.Iowait
prevNonIdle := prevStat.User + prevStat.Nice + prevStat.System + prevStat.IRQ + prevStat.SoftIRQ + prevStat.Steal
nonIdle := stat.User + stat.Nice + stat.System + stat.IRQ + stat.SoftIRQ + stat.Steal
prevNonIdle := prevStat.User + prevStat.Nice + prevStat.System + prevStat.Irq + prevStat.Softirq + prevStat.Steal
nonIdle := stat.User + stat.Nice + stat.System + stat.Irq + stat.Softirq + stat.Steal
prevTotal := prevIdle + prevNonIdle
total := idle + nonIdle
@ -78,17 +79,16 @@ func MakeGraph(widget *Widget) {
prevStats[i] = stat
}
memInfo, err := linux.ReadMemInfo("/proc/meminfo")
//memInfo, err := linux.ReadMemInfo("/proc/meminfo")
memInfo, err := mem.VirtualMemory()
if err != nil {
return
}
memIndex := len(cpuStat.CPUStats)
memUsed := memInfo.MemTotal - memInfo.MemAvailable
memPercent := float64(memUsed) / float64(memInfo.MemTotal)
memIndex := len(cpuStats)
usedMemLabel := bytefmt.ByteSize(memUsed * bytefmt.KILOBYTE)
totalMemLabel := bytefmt.ByteSize(memInfo.MemTotal * bytefmt.KILOBYTE)
usedMemLabel := bytefmt.ByteSize(memInfo.Used)
totalMemLabel := bytefmt.ByteSize(memInfo.Total)
if usedMemLabel[len(usedMemLabel)-1] == totalMemLabel[len(totalMemLabel)-1] {
usedMemLabel = usedMemLabel[:len(usedMemLabel)-1]
@ -96,16 +96,16 @@ func MakeGraph(widget *Widget) {
stats[memIndex] = wtf.Bar{
Label: "Mem",
Percent: int(memPercent * 100),
Percent: int(memInfo.UsedPercent),
ValueLabel: fmt.Sprintf("%s/%s", usedMemLabel, totalMemLabel),
}
swapIndex := len(cpuStat.CPUStats) + 1
swapIndex := len(cpuStats) + 1
swapUsed := memInfo.SwapTotal - memInfo.SwapFree
swapPercent := float64(swapUsed) / float64(memInfo.SwapTotal)
usedSwapLabel := bytefmt.ByteSize(swapUsed * bytefmt.KILOBYTE)
totalSwapLabel := bytefmt.ByteSize(memInfo.SwapTotal * bytefmt.KILOBYTE)
usedSwapLabel := bytefmt.ByteSize(swapUsed)
totalSwapLabel := bytefmt.ByteSize(memInfo.SwapTotal)
if usedSwapLabel[len(usedSwapLabel)-1] == totalMemLabel[len(totalSwapLabel)-1] {
usedSwapLabel = usedSwapLabel[:len(usedSwapLabel)-1]

20
vendor/github.com/StackExchange/wmi/LICENSE generated vendored Normal file
View File

@ -0,0 +1,20 @@
The MIT License (MIT)
Copyright (c) 2013 Stack Exchange
Permission is hereby granted, free of charge, to any person obtaining a copy of
this software and associated documentation files (the "Software"), to deal in
the Software without restriction, including without limitation the rights to
use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
the Software, and to permit persons to whom the Software is furnished to do so,
subject to the following conditions:
The above copyright notice and this permission notice shall be included in all
copies or substantial portions of the Software.
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

6
vendor/github.com/StackExchange/wmi/README.md generated vendored Normal file
View File

@ -0,0 +1,6 @@
wmi
===
Package wmi provides a WQL interface to Windows WMI.
Note: It interfaces with WMI on the local machine, therefore it only runs on Windows.

260
vendor/github.com/StackExchange/wmi/swbemservices.go generated vendored Normal file
View File

@ -0,0 +1,260 @@
// +build windows
package wmi
import (
"fmt"
"reflect"
"runtime"
"sync"
"github.com/go-ole/go-ole"
"github.com/go-ole/go-ole/oleutil"
)
// SWbemServices is used to access wmi. See https://msdn.microsoft.com/en-us/library/aa393719(v=vs.85).aspx
type SWbemServices struct {
//TODO: track namespace. Not sure if we can re connect to a different namespace using the same instance
cWMIClient *Client //This could also be an embedded struct, but then we would need to branch on Client vs SWbemServices in the Query method
sWbemLocatorIUnknown *ole.IUnknown
sWbemLocatorIDispatch *ole.IDispatch
queries chan *queryRequest
closeError chan error
lQueryorClose sync.Mutex
}
type queryRequest struct {
query string
dst interface{}
args []interface{}
finished chan error
}
// InitializeSWbemServices will return a new SWbemServices object that can be used to query WMI
func InitializeSWbemServices(c *Client, connectServerArgs ...interface{}) (*SWbemServices, error) {
//fmt.Println("InitializeSWbemServices: Starting")
//TODO: implement connectServerArgs as optional argument for init with connectServer call
s := new(SWbemServices)
s.cWMIClient = c
s.queries = make(chan *queryRequest)
initError := make(chan error)
go s.process(initError)
err, ok := <-initError
if ok {
return nil, err //Send error to caller
}
//fmt.Println("InitializeSWbemServices: Finished")
return s, nil
}
// Close will clear and release all of the SWbemServices resources
func (s *SWbemServices) Close() error {
s.lQueryorClose.Lock()
if s == nil || s.sWbemLocatorIDispatch == nil {
s.lQueryorClose.Unlock()
return fmt.Errorf("SWbemServices is not Initialized")
}
if s.queries == nil {
s.lQueryorClose.Unlock()
return fmt.Errorf("SWbemServices has been closed")
}
//fmt.Println("Close: sending close request")
var result error
ce := make(chan error)
s.closeError = ce //Race condition if multiple callers to close. May need to lock here
close(s.queries) //Tell background to shut things down
s.lQueryorClose.Unlock()
err, ok := <-ce
if ok {
result = err
}
//fmt.Println("Close: finished")
return result
}
func (s *SWbemServices) process(initError chan error) {
//fmt.Println("process: starting background thread initialization")
//All OLE/WMI calls must happen on the same initialized thead, so lock this goroutine
runtime.LockOSThread()
defer runtime.LockOSThread()
err := ole.CoInitializeEx(0, ole.COINIT_MULTITHREADED)
if err != nil {
oleCode := err.(*ole.OleError).Code()
if oleCode != ole.S_OK && oleCode != S_FALSE {
initError <- fmt.Errorf("ole.CoInitializeEx error: %v", err)
return
}
}
defer ole.CoUninitialize()
unknown, err := oleutil.CreateObject("WbemScripting.SWbemLocator")
if err != nil {
initError <- fmt.Errorf("CreateObject SWbemLocator error: %v", err)
return
} else if unknown == nil {
initError <- ErrNilCreateObject
return
}
defer unknown.Release()
s.sWbemLocatorIUnknown = unknown
dispatch, err := s.sWbemLocatorIUnknown.QueryInterface(ole.IID_IDispatch)
if err != nil {
initError <- fmt.Errorf("SWbemLocator QueryInterface error: %v", err)
return
}
defer dispatch.Release()
s.sWbemLocatorIDispatch = dispatch
// we can't do the ConnectServer call outside the loop unless we find a way to track and re-init the connectServerArgs
//fmt.Println("process: initialized. closing initError")
close(initError)
//fmt.Println("process: waiting for queries")
for q := range s.queries {
//fmt.Printf("process: new query: len(query)=%d\n", len(q.query))
errQuery := s.queryBackground(q)
//fmt.Println("process: s.queryBackground finished")
if errQuery != nil {
q.finished <- errQuery
}
close(q.finished)
}
//fmt.Println("process: queries channel closed")
s.queries = nil //set channel to nil so we know it is closed
//TODO: I think the Release/Clear calls can panic if things are in a bad state.
//TODO: May need to recover from panics and send error to method caller instead.
close(s.closeError)
}
// Query runs the WQL query using a SWbemServices instance and appends the values to dst.
//
// dst must have type *[]S or *[]*S, for some struct type S. Fields selected in
// the query must have the same name in dst. Supported types are all signed and
// unsigned integers, time.Time, string, bool, or a pointer to one of those.
// Array types are not supported.
//
// By default, the local machine and default namespace are used. These can be
// changed using connectServerArgs. See
// http://msdn.microsoft.com/en-us/library/aa393720.aspx for details.
func (s *SWbemServices) Query(query string, dst interface{}, connectServerArgs ...interface{}) error {
s.lQueryorClose.Lock()
if s == nil || s.sWbemLocatorIDispatch == nil {
s.lQueryorClose.Unlock()
return fmt.Errorf("SWbemServices is not Initialized")
}
if s.queries == nil {
s.lQueryorClose.Unlock()
return fmt.Errorf("SWbemServices has been closed")
}
//fmt.Println("Query: Sending query request")
qr := queryRequest{
query: query,
dst: dst,
args: connectServerArgs,
finished: make(chan error),
}
s.queries <- &qr
s.lQueryorClose.Unlock()
err, ok := <-qr.finished
if ok {
//fmt.Println("Query: Finished with error")
return err //Send error to caller
}
//fmt.Println("Query: Finished")
return nil
}
func (s *SWbemServices) queryBackground(q *queryRequest) error {
if s == nil || s.sWbemLocatorIDispatch == nil {
return fmt.Errorf("SWbemServices is not Initialized")
}
wmi := s.sWbemLocatorIDispatch //Should just rename in the code, but this will help as we break things apart
//fmt.Println("queryBackground: Starting")
dv := reflect.ValueOf(q.dst)
if dv.Kind() != reflect.Ptr || dv.IsNil() {
return ErrInvalidEntityType
}
dv = dv.Elem()
mat, elemType := checkMultiArg(dv)
if mat == multiArgTypeInvalid {
return ErrInvalidEntityType
}
// service is a SWbemServices
serviceRaw, err := oleutil.CallMethod(wmi, "ConnectServer", q.args...)
if err != nil {
return err
}
service := serviceRaw.ToIDispatch()
defer serviceRaw.Clear()
// result is a SWBemObjectSet
resultRaw, err := oleutil.CallMethod(service, "ExecQuery", q.query)
if err != nil {
return err
}
result := resultRaw.ToIDispatch()
defer resultRaw.Clear()
count, err := oleInt64(result, "Count")
if err != nil {
return err
}
enumProperty, err := result.GetProperty("_NewEnum")
if err != nil {
return err
}
defer enumProperty.Clear()
enum, err := enumProperty.ToIUnknown().IEnumVARIANT(ole.IID_IEnumVariant)
if err != nil {
return err
}
if enum == nil {
return fmt.Errorf("can't get IEnumVARIANT, enum is nil")
}
defer enum.Release()
// Initialize a slice with Count capacity
dv.Set(reflect.MakeSlice(dv.Type(), 0, int(count)))
var errFieldMismatch error
for itemRaw, length, err := enum.Next(1); length > 0; itemRaw, length, err = enum.Next(1) {
if err != nil {
return err
}
err := func() error {
// item is a SWbemObject, but really a Win32_Process
item := itemRaw.ToIDispatch()
defer item.Release()
ev := reflect.New(elemType)
if err = s.cWMIClient.loadEntity(ev.Interface(), item); err != nil {
if _, ok := err.(*ErrFieldMismatch); ok {
// We continue loading entities even in the face of field mismatch errors.
// If we encounter any other error, that other error is returned. Otherwise,
// an ErrFieldMismatch is returned.
errFieldMismatch = err
} else {
return err
}
}
if mat != multiArgTypeStructPtr {
ev = ev.Elem()
}
dv.Set(reflect.Append(dv, ev))
return nil
}()
if err != nil {
return err
}
}
//fmt.Println("queryBackground: Finished")
return errFieldMismatch
}

486
vendor/github.com/StackExchange/wmi/wmi.go generated vendored Normal file
View File

@ -0,0 +1,486 @@
// +build windows
/*
Package wmi provides a WQL interface for WMI on Windows.
Example code to print names of running processes:
type Win32_Process struct {
Name string
}
func main() {
var dst []Win32_Process
q := wmi.CreateQuery(&dst, "")
err := wmi.Query(q, &dst)
if err != nil {
log.Fatal(err)
}
for i, v := range dst {
println(i, v.Name)
}
}
*/
package wmi
import (
"bytes"
"errors"
"fmt"
"log"
"os"
"reflect"
"runtime"
"strconv"
"strings"
"sync"
"time"
"github.com/go-ole/go-ole"
"github.com/go-ole/go-ole/oleutil"
)
var l = log.New(os.Stdout, "", log.LstdFlags)
var (
ErrInvalidEntityType = errors.New("wmi: invalid entity type")
// ErrNilCreateObject is the error returned if CreateObject returns nil even
// if the error was nil.
ErrNilCreateObject = errors.New("wmi: create object returned nil")
lock sync.Mutex
)
// S_FALSE is returned by CoInitializeEx if it was already called on this thread.
const S_FALSE = 0x00000001
// QueryNamespace invokes Query with the given namespace on the local machine.
func QueryNamespace(query string, dst interface{}, namespace string) error {
return Query(query, dst, nil, namespace)
}
// Query runs the WQL query and appends the values to dst.
//
// dst must have type *[]S or *[]*S, for some struct type S. Fields selected in
// the query must have the same name in dst. Supported types are all signed and
// unsigned integers, time.Time, string, bool, or a pointer to one of those.
// Array types are not supported.
//
// By default, the local machine and default namespace are used. These can be
// changed using connectServerArgs. See
// http://msdn.microsoft.com/en-us/library/aa393720.aspx for details.
//
// Query is a wrapper around DefaultClient.Query.
func Query(query string, dst interface{}, connectServerArgs ...interface{}) error {
if DefaultClient.SWbemServicesClient == nil {
return DefaultClient.Query(query, dst, connectServerArgs...)
}
return DefaultClient.SWbemServicesClient.Query(query, dst, connectServerArgs...)
}
// A Client is an WMI query client.
//
// Its zero value (DefaultClient) is a usable client.
type Client struct {
// NonePtrZero specifies if nil values for fields which aren't pointers
// should be returned as the field types zero value.
//
// Setting this to true allows stucts without pointer fields to be used
// without the risk failure should a nil value returned from WMI.
NonePtrZero bool
// PtrNil specifies if nil values for pointer fields should be returned
// as nil.
//
// Setting this to true will set pointer fields to nil where WMI
// returned nil, otherwise the types zero value will be returned.
PtrNil bool
// AllowMissingFields specifies that struct fields not present in the
// query result should not result in an error.
//
// Setting this to true allows custom queries to be used with full
// struct definitions instead of having to define multiple structs.
AllowMissingFields bool
// SWbemServiceClient is an optional SWbemServices object that can be
// initialized and then reused across multiple queries. If it is null
// then the method will initialize a new temporary client each time.
SWbemServicesClient *SWbemServices
}
// DefaultClient is the default Client and is used by Query, QueryNamespace
var DefaultClient = &Client{}
// Query runs the WQL query and appends the values to dst.
//
// dst must have type *[]S or *[]*S, for some struct type S. Fields selected in
// the query must have the same name in dst. Supported types are all signed and
// unsigned integers, time.Time, string, bool, or a pointer to one of those.
// Array types are not supported.
//
// By default, the local machine and default namespace are used. These can be
// changed using connectServerArgs. See
// http://msdn.microsoft.com/en-us/library/aa393720.aspx for details.
func (c *Client) Query(query string, dst interface{}, connectServerArgs ...interface{}) error {
dv := reflect.ValueOf(dst)
if dv.Kind() != reflect.Ptr || dv.IsNil() {
return ErrInvalidEntityType
}
dv = dv.Elem()
mat, elemType := checkMultiArg(dv)
if mat == multiArgTypeInvalid {
return ErrInvalidEntityType
}
lock.Lock()
defer lock.Unlock()
runtime.LockOSThread()
defer runtime.UnlockOSThread()
err := ole.CoInitializeEx(0, ole.COINIT_MULTITHREADED)
if err != nil {
oleCode := err.(*ole.OleError).Code()
if oleCode != ole.S_OK && oleCode != S_FALSE {
return err
}
}
defer ole.CoUninitialize()
unknown, err := oleutil.CreateObject("WbemScripting.SWbemLocator")
if err != nil {
return err
} else if unknown == nil {
return ErrNilCreateObject
}
defer unknown.Release()
wmi, err := unknown.QueryInterface(ole.IID_IDispatch)
if err != nil {
return err
}
defer wmi.Release()
// service is a SWbemServices
serviceRaw, err := oleutil.CallMethod(wmi, "ConnectServer", connectServerArgs...)
if err != nil {
return err
}
service := serviceRaw.ToIDispatch()
defer serviceRaw.Clear()
// result is a SWBemObjectSet
resultRaw, err := oleutil.CallMethod(service, "ExecQuery", query)
if err != nil {
return err
}
result := resultRaw.ToIDispatch()
defer resultRaw.Clear()
count, err := oleInt64(result, "Count")
if err != nil {
return err
}
enumProperty, err := result.GetProperty("_NewEnum")
if err != nil {
return err
}
defer enumProperty.Clear()
enum, err := enumProperty.ToIUnknown().IEnumVARIANT(ole.IID_IEnumVariant)
if err != nil {
return err
}
if enum == nil {
return fmt.Errorf("can't get IEnumVARIANT, enum is nil")
}
defer enum.Release()
// Initialize a slice with Count capacity
dv.Set(reflect.MakeSlice(dv.Type(), 0, int(count)))
var errFieldMismatch error
for itemRaw, length, err := enum.Next(1); length > 0; itemRaw, length, err = enum.Next(1) {
if err != nil {
return err
}
err := func() error {
// item is a SWbemObject, but really a Win32_Process
item := itemRaw.ToIDispatch()
defer item.Release()
ev := reflect.New(elemType)
if err = c.loadEntity(ev.Interface(), item); err != nil {
if _, ok := err.(*ErrFieldMismatch); ok {
// We continue loading entities even in the face of field mismatch errors.
// If we encounter any other error, that other error is returned. Otherwise,
// an ErrFieldMismatch is returned.
errFieldMismatch = err
} else {
return err
}
}
if mat != multiArgTypeStructPtr {
ev = ev.Elem()
}
dv.Set(reflect.Append(dv, ev))
return nil
}()
if err != nil {
return err
}
}
return errFieldMismatch
}
// ErrFieldMismatch is returned when a field is to be loaded into a different
// type than the one it was stored from, or when a field is missing or
// unexported in the destination struct.
// StructType is the type of the struct pointed to by the destination argument.
type ErrFieldMismatch struct {
StructType reflect.Type
FieldName string
Reason string
}
func (e *ErrFieldMismatch) Error() string {
return fmt.Sprintf("wmi: cannot load field %q into a %q: %s",
e.FieldName, e.StructType, e.Reason)
}
var timeType = reflect.TypeOf(time.Time{})
// loadEntity loads a SWbemObject into a struct pointer.
func (c *Client) loadEntity(dst interface{}, src *ole.IDispatch) (errFieldMismatch error) {
v := reflect.ValueOf(dst).Elem()
for i := 0; i < v.NumField(); i++ {
f := v.Field(i)
of := f
isPtr := f.Kind() == reflect.Ptr
if isPtr {
ptr := reflect.New(f.Type().Elem())
f.Set(ptr)
f = f.Elem()
}
n := v.Type().Field(i).Name
if !f.CanSet() {
return &ErrFieldMismatch{
StructType: of.Type(),
FieldName: n,
Reason: "CanSet() is false",
}
}
prop, err := oleutil.GetProperty(src, n)
if err != nil {
if !c.AllowMissingFields {
errFieldMismatch = &ErrFieldMismatch{
StructType: of.Type(),
FieldName: n,
Reason: "no such struct field",
}
}
continue
}
defer prop.Clear()
switch val := prop.Value().(type) {
case int8, int16, int32, int64, int:
v := reflect.ValueOf(val).Int()
switch f.Kind() {
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
f.SetInt(v)
case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64:
f.SetUint(uint64(v))
default:
return &ErrFieldMismatch{
StructType: of.Type(),
FieldName: n,
Reason: "not an integer class",
}
}
case uint8, uint16, uint32, uint64:
v := reflect.ValueOf(val).Uint()
switch f.Kind() {
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
f.SetInt(int64(v))
case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64:
f.SetUint(v)
default:
return &ErrFieldMismatch{
StructType: of.Type(),
FieldName: n,
Reason: "not an integer class",
}
}
case string:
switch f.Kind() {
case reflect.String:
f.SetString(val)
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
iv, err := strconv.ParseInt(val, 10, 64)
if err != nil {
return err
}
f.SetInt(iv)
case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64:
uv, err := strconv.ParseUint(val, 10, 64)
if err != nil {
return err
}
f.SetUint(uv)
case reflect.Struct:
switch f.Type() {
case timeType:
if len(val) == 25 {
mins, err := strconv.Atoi(val[22:])
if err != nil {
return err
}
val = val[:22] + fmt.Sprintf("%02d%02d", mins/60, mins%60)
}
t, err := time.Parse("20060102150405.000000-0700", val)
if err != nil {
return err
}
f.Set(reflect.ValueOf(t))
}
}
case bool:
switch f.Kind() {
case reflect.Bool:
f.SetBool(val)
default:
return &ErrFieldMismatch{
StructType: of.Type(),
FieldName: n,
Reason: "not a bool",
}
}
case float32:
switch f.Kind() {
case reflect.Float32:
f.SetFloat(float64(val))
default:
return &ErrFieldMismatch{
StructType: of.Type(),
FieldName: n,
Reason: "not a Float32",
}
}
default:
if f.Kind() == reflect.Slice {
switch f.Type().Elem().Kind() {
case reflect.String:
safeArray := prop.ToArray()
if safeArray != nil {
arr := safeArray.ToValueArray()
fArr := reflect.MakeSlice(f.Type(), len(arr), len(arr))
for i, v := range arr {
s := fArr.Index(i)
s.SetString(v.(string))
}
f.Set(fArr)
}
case reflect.Uint8:
safeArray := prop.ToArray()
if safeArray != nil {
arr := safeArray.ToValueArray()
fArr := reflect.MakeSlice(f.Type(), len(arr), len(arr))
for i, v := range arr {
s := fArr.Index(i)
s.SetUint(reflect.ValueOf(v).Uint())
}
f.Set(fArr)
}
default:
return &ErrFieldMismatch{
StructType: of.Type(),
FieldName: n,
Reason: fmt.Sprintf("unsupported slice type (%T)", val),
}
}
} else {
typeof := reflect.TypeOf(val)
if typeof == nil && (isPtr || c.NonePtrZero) {
if (isPtr && c.PtrNil) || (!isPtr && c.NonePtrZero) {
of.Set(reflect.Zero(of.Type()))
}
break
}
return &ErrFieldMismatch{
StructType: of.Type(),
FieldName: n,
Reason: fmt.Sprintf("unsupported type (%T)", val),
}
}
}
}
return errFieldMismatch
}
type multiArgType int
const (
multiArgTypeInvalid multiArgType = iota
multiArgTypeStruct
multiArgTypeStructPtr
)
// checkMultiArg checks that v has type []S, []*S for some struct type S.
//
// It returns what category the slice's elements are, and the reflect.Type
// that represents S.
func checkMultiArg(v reflect.Value) (m multiArgType, elemType reflect.Type) {
if v.Kind() != reflect.Slice {
return multiArgTypeInvalid, nil
}
elemType = v.Type().Elem()
switch elemType.Kind() {
case reflect.Struct:
return multiArgTypeStruct, elemType
case reflect.Ptr:
elemType = elemType.Elem()
if elemType.Kind() == reflect.Struct {
return multiArgTypeStructPtr, elemType
}
}
return multiArgTypeInvalid, nil
}
func oleInt64(item *ole.IDispatch, prop string) (int64, error) {
v, err := oleutil.GetProperty(item, prop)
if err != nil {
return 0, err
}
defer v.Clear()
i := int64(v.Val)
return i, nil
}
// CreateQuery returns a WQL query string that queries all columns of src. where
// is an optional string that is appended to the query, to be used with WHERE
// clauses. In such a case, the "WHERE" string should appear at the beginning.
func CreateQuery(src interface{}, where string) string {
var b bytes.Buffer
b.WriteString("SELECT ")
s := reflect.Indirect(reflect.ValueOf(src))
t := s.Type()
if s.Kind() == reflect.Slice {
t = t.Elem()
}
if t.Kind() != reflect.Struct {
return ""
}
var fields []string
for i := 0; i < t.NumField(); i++ {
fields = append(fields, t.Field(i).Name)
}
b.WriteString(strings.Join(fields, ", "))
b.WriteString(" FROM ")
b.WriteString(t.Name())
b.WriteString(" " + where)
return b.String()
}

21
vendor/github.com/c9s/goprocinfo/LICENSE generated vendored
View File

@ -1,21 +0,0 @@
The MIT License (MIT)
Copyright (c) 2013-2014 Yo-An Lin
Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files (the "Software"), to deal
in the Software without restriction, including without limitation the rights
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
copies of the Software, and to permit persons to whom the Software is
furnished to do so, subject to the following conditions:
The above copyright notice and this permission notice shall be included in
all copies or substantial portions of the Software.
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
THE SOFTWARE.

133
vendor/github.com/c9s/goprocinfo/linux/cpuinfo.go generated vendored
View File

@ -1,133 +0,0 @@
package linux
import (
"io/ioutil"
"regexp"
"strconv"
"strings"
)
type CPUInfo struct {
Processors []Processor `json:"processors"`
}
func (self *CPUInfo) NumCPU() int {
return len(self.Processors)
}
func (self *CPUInfo) NumCore() int {
core := make(map[string]bool)
for _, p := range self.Processors {
pid := p.PhysicalId
cid := p.CoreId
if pid == -1 {
return self.NumCPU()
} else {
// to avoid fmt import
key := strconv.FormatInt(int64(pid), 10) + ":" + strconv.FormatInt(int64(cid), 10)
core[key] = true
}
}
return len(core)
}
func (self *CPUInfo) NumPhysicalCPU() int {
pcpu := make(map[string]bool)
for _, p := range self.Processors {
pid := p.PhysicalId
if pid == -1 {
return self.NumCPU()
} else {
// to avoid fmt import
key := strconv.FormatInt(int64(pid), 10)
pcpu[key] = true
}
}
return len(pcpu)
}
type Processor struct {
Id int64 `json:"id"`
VendorId string `json:"vendor_id"`
Model int64 `json:"model"`
ModelName string `json:"model_name"`
Flags []string `json:"flags"`
Cores int64 `json:"cores"`
MHz float64 `json:"mhz"`
CacheSize int64 `json:"cache_size"` // KB
PhysicalId int64 `json:"physical_id"`
CoreId int64 `json:"core_id"`
}
var cpuinfoRegExp = regexp.MustCompile("([^:]*?)\\s*:\\s*(.*)$")
func ReadCPUInfo(path string) (*CPUInfo, error) {
b, err := ioutil.ReadFile(path)
if err != nil {
return nil, err
}
content := string(b)
lines := strings.Split(content, "\n")
var cpuinfo = CPUInfo{}
var processor = &Processor{CoreId: -1, PhysicalId: -1}
for i, line := range lines {
var key string
var value string
if len(line) == 0 && i != len(lines)-1 {
// end of processor
cpuinfo.Processors = append(cpuinfo.Processors, *processor)
processor = &Processor{}
continue
} else if i == len(lines)-1 {
continue
}
submatches := cpuinfoRegExp.FindStringSubmatch(line)
key = submatches[1]
value = submatches[2]
switch key {
case "processor":
processor.Id, _ = strconv.ParseInt(value, 10, 64)
case "vendor_id":
processor.VendorId = value
case "model":
processor.Model, _ = strconv.ParseInt(value, 10, 64)
case "model name":
processor.ModelName = value
case "flags":
processor.Flags = strings.Fields(value)
case "cpu cores":
processor.Cores, _ = strconv.ParseInt(value, 10, 64)
case "cpu MHz":
processor.MHz, _ = strconv.ParseFloat(value, 64)
case "cache size":
processor.CacheSize, _ = strconv.ParseInt(value[:strings.IndexAny(value, " \t\n")], 10, 64)
if strings.HasSuffix(line, "MB") {
processor.CacheSize *= 1024
}
case "physical id":
processor.PhysicalId, _ = strconv.ParseInt(value, 10, 64)
case "core id":
processor.CoreId, _ = strconv.ParseInt(value, 10, 64)
}
/*
processor : 0
vendor_id : GenuineIntel
cpu family : 6
model : 26
model name : Intel(R) Xeon(R) CPU L5520 @ 2.27GHz
*/
}
return &cpuinfo, nil
}

26
vendor/github.com/c9s/goprocinfo/linux/disk.go generated vendored
View File

@ -1,26 +0,0 @@
package linux
import (
"syscall"
)
type Disk struct {
All uint64 `json:"all"`
Used uint64 `json:"used"`
Free uint64 `json:"free"`
FreeInodes uint64 `json:"freeInodes"`
}
func ReadDisk(path string) (*Disk, error) {
fs := syscall.Statfs_t{}
err := syscall.Statfs(path, &fs)
if err != nil {
return nil, err
}
disk := Disk{}
disk.All = fs.Blocks * uint64(fs.Bsize)
disk.Free = fs.Bfree * uint64(fs.Bsize)
disk.Used = disk.All - disk.Free
disk.FreeInodes = fs.Ffree
return &disk, nil
}

100
vendor/github.com/c9s/goprocinfo/linux/diskstat.go generated vendored
View File

@ -1,100 +0,0 @@
package linux
import (
"io/ioutil"
"strconv"
"strings"
"time"
)
// DiskStat is disk statistics to help measure disk activity.
//
// Note:
// * On a very busy or long-lived system values may wrap.
// * No kernel locks are held while modifying these counters. This implies that
// minor inaccuracies may occur.
//
// More more info see:
// https://www.kernel.org/doc/Documentation/iostats.txt and
// https://www.kernel.org/doc/Documentation/block/stat.txt
type DiskStat struct {
Major int `json:"major"` // major device number
Minor int `json:"minor"` // minor device number
Name string `json:"name"` // device name
ReadIOs uint64 `json:"read_ios"` // number of read I/Os processed
ReadMerges uint64 `json:"read_merges"` // number of read I/Os merged with in-queue I/O
ReadSectors uint64 `json:"read_sectors"` // number of 512 byte sectors read
ReadTicks uint64 `json:"read_ticks"` // total wait time for read requests in milliseconds
WriteIOs uint64 `json:"write_ios"` // number of write I/Os processed
WriteMerges uint64 `json:"write_merges"` // number of write I/Os merged with in-queue I/O
WriteSectors uint64 `json:"write_sectors"` // number of 512 byte sectors written
WriteTicks uint64 `json:"write_ticks"` // total wait time for write requests in milliseconds
InFlight uint64 `json:"in_flight"` // number of I/Os currently in flight
IOTicks uint64 `json:"io_ticks"` // total time this block device has been active in milliseconds
TimeInQueue uint64 `json:"time_in_queue"` // total wait time for all requests in milliseconds
}
// ReadDiskStats reads and parses the file.
//
// Note:
// * Assumes a well formed file and will panic if it isn't.
func ReadDiskStats(path string) ([]DiskStat, error) {
data, err := ioutil.ReadFile(path)
if err != nil {
return nil, err
}
devices := strings.Split(string(data), "\n")
results := make([]DiskStat, len(devices)-1)
for i := range results {
fields := strings.Fields(devices[i])
Major, _ := strconv.ParseInt(fields[0], 10, strconv.IntSize)
results[i].Major = int(Major)
Minor, _ := strconv.ParseInt(fields[1], 10, strconv.IntSize)
results[i].Minor = int(Minor)
results[i].Name = fields[2]
results[i].ReadIOs, _ = strconv.ParseUint(fields[3], 10, 64)
results[i].ReadMerges, _ = strconv.ParseUint(fields[4], 10, 64)
results[i].ReadSectors, _ = strconv.ParseUint(fields[5], 10, 64)
results[i].ReadTicks, _ = strconv.ParseUint(fields[6], 10, 64)
results[i].WriteIOs, _ = strconv.ParseUint(fields[7], 10, 64)
results[i].WriteMerges, _ = strconv.ParseUint(fields[8], 10, 64)
results[i].WriteSectors, _ = strconv.ParseUint(fields[9], 10, 64)
results[i].WriteTicks, _ = strconv.ParseUint(fields[10], 10, 64)
results[i].InFlight, _ = strconv.ParseUint(fields[11], 10, 64)
results[i].IOTicks, _ = strconv.ParseUint(fields[12], 10, 64)
results[i].TimeInQueue, _ = strconv.ParseUint(fields[13], 10, 64)
}
return results, nil
}
// GetReadBytes returns the number of bytes read.
func (ds *DiskStat) GetReadBytes() int64 {
return int64(ds.ReadSectors) * 512
}
// GetReadTicks returns the duration waited for read requests.
func (ds *DiskStat) GetReadTicks() time.Duration {
return time.Duration(ds.ReadTicks) * time.Millisecond
}
// GetWriteBytes returns the number of bytes written.
func (ds *DiskStat) GetWriteBytes() int64 {
return int64(ds.WriteSectors) * 512
}
// GetReadTicks returns the duration waited for write requests.
func (ds *DiskStat) GetWriteTicks() time.Duration {
return time.Duration(ds.WriteTicks) * time.Millisecond
}
// GetIOTicks returns the duration the disk has been active.
func (ds *DiskStat) GetIOTicks() time.Duration {
return time.Duration(ds.IOTicks) * time.Millisecond
}
// GetTimeInQueue returns the duration waited for all requests.
func (ds *DiskStat) GetTimeInQueue() time.Duration {
return time.Duration(ds.TimeInQueue) * time.Millisecond
}

56
vendor/github.com/c9s/goprocinfo/linux/interrupts.go generated vendored
View File

@ -1,56 +0,0 @@
package linux
import (
"io/ioutil"
"strconv"
"strings"
)
type Interrupt struct {
Name string
Counts []uint64
Description string
}
type Interrupts struct {
Interrupts []Interrupt
}
func ReadInterrupts(path string) (*Interrupts, error) {
b, err := ioutil.ReadFile(path)
if err != nil {
return nil, err
}
content := string(b)
lines := strings.Split(content, "\n")
cpus := lines[0]
lines = append(lines[:0], lines[1:]...)
numCpus := len(strings.Fields(cpus))
interrupts := make([]Interrupt, 0)
for _, line := range lines {
fields := strings.Fields(line)
if len(fields) == 0 {
continue
}
counts := make([]uint64, 0)
i := 0
for ; i < numCpus; i++ {
if len(fields) <= i+1 {
break
}
count, err := strconv.ParseInt(fields[i+1], 10, 64)
if err != nil {
return nil, err
}
counts = append(counts, uint64(count))
}
name := strings.TrimSuffix(fields[0], ":")
description := strings.Join(fields[i+1:], " ")
interrupts = append(interrupts, Interrupt{
Name: name,
Counts: counts,
Description: description,
})
}
return &Interrupts{Interrupts: interrupts}, nil
}

67
vendor/github.com/c9s/goprocinfo/linux/loadavg.go generated vendored
View File

@ -1,67 +0,0 @@
package linux
import (
"errors"
"io/ioutil"
"strconv"
"strings"
)
type LoadAvg struct {
Last1Min float64 `json:"last1min"`
Last5Min float64 `json:"last5min"`
Last15Min float64 `json:"last15min"`
ProcessRunning uint64 `json:"process_running"`
ProcessTotal uint64 `json:"process_total"`
LastPID uint64 `json:"last_pid"`
}
func ReadLoadAvg(path string) (*LoadAvg, error) {
b, err := ioutil.ReadFile(path)
if err != nil {
return nil, err
}
content := strings.TrimSpace(string(b))
fields := strings.Fields(content)
if len(fields) < 5 {
return nil, errors.New("Cannot parse loadavg: " + content)
}
process := strings.Split(fields[3], "/")
if len(process) != 2 {
return nil, errors.New("Cannot parse loadavg: " + content)
}
loadavg := LoadAvg{}
if loadavg.Last1Min, err = strconv.ParseFloat(fields[0], 64); err != nil {
return nil, err
}
if loadavg.Last5Min, err = strconv.ParseFloat(fields[1], 64); err != nil {
return nil, err
}
if loadavg.Last15Min, err = strconv.ParseFloat(fields[2], 64); err != nil {
return nil, err
}
if loadavg.ProcessRunning, err = strconv.ParseUint(process[0], 10, 64); err != nil {
return nil, err
}
if loadavg.ProcessTotal, err = strconv.ParseUint(process[1], 10, 64); err != nil {
return nil, err
}
if loadavg.LastPID, err = strconv.ParseUint(fields[4], 10, 64); err != nil {
return nil, err
}
return &loadavg, nil
}

97
vendor/github.com/c9s/goprocinfo/linux/meminfo.go generated vendored
View File

@ -1,97 +0,0 @@
package linux
import (
"io/ioutil"
"reflect"
"strconv"
"strings"
)
type MemInfo struct {
MemTotal uint64 `json:"mem_total"`
MemFree uint64 `json:"mem_free"`
MemAvailable uint64 `json:"mem_available"`
Buffers uint64 `json:"buffers"`
Cached uint64 `json:"cached"`
SwapCached uint64 `json:"swap_cached"`
Active uint64 `json:"active"`
Inactive uint64 `json:"inactive"`
ActiveAnon uint64 `json:"active_anon" field:"Active(anon)"`
InactiveAnon uint64 `json:"inactive_anon" field:"Inactive(anon)"`
ActiveFile uint64 `json:"active_file" field:"Active(file)"`
InactiveFile uint64 `json:"inactive_file" field:"Inactive(file)"`
Unevictable uint64 `json:"unevictable"`
Mlocked uint64 `json:"mlocked"`
SwapTotal uint64 `json:"swap_total"`
SwapFree uint64 `json:"swap_free"`
Dirty uint64 `json:"dirty"`
Writeback uint64 `json:"write_back"`
AnonPages uint64 `json:"anon_pages"`
Mapped uint64 `json:"mapped"`
Shmem uint64 `json:"shmem"`
Slab uint64 `json:"slab"`
SReclaimable uint64 `json:"s_reclaimable"`
SUnreclaim uint64 `json:"s_unclaim"`
KernelStack uint64 `json:"kernel_stack"`
PageTables uint64 `json:"page_tables"`
NFS_Unstable uint64 `json:"nfs_unstable"`
Bounce uint64 `json:"bounce"`
WritebackTmp uint64 `json:"writeback_tmp"`
CommitLimit uint64 `json:"commit_limit"`
Committed_AS uint64 `json:"committed_as"`
VmallocTotal uint64 `json:"vmalloc_total"`
VmallocUsed uint64 `json:"vmalloc_used"`
VmallocChunk uint64 `json:"vmalloc_chunk"`
HardwareCorrupted uint64 `json:"hardware_corrupted"`
AnonHugePages uint64 `json:"anon_huge_pages"`
HugePages_Total uint64 `json:"huge_pages_total"`
HugePages_Free uint64 `json:"huge_pages_free"`
HugePages_Rsvd uint64 `json:"huge_pages_rsvd"`
HugePages_Surp uint64 `json:"huge_pages_surp"`
Hugepagesize uint64 `json:"hugepagesize"`
DirectMap4k uint64 `json:"direct_map_4k"`
DirectMap2M uint64 `json:"direct_map_2M"`
DirectMap1G uint64 `json:"direct_map_1G"`
}
func ReadMemInfo(path string) (*MemInfo, error) {
data, err := ioutil.ReadFile(path)
if err != nil {
return nil, err
}
lines := strings.Split(string(data), "\n")
// Maps a meminfo metric to its value (i.e. MemTotal --> 100000)
statMap := make(map[string]uint64)
var info = MemInfo{}
for _, line := range lines {
fields := strings.SplitN(line, ":", 2)
if len(fields) < 2 {
continue
}
valFields := strings.Fields(fields[1])
val, _ := strconv.ParseUint(valFields[0], 10, 64)
statMap[fields[0]] = val
}
elem := reflect.ValueOf(&info).Elem()
typeOfElem := elem.Type()
for i := 0; i < elem.NumField(); i++ {
val, ok := statMap[typeOfElem.Field(i).Name]
if ok {
elem.Field(i).SetUint(val)
continue
}
val, ok = statMap[typeOfElem.Field(i).Tag.Get("field")]
if ok {
elem.Field(i).SetUint(val)
}
}
return &info, nil
}

49
vendor/github.com/c9s/goprocinfo/linux/mounts.go generated vendored
View File

@ -1,49 +0,0 @@
package linux
import (
"bufio"
"os"
"strings"
)
type Mounts struct {
Mounts []Mount `json:"mounts"`
}
type Mount struct {
Device string `json:"device"`
MountPoint string `json:"mountpoint"`
FSType string `json:"fstype"`
Options string `json:"options"`
}
const (
DefaultBufferSize = 1024
)
func ReadMounts(path string) (*Mounts, error) {
fin, err := os.Open(path)
if err != nil {
return nil, err
}
defer fin.Close()
var mounts = Mounts{}
scanner := bufio.NewScanner(fin)
for scanner.Scan() {
fields := strings.Fields(scanner.Text())
var mount = &Mount{
fields[0],
fields[1],
fields[2],
fields[3],
}
mounts.Mounts = append(mounts.Mounts, *mount)
}
if err := scanner.Err(); err != nil {
return nil, err
}
return &mounts, nil
}

173
vendor/github.com/c9s/goprocinfo/linux/net_ip.go generated vendored
View File

@ -1,173 +0,0 @@
package linux
import (
"errors"
"net"
"regexp"
"strconv"
"strings"
)
var (
ipv4RegExp = regexp.MustCompile("^[0-9a-fA-F]{8}:[0-9a-fA-F]{4}$") // Regex for NetIPv4Decoder
ipv6RegExp = regexp.MustCompile("^[0-9a-fA-F]{32}:[0-9a-fA-F]{4}$") // Regex for NetIPv6Decoder
)
type NetIPDecoder func(string) (string, error) // Either NetIPv4Decoder or NetIPv6Decoder
type NetSocket struct {
LocalAddress string `json:"local_address"`
RemoteAddress string `json:"remote_address"`
Status uint8 `json:"st"`
TxQueue uint64 `json:"tx_queue"`
RxQueue uint64 `json:"rx_queue"`
Uid uint32 `json:"uid"`
Inode uint64 `json:"inode"`
SocketReferenceCount uint64 `json:"ref"`
}
func parseNetSocket(f []string, ip NetIPDecoder) (*NetSocket, error) {
if len(f) < 11 {
return nil, errors.New("Cannot parse net socket line: " + strings.Join(f, " "))
}
if strings.Index(f[4], ":") == -1 {
return nil, errors.New("Cannot parse tx/rx queues: " + f[4])
}
q := strings.Split(f[4], ":")
socket := &NetSocket{}
var s uint64 // socket.Status
var u uint64 // socket.Uid
var err error // parse error
if socket.LocalAddress, err = ip(f[1]); err != nil {
return nil, err
}
if socket.RemoteAddress, err = ip(f[2]); err != nil {
return nil, err
}
if s, err = strconv.ParseUint(f[3], 16, 8); err != nil {
return nil, err
}
if socket.TxQueue, err = strconv.ParseUint(q[0], 16, 64); err != nil {
return nil, err
}
if socket.RxQueue, err = strconv.ParseUint(q[1], 16, 64); err != nil {
return nil, err
}
if u, err = strconv.ParseUint(f[7], 10, 32); err != nil {
return nil, err
}
if socket.Inode, err = strconv.ParseUint(f[9], 10, 64); err != nil {
return nil, err
}
if socket.SocketReferenceCount, err = strconv.ParseUint(f[10], 10, 64); err != nil {
return nil, err
}
socket.Status = uint8(s)
socket.Uid = uint32(u)
return socket, nil
}
// Decode an IPv4 address with port from a given hex string
// NOTE: This function match NetIPDecoder type
func NetIPv4Decoder(s string) (string, error) {
if !ipv4RegExp.MatchString(s) {
return "", errors.New("Cannot decode ipv4 address: " + s)
}
i := strings.Split(s, ":")
b := make([]byte, 4)
for j := 0; j < 4; j++ {
x := j * 2
y := x + 2
z := 3 - j
// Extract 2 characters from hex string, 4 times.
//
// s: "0100007F" -> [
// h: "01", h: "00", h: "00", h: "7F",
// ]
h := i[0][x:y]
// Reverse byte order
n, _ := strconv.ParseUint(h, 16, 8)
b[z] = byte(n)
}
h := net.IP(b).String()
n, _ := strconv.ParseUint(i[1], 16, 64)
p := strconv.FormatUint(n, 10)
// ipv4:port
v := h + ":" + p
return v, nil
}
// Decode an IPv6 address with port from a given hex string
// NOTE: This function match NetIPDecoder type
func NetIPv6Decoder(s string) (string, error) {
if !ipv6RegExp.MatchString(s) {
return "", errors.New("Cannot decode ipv6 address: " + s)
}
i := strings.Split(s, ":")
b := make([]byte, 16)
for j := 0; j < 4; j++ {
x := j * 8
y := x + 8
// Extract 8 characters from hex string, 4 times.
//
// s: "350E012A900F122E85EDEAADA64DAAD1" -> [
// h: "350E012A", h: "900F122E",
// h: "85EDEAAD", h: "A64DAAD1",
// ]
h := i[0][x:y]
for k := 0; k < 4; k++ {
// Reverse byte order
// "350E012A" -> [ 0x2A, 0x01, 0x0E, 0x35 ]
z := (j * 4) + k
g := 8 - (k * 2)
f := g - 2
n, _ := strconv.ParseUint(h[f:g], 16, 8)
b[z] = byte(n)
}
}
h := net.IP(b).String()
n, _ := strconv.ParseUint(i[1], 16, 64)
p := strconv.FormatUint(n, 10)
// ipv6:port
v := h + ":" + p
return v, nil
}

82
vendor/github.com/c9s/goprocinfo/linux/net_tcp.go generated vendored
View File

@ -1,82 +0,0 @@
package linux
import (
"io/ioutil"
"strconv"
"strings"
)
type NetTCPSockets struct {
Sockets []NetTCPSocket `json:"sockets"`
}
type NetTCPSocket struct {
NetSocket
RetransmitTimeout uint64 `json:"retransmit_timeout"`
PredictedTick uint64 `json:"predicted_tick"`
AckQuick uint8 `json:"ack_quick"`
AckPingpong bool `json:"ack_pingpong"`
SendingCongestionWindow uint64 `json:"sending_congestion_window"`
SlowStartSizeThreshold int64 `json:"slow_start_size_threshold"`
}
func ReadNetTCPSockets(path string, ip NetIPDecoder) (*NetTCPSockets, error) {
b, err := ioutil.ReadFile(path)
if err != nil {
return nil, err
}
lines := strings.Split(string(b), "\n")
tcp := &NetTCPSockets{}
for i := 1; i < len(lines); i++ {
line := lines[i]
f := strings.Fields(line)
if len(f) < 17 {
continue
}
s, err := parseNetSocket(f, ip)
if err != nil {
return nil, err
}
var n int64
e := &NetTCPSocket{
NetSocket: *s,
}
if e.RetransmitTimeout, err = strconv.ParseUint(f[12], 10, 64); err != nil {
return nil, err
}
if e.PredictedTick, err = strconv.ParseUint(f[13], 10, 64); err != nil {
return nil, err
}
if n, err = strconv.ParseInt(f[14], 10, 8); err != nil {
return nil, err
}
e.AckQuick = uint8(n >> 1)
e.AckPingpong = ((n & 1) == 1)
if e.SendingCongestionWindow, err = strconv.ParseUint(f[15], 10, 64); err != nil {
return nil, err
}
if e.SlowStartSizeThreshold, err = strconv.ParseInt(f[16], 10, 32); err != nil {
return nil, err
}
tcp.Sockets = append(tcp.Sockets, *e)
}
return tcp, nil
}

59
vendor/github.com/c9s/goprocinfo/linux/net_udp.go generated vendored
View File

@ -1,59 +0,0 @@
package linux
import (
"io/ioutil"
"strconv"
"strings"
)
type NetUDPSockets struct {
Sockets []NetUDPSocket `json:"sockets"`
}
type NetUDPSocket struct {
NetSocket
Drops uint64 `json:"drops"`
}
func ReadNetUDPSockets(path string, ip NetIPDecoder) (*NetUDPSockets, error) {
b, err := ioutil.ReadFile(path)
if err != nil {
return nil, err
}
lines := strings.Split(string(b), "\n")
udp := &NetUDPSockets{}
for i := 1; i < len(lines); i++ {
line := lines[i]
f := strings.Fields(line)
if len(f) < 13 {
continue
}
s, err := parseNetSocket(f, ip)
if err != nil {
return nil, err
}
e := &NetUDPSocket{
NetSocket: *s,
Drops: 0,
}
if e.Drops, err = strconv.ParseUint(f[12], 10, 64); err != nil {
return nil, err
}
udp.Sockets = append(udp.Sockets, *e)
}
return udp, nil
}

174
vendor/github.com/c9s/goprocinfo/linux/netstat.go generated vendored
View File

@ -1,174 +0,0 @@
package linux
import (
"io/ioutil"
"reflect"
"strconv"
"strings"
)
type NetStat struct {
// TcpExt
SyncookiesSent uint64 `json:"syncookie_sent"`
SyncookiesRecv uint64 `json:"syncookies_recv"`
SyncookiesFailed uint64 `json:"syncookies_failed"`
EmbryonicRsts uint64 `json:"embryonic_rsts"`
PruneCalled uint64 `json:"prune_called"`
RcvPruned uint64 `json:"rcv_pruned"`
OfoPruned uint64 `json:"ofo_pruned"`
OutOfWindowIcmps uint64 `json:"out_of_window_icmps"`
LockDroppedIcmps uint64 `json:"lock_dropped_icmps"`
ArpFilter uint64 `json:"arp_filter"`
TW uint64 `json:"tw"`
TWRecycled uint64 `json:"tw_recycled"`
TWKilled uint64 `json:"tw_killed"`
PAWSPassive uint64 `json:"paws_passive"`
PAWSActive uint64 `json:"paws_active"`
PAWSEstab uint64 `json:"paws_estab"`
DelayedACKs uint64 `json:"delayed_acks"`
DelayedACKLocked uint64 `json:"delayed_ack_locked"`
DelayedACKLost uint64 `json:"delayed_ack_lost"`
ListenOverflows uint64 `json:"listen_overflows"`
ListenDrops uint64 `json:"listen_drops"`
TCPPrequeued uint64 `json:"tcp_prequeued"`
TCPDirectCopyFromBacklog uint64 `json:"tcp_direct_copy_from_backlog"`
TCPDirectCopyFromPrequeue uint64 `json:"tcp_direct_copy_from_prequeue"`
TCPPrequeueDropped uint64 `json:"tcp_prequeue_dropped"`
TCPHPHits uint64 `json:"tcp_hp_hits"`
TCPHPHitsToUser uint64 `json:"tcp_hp_hits_to_user"`
TCPPureAcks uint64 `json:"tcp_pure_acks"`
TCPHPAcks uint64 `json:"tcp_hp_acks"`
TCPRenoRecovery uint64 `json:"tcp_reno_recovery"`
TCPSackRecovery uint64 `json:"tcp_sack_recovery"`
TCPSACKReneging uint64 `json:"tcp_sack_reneging"`
TCPFACKReorder uint64 `json:"tcp_fack_reorder"`
TCPSACKReorder uint64 `json:"tcp_sack_reorder"`
TCPRenoReorder uint64 `json:"tcp_reno_reorder"`
TCPTSReorder uint64 `json:"tcp_ts_reorder"`
TCPFullUndo uint64 `json:"tcp_full_undo"`
TCPPartialUndo uint64 `json:"tcp_partial_undo"`
TCPDSACKUndo uint64 `json:"tcp_dsack_undo"`
TCPLossUndo uint64 `json:"tcp_loss_undo"`
TCPLoss uint64 `json:"tcp_loss"`
TCPLostRetransmit uint64 `json:"tcp_lost_retransmit"`
TCPRenoFailures uint64 `json:"tcp_reno_failures"`
TCPSackFailures uint64 `json:"tcp_sack_failures"`
TCPLossFailures uint64 `json:"tcp_loss_failures"`
TCPFastRetrans uint64 `json:"tcp_fast_retrans"`
TCPForwardRetrans uint64 `json:"tcp_forward_retrans"`
TCPSlowStartRetrans uint64 `json:"tcp_slow_start_retrans"`
TCPTimeouts uint64 `json:"tcp_timeouts"`
TCPLossProbes uint64 `json:"tcp_loss_probes"`
TCPLossProbeRecovery uint64 `json:"tcp_loss_probe_recovery"`
TCPRenoRecoveryFail uint64 `json:"tcp_reno_recovery_fail"`
TCPSackRecoveryFail uint64 `json:"tcp_sack_recovery_fail"`
TCPSchedulerFailed uint64 `json:"tcp_scheduler_failed"`
TCPRcvCollapsed uint64 `json:"tcp_rcv_collapsed"`
TCPDSACKOldSent uint64 `json:"tcp_dsack_old_sent"`
TCPDSACKOfoSent uint64 `json:"tcp_dsack_ofo_sent"`
TCPDSACKRecv uint64 `json:"tcp_dsack_recv"`
TCPDSACKOfoRecv uint64 `json:"tcp_dsack_ofo_recv"`
TCPAbortOnSyn uint64 `json:"tcp_abort_on_syn"`
TCPAbortOnData uint64 `json:"tcp_abort_on_data"`
TCPAbortOnClose uint64 `json:"tcp_abort_on_close"`
TCPAbortOnMemory uint64 `json:"tcp_abort_on_memory"`
TCPAbortOnTimeout uint64 `json:"tcp_abort_on_timeout"`
TCPAbortOnLinger uint64 `json:"tcp_abort_on_linger"`
TCPAbortFailed uint64 `json:"tcp_abort_failed"`
TCPMemoryPressures uint64 `json:"tcp_memory_pressures"`
TCPSACKDiscard uint64 `json:"tcp_sack_discard"`
TCPDSACKIgnoredOld uint64 `json:"tcp_dsack_ignored_old"`
TCPDSACKIgnoredNoUndo uint64 `json:"tcp_dsack_ignored_no_undo"`
TCPSpuriousRTOs uint64 `json:"tcp_spurious_rtos"`
TCPMD5NotFound uint64 `json:"tcp_md5_not_found"`
TCPMD5Unexpected uint64 `json:"tcp_md5_unexpected"`
TCPSackShifted uint64 `json:"tcp_sack_shifted"`
TCPSackMerged uint64 `json:"tcp_sack_merged"`
TCPSackShiftFallback uint64 `json:"tcp_sack_shift_fallback"`
TCPBacklogDrop uint64 `json:"tcp_backlog_drop"`
TCPMinTTLDrop uint64 `json:"tcp_min_ttl_drop"`
TCPDeferAcceptDrop uint64 `json:"tcp_defer_accept_drop"`
IPReversePathFilter uint64 `json:"ip_reverse_path_filter"`
TCPTimeWaitOverflow uint64 `json:"tcp_time_wait_overflow"`
TCPReqQFullDoCookies uint64 `json:"tcp_req_q_full_do_cookies"`
TCPReqQFullDrop uint64 `json:"tcp_req_q_full_drop"`
TCPRetransFail uint64 `json:"tcp_retrans_fail"`
TCPRcvCoalesce uint64 `json:"tcp_rcv_coalesce"`
TCPOFOQueue uint64 `json:"tcp_ofo_drop"`
TCPOFODrop uint64 `json:"tcp_ofo_drop"`
TCPOFOMerge uint64 `json:"tcp_ofo_merge"`
TCPChallengeACK uint64 `json:"tcp_challenge_ack"`
TCPSYNChallenge uint64 `json:"tcp_syn_challenge"`
TCPFastOpenActive uint64 `json:"tcp_fast_open_active"`
TCPFastOpenActiveFail uint64 `json:"tcp_fast_open_active_fail"`
TCPFastOpenPassive uint64 `json:"tcp_fast_open_passive"`
TCPFastOpenPassiveFail uint64 `json:"tcp_fast_open_passive_fail"`
TCPFastOpenListenOverflow uint64 `json:"tcp_fast_open_listen_overflow"`
TCPFastOpenCookieReqd uint64 `json:"tcp_fast_open_cookie_reqd"`
TCPSpuriousRtxHostQueues uint64 `json:"tcp_spurious_rtx_host_queues"`
BusyPollRxPackets uint64 `json:"busy_poll_rx_packets"`
TCPAutoCorking uint64 `json:"tcp_auto_corking"`
TCPFromZeroWindowAdv uint64 `json:"tcp_from_zero_window_adv"`
TCPToZeroWindowAdv uint64 `json:"tcp_to_zero_window_adv"`
TCPWantZeroWindowAdv uint64 `json:"tcp_want_zero_window_adv"`
TCPSynRetrans uint64 `json:"tcp_syn_retrans"`
TCPOrigDataSent uint64 `json:"tcp_orig_data_sent"`
// IpExt
InNoRoutes uint64 `json:"in_no_routes"`
InTruncatedPkts uint64 `json:"in_truncated_pkts"`
InMcastPkts uint64 `json:"in_mcast_pkts"`
OutMcastPkts uint64 `json:"out_mcast_pkts"`
InBcastPkts uint64 `json:"in_bcast_pkts"`
OutBcastPkts uint64 `json:"out_bcast_pkts"`
InOctets uint64 `json:"in_octets"`
OutOctets uint64 `json:"out_octets"`
InMcastOctets uint64 `json:"in_mcast_octets"`
OutMcastOctets uint64 `json:"out_mcast_octets"`
InBcastOctets uint64 `json:"in_bcast_octets"`
OutBcastOctets uint64 `json:"out_bcast_octets"`
InCsumErrors uint64 `json:"in_csum_errors"`
InNoECTPkts uint64 `json:"in_no_ect_pkts"`
InECT1Pkts uint64 `json:"in_ect1_pkts"`
InECT0Pkts uint64 `json:"in_ect0_pkts"`
InCEPkts uint64 `json:"in_ce_pkts"`
}
func ReadNetStat(path string) (*NetStat, error) {
data, err := ioutil.ReadFile(path)
if err != nil {
return nil, err
}
lines := strings.Split(string(data), "\n")
// Maps a netstat metric to its value (i.e. SyncookiesSent --> 0)
statMap := make(map[string]string)
// patterns
// TcpExt: SyncookiesSent SyncookiesRecv SyncookiesFailed... <-- header
// TcpExt: 0 0 1764... <-- values
for i := 1; i < len(lines); i = i + 2 {
headers := strings.Fields(lines[i-1][strings.Index(lines[i-1], ":")+1:])
values := strings.Fields(lines[i][strings.Index(lines[i], ":")+1:])
for j, header := range headers {
statMap[header] = values[j]
}
}
var netstat NetStat = NetStat{}
elem := reflect.ValueOf(&netstat).Elem()
typeOfElem := elem.Type()
for i := 0; i < elem.NumField(); i++ {
if val, ok := statMap[typeOfElem.Field(i).Name]; ok {
parsedVal, _ := strconv.ParseUint(val, 10, 64)
elem.Field(i).SetUint(parsedVal)
}
}
return &netstat, nil
}

73
vendor/github.com/c9s/goprocinfo/linux/network_stat.go generated vendored
View File

@ -1,73 +0,0 @@
package linux
import (
"io/ioutil"
"strconv"
"strings"
)
type NetworkStat struct {
Iface string `json:"iface"`
RxBytes uint64 `json:"rxbytes"`
RxPackets uint64 `json:"rxpackets"`
RxErrs uint64 `json:"rxerrs"`
RxDrop uint64 `json:"rxdrop"`
RxFifo uint64 `json:"rxfifo"`
RxFrame uint64 `json:"rxframe"`
RxCompressed uint64 `json:"rxcompressed"`
RxMulticast uint64 `json:"rxmulticast"`
TxBytes uint64 `json:"txbytes"`
TxPackets uint64 `json:"txpackets"`
TxErrs uint64 `json:"txerrs"`
TxDrop uint64 `json:"txdrop"`
TxFifo uint64 `json:"txfifo"`
TxColls uint64 `json:"txcolls"`
TxCarrier uint64 `json:"txcarrier"`
TxCompressed uint64 `json:"txcompressed"`
}
func ReadNetworkStat(path string) ([]NetworkStat, error) {
data, err := ioutil.ReadFile(path)
if err != nil {
return nil, err
}
lines := strings.Split(string(data), "\n")
// lines[2:] remove /proc/net/dev header
results := make([]NetworkStat, len(lines[2:])-1)
for i, line := range lines[2:] {
// patterns
// <iface>: 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0
// or
// <iface>:0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 (without space after colon)
colon := strings.Index(line, ":")
if colon > 0 {
metrics := line[colon+1:]
fields := strings.Fields(metrics)
results[i].Iface = strings.Replace(line[0:colon], " ", "", -1)
results[i].RxBytes, _ = strconv.ParseUint(fields[0], 10, 64)
results[i].RxPackets, _ = strconv.ParseUint(fields[1], 10, 64)
results[i].RxErrs, _ = strconv.ParseUint(fields[2], 10, 64)
results[i].RxDrop, _ = strconv.ParseUint(fields[3], 10, 64)
results[i].RxFifo, _ = strconv.ParseUint(fields[4], 10, 64)
results[i].RxFrame, _ = strconv.ParseUint(fields[5], 10, 64)
results[i].RxCompressed, _ = strconv.ParseUint(fields[6], 10, 64)
results[i].RxMulticast, _ = strconv.ParseUint(fields[7], 10, 64)
results[i].TxBytes, _ = strconv.ParseUint(fields[8], 10, 64)
results[i].TxPackets, _ = strconv.ParseUint(fields[9], 10, 64)
results[i].TxErrs, _ = strconv.ParseUint(fields[10], 10, 64)
results[i].TxDrop, _ = strconv.ParseUint(fields[11], 10, 64)
results[i].TxFifo, _ = strconv.ParseUint(fields[12], 10, 64)
results[i].TxColls, _ = strconv.ParseUint(fields[13], 10, 64)
results[i].TxCarrier, _ = strconv.ParseUint(fields[14], 10, 64)
results[i].TxCompressed, _ = strconv.ParseUint(fields[15], 10, 64)
}
}
return results, nil
}

62
vendor/github.com/c9s/goprocinfo/linux/process.go generated vendored
View File

@ -1,62 +0,0 @@
package linux
import (
"os"
"path/filepath"
"strconv"
)
type Process struct {
Status ProcessStatus `json:"status"`
Statm ProcessStatm `json:"statm"`
Stat ProcessStat `json:"stat"`
IO ProcessIO `json:"io"`
Cmdline string `json:"cmdline"`
}
func ReadProcess(pid uint64, path string) (*Process, error) {
var err error
p := filepath.Join(path, strconv.FormatUint(pid, 10))
if _, err = os.Stat(p); err != nil {
return nil, err
}
process := Process{}
var io *ProcessIO
var stat *ProcessStat
var statm *ProcessStatm
var status *ProcessStatus
var cmdline string
if io, err = ReadProcessIO(filepath.Join(p, "io")); err != nil {
return nil, err
}
if stat, err = ReadProcessStat(filepath.Join(p, "stat")); err != nil {
return nil, err
}
if statm, err = ReadProcessStatm(filepath.Join(p, "statm")); err != nil {
return nil, err
}
if status, err = ReadProcessStatus(filepath.Join(p, "status")); err != nil {
return nil, err
}
if cmdline, err = ReadProcessCmdline(filepath.Join(p, "cmdline")); err != nil {
return nil, err
}
process.IO = *io
process.Stat = *stat
process.Statm = *statm
process.Status = *status
process.Cmdline = cmdline
return &process, nil
}

39
vendor/github.com/c9s/goprocinfo/linux/process_cmdline.go generated vendored
View File

@ -1,39 +0,0 @@
package linux
import (
"io/ioutil"
"strings"
)
func ReadProcessCmdline(path string) (string, error) {
b, err := ioutil.ReadFile(path)
if err != nil {
return "", err
}
l := len(b) - 1 // Define limit before last byte ('\0')
z := byte(0) // '\0' or null byte
s := byte(0x20) // space byte
c := 0 // cursor of useful bytes
for i := 0; i < l; i++ {
// Check if next byte is not a '\0' byte.
if b[i+1] != z {
// Offset must match a '\0' byte.
c = i + 2
// If current byte is '\0', replace it with a space byte.
if b[i] == z {
b[i] = s
}
}
}
x := strings.TrimSpace(string(b[0:c]))
return x, nil
}

71
vendor/github.com/c9s/goprocinfo/linux/process_io.go generated vendored
View File

@ -1,71 +0,0 @@
package linux
import (
"io/ioutil"
"reflect"
"strconv"
"strings"
)
// I/O statistics for the process.
type ProcessIO struct {
RChar uint64 `json:"rchar" field:"rchar"` // chars read
WChar uint64 `json:"wchar" field:"wchar"` // chars written
Syscr uint64 `json:"syscr" field:"syscr"` // read syscalls
Syscw uint64 `json:"syscw" field:"syscw"` // write syscalls
ReadBytes uint64 `json:"read_bytes" field:"read_bytes"` // bytes read
WriteBytes uint64 `json:"write_bytes" field:"write_bytes"` // bytes written
CancelledWriteBytes uint64 `json:"cancelled_write_bytes" field:"cancelled_write_bytes"` // bytes truncated
}
func ReadProcessIO(path string) (*ProcessIO, error) {
b, err := ioutil.ReadFile(path)
if err != nil {
return nil, err
}
// Maps a io metric to its value (i.e. rchar --> 100000)
m := map[string]uint64{}
io := ProcessIO{}
lines := strings.Split(string(b), "\n")
for _, line := range lines {
if strings.Index(line, ": ") == -1 {
continue
}
l := strings.Split(line, ": ")
k := l[0]
v, err := strconv.ParseUint(l[1], 10, 64)
if err != nil {
return nil, err
}
m[k] = v
}
e := reflect.ValueOf(&io).Elem()
t := e.Type()
for i := 0; i < e.NumField(); i++ {
k := t.Field(i).Tag.Get("field")
v, ok := m[k]
if ok {
e.Field(i).SetUint(v)
}
}
return &io, nil
}

54
vendor/github.com/c9s/goprocinfo/linux/process_pid.go generated vendored
View File

@ -1,54 +0,0 @@
package linux
import (
"io/ioutil"
"os"
"path/filepath"
"strconv"
"strings"
)
func ReadMaxPID(path string) (uint64, error) {
b, err := ioutil.ReadFile(path)
if err != nil {
return 0, err
}
s := strings.TrimSpace(string(b))
i, err := strconv.ParseUint(s, 10, 64)
if err != nil {
return 0, err
}
return i, nil
}
func ListPID(path string, max uint64) ([]uint64, error) {
l := make([]uint64, 0, 5)
for i := uint64(1); i <= max; i++ {
p := filepath.Join(path, strconv.FormatUint(i, 10))
s, err := os.Stat(p)
if err != nil && !os.IsNotExist(err) {
return nil, err
}
if err != nil || !s.IsDir() {
continue
}
l = append(l, i)
}
return l, nil
}

303
vendor/github.com/c9s/goprocinfo/linux/process_stat.go generated vendored
View File

@ -1,303 +0,0 @@
package linux
import (
"io/ioutil"
"regexp"
"strconv"
"strings"
)
// Status information about the process.
type ProcessStat struct {
Pid uint64 `json:"pid"`
Comm string `json:"comm"`
State string `json:"state"`
Ppid int64 `json:"ppid"`
Pgrp int64 `json:"pgrp"`
Session int64 `json:"session"`
TtyNr int64 `json:"tty_nr"`
Tpgid int64 `json:"tpgid"`
Flags uint64 `json:"flags"`
Minflt uint64 `json:"minflt"`
Cminflt uint64 `json:"cminflt"`
Majflt uint64 `json:"majflt"`
Cmajflt uint64 `json:"cmajflt"`
Utime uint64 `json:"utime"`
Stime uint64 `json:"stime"`
Cutime int64 `json:"cutime"`
Cstime int64 `json:"cstime"`
Priority int64 `json:"priority"`
Nice int64 `json:"nice"`
NumThreads int64 `json:"num_threads"`
Itrealvalue int64 `json:"itrealvalue"`
Starttime uint64 `json:"starttime"`
Vsize uint64 `json:"vsize"`
Rss int64 `json:"rss"`
Rsslim uint64 `json:"rsslim"`
Startcode uint64 `json:"startcode"`
Endcode uint64 `json:"endcode"`
Startstack uint64 `json:"startstack"`
Kstkesp uint64 `json:"kstkesp"`
Kstkeip uint64 `json:"kstkeip"`
Signal uint64 `json:"signal"`
Blocked uint64 `json:"blocked"`
Sigignore uint64 `json:"sigignore"`
Sigcatch uint64 `json:"sigcatch"`
Wchan uint64 `json:"wchan"`
Nswap uint64 `json:"nswap"`
Cnswap uint64 `json:"cnswap"`
ExitSignal int64 `json:"exit_signal"`
Processor int64 `json:"processor"`
RtPriority uint64 `json:"rt_priority"`
Policy uint64 `json:"policy"`
DelayacctBlkioTicks uint64 `json:"delayacct_blkio_ticks"`
GuestTime uint64 `json:"guest_time"`
CguestTime int64 `json:"cguest_time"`
StartData uint64 `json:"start_data"`
EndData uint64 `json:"end_data"`
StartBrk uint64 `json:"start_brk"`
ArgStart uint64 `json:"arg_start"`
ArgEnd uint64 `json:"arg_end"`
EnvStart uint64 `json:"env_start"`
EnvEnd uint64 `json:"env_end"`
ExitCode int64 `json:"exit_code"`
}
var processStatRegExp = regexp.MustCompile("^(\\d+)( \\(.*?\\) )(.*)$")
func ReadProcessStat(path string) (*ProcessStat, error) {
b, err := ioutil.ReadFile(path)
if err != nil {
return nil, err
}
s := string(b)
f := make([]string, 0, 32)
e := processStatRegExp.FindStringSubmatch(strings.TrimSpace(s))
// Inject process Pid
f = append(f, e[1])
// Inject process Comm
f = append(f, strings.TrimSpace(e[2]))
// Inject all remaining process info
f = append(f, (strings.Fields(e[3]))...)
stat := ProcessStat{}
for i := 0; i < len(f); i++ {
switch i {
case 0:
if stat.Pid, err = strconv.ParseUint(f[i], 10, 64); err != nil {
return nil, err
}
case 1:
stat.Comm = f[i]
case 2:
stat.State = f[i]
case 3:
if stat.Ppid, err = strconv.ParseInt(f[i], 10, 64); err != nil {
return nil, err
}
case 4:
if stat.Pgrp, err = strconv.ParseInt(f[i], 10, 64); err != nil {
return nil, err
}
case 5:
if stat.Session, err = strconv.ParseInt(f[i], 10, 64); err != nil {
return nil, err
}
case 6:
if stat.TtyNr, err = strconv.ParseInt(f[i], 10, 64); err != nil {
return nil, err
}
case 7:
if stat.Tpgid, err = strconv.ParseInt(f[i], 10, 64); err != nil {
return nil, err
}
case 8:
if stat.Flags, err = strconv.ParseUint(f[i], 10, 64); err != nil {
return nil, err
}
case 9:
if stat.Minflt, err = strconv.ParseUint(f[i], 10, 64); err != nil {
return nil, err
}
case 10:
if stat.Cminflt, err = strconv.ParseUint(f[i], 10, 64); err != nil {
return nil, err
}
case 11:
if stat.Majflt, err = strconv.ParseUint(f[i], 10, 64); err != nil {
return nil, err
}
case 12:
if stat.Cmajflt, err = strconv.ParseUint(f[i], 10, 64); err != nil {
return nil, err
}
case 13:
if stat.Utime, err = strconv.ParseUint(f[i], 10, 64); err != nil {
return nil, err
}
case 14:
if stat.Stime, err = strconv.ParseUint(f[i], 10, 64); err != nil {
return nil, err
}
case 15:
if stat.Cutime, err = strconv.ParseInt(f[i], 10, 64); err != nil {
return nil, err
}
case 16:
if stat.Cstime, err = strconv.ParseInt(f[i], 10, 64); err != nil {
return nil, err
}
case 17:
if stat.Priority, err = strconv.ParseInt(f[i], 10, 64); err != nil {
return nil, err
}
case 18:
if stat.Nice, err = strconv.ParseInt(f[i], 10, 64); err != nil {
return nil, err
}
case 19:
if stat.NumThreads, err = strconv.ParseInt(f[i], 10, 64); err != nil {
return nil, err
}
case 20:
if stat.Itrealvalue, err = strconv.ParseInt(f[i], 10, 64); err != nil {
return nil, err
}
case 21:
if stat.Starttime, err = strconv.ParseUint(f[i], 10, 64); err != nil {
return nil, err
}
case 22:
if stat.Vsize, err = strconv.ParseUint(f[i], 10, 64); err != nil {
return nil, err
}
case 23:
if stat.Rss, err = strconv.ParseInt(f[i], 10, 64); err != nil {
return nil, err
}
case 24:
if stat.Rsslim, err = strconv.ParseUint(f[i], 10, 64); err != nil {
return nil, err
}
case 25:
if stat.Startcode, err = strconv.ParseUint(f[i], 10, 64); err != nil {
return nil, err
}
case 26:
if stat.Endcode, err = strconv.ParseUint(f[i], 10, 64); err != nil {
return nil, err
}
case 27:
if stat.Startstack, err = strconv.ParseUint(f[i], 10, 64); err != nil {
return nil, err
}
case 28:
if stat.Kstkesp, err = strconv.ParseUint(f[i], 10, 64); err != nil {
return nil, err
}
case 29:
if stat.Kstkeip, err = strconv.ParseUint(f[i], 10, 64); err != nil {
return nil, err
}
case 30:
if stat.Signal, err = strconv.ParseUint(f[i], 10, 64); err != nil {
return nil, err
}
case 31:
if stat.Blocked, err = strconv.ParseUint(f[i], 10, 64); err != nil {
return nil, err
}
case 32:
if stat.Sigignore, err = strconv.ParseUint(f[i], 10, 64); err != nil {
return nil, err
}
case 33:
if stat.Sigcatch, err = strconv.ParseUint(f[i], 10, 64); err != nil {
return nil, err
}
case 34:
if stat.Wchan, err = strconv.ParseUint(f[i], 10, 64); err != nil {
return nil, err
}
case 35:
if stat.Nswap, err = strconv.ParseUint(f[i], 10, 64); err != nil {
return nil, err
}
case 36:
if stat.Cnswap, err = strconv.ParseUint(f[i], 10, 64); err != nil {
return nil, err
}
case 37:
if stat.ExitSignal, err = strconv.ParseInt(f[i], 10, 64); err != nil {
return nil, err
}
case 38:
if stat.Processor, err = strconv.ParseInt(f[i], 10, 64); err != nil {
return nil, err
}
case 39:
if stat.RtPriority, err = strconv.ParseUint(f[i], 10, 64); err != nil {
return nil, err
}
case 40:
if stat.Policy, err = strconv.ParseUint(f[i], 10, 64); err != nil {
return nil, err
}
case 41:
if stat.DelayacctBlkioTicks, err = strconv.ParseUint(f[i], 10, 64); err != nil {
return nil, err
}
case 42:
if stat.GuestTime, err = strconv.ParseUint(f[i], 10, 64); err != nil {
return nil, err
}
case 43:
if stat.CguestTime, err = strconv.ParseInt(f[i], 10, 64); err != nil {
return nil, err
}
case 44:
if stat.StartData, err = strconv.ParseUint(f[i], 10, 64); err != nil {
return nil, err
}
case 45:
if stat.EndData, err = strconv.ParseUint(f[i], 10, 64); err != nil {
return nil, err
}
case 46:
if stat.StartBrk, err = strconv.ParseUint(f[i], 10, 64); err != nil {
return nil, err
}
case 47:
if stat.ArgStart, err = strconv.ParseUint(f[i], 10, 64); err != nil {
return nil, err
}
case 48:
if stat.ArgEnd, err = strconv.ParseUint(f[i], 10, 64); err != nil {
return nil, err
}
case 49:
if stat.EnvStart, err = strconv.ParseUint(f[i], 10, 64); err != nil {
return nil, err
}
case 50:
if stat.EnvEnd, err = strconv.ParseUint(f[i], 10, 64); err != nil {
return nil, err
}
case 51:
if stat.ExitCode, err = strconv.ParseInt(f[i], 10, 64); err != nil {
return nil, err
}
}
}
return &stat, nil
}

61
vendor/github.com/c9s/goprocinfo/linux/process_statm.go generated vendored
View File

@ -1,61 +0,0 @@
package linux
import (
"io/ioutil"
"strconv"
"strings"
)
// Provides information about memory usage, measured in pages.
type ProcessStatm struct {
Size uint64 `json:"size"` // total program size
Resident uint64 `json:"resident"` // resident set size
Share uint64 `json:"share"` // shared pages
Text uint64 `json:"text"` // text (code)
Lib uint64 `json:"lib"` // library (unused in Linux 2.6)
Data uint64 `json:"data"` // data + stack
Dirty uint64 `json:"dirty"` // dirty pages (unused in Linux 2.6)
}
func ReadProcessStatm(path string) (*ProcessStatm, error) {
b, err := ioutil.ReadFile(path)
if err != nil {
return nil, err
}
s := string(b)
f := strings.Fields(s)
statm := ProcessStatm{}
var n uint64
for i := 0; i < len(f); i++ {
if n, err = strconv.ParseUint(f[i], 10, 64); err != nil {
return nil, err
}
switch i {
case 0:
statm.Size = n
case 1:
statm.Resident = n
case 2:
statm.Share = n
case 3:
statm.Text = n
case 4:
statm.Lib = n
case 5:
statm.Data = n
case 6:
statm.Dirty = n
}
}
return &statm, nil
}

331
vendor/github.com/c9s/goprocinfo/linux/process_status.go generated vendored
View File

@ -1,331 +0,0 @@
package linux
import (
"io/ioutil"
"strconv"
"strings"
)
// Provides much of the information from ProcessStatm and ProcessStat
type ProcessStatus struct {
Name string
State string
Tgid uint64
Pid uint64
PPid int64
TracerPid uint64
RealUid uint64
EffectiveUid uint64
SavedSetUid uint64
FilesystemUid uint64
RealGid uint64
EffectiveGid uint64
SavedSetGid uint64
FilesystemGid uint64
FDSize uint64
Groups []int64
VmPeak uint64
VmSize uint64
VmLck uint64
VmHWM uint64
VmRSS uint64
VmData uint64
VmStk uint64
VmExe uint64
VmLib uint64
VmPTE uint64
VmSwap uint64
Threads uint64
SigQLength uint64
SigQLimit uint64
SigPnd uint64
ShdPnd uint64
SigBlk uint64
SigIgn uint64
SigCgt uint64
CapInh uint64
CapPrm uint64
CapEff uint64
CapBnd uint64
Seccomp uint8
CpusAllowed []uint32
MemsAllowed []uint32
VoluntaryCtxtSwitches uint64
NonvoluntaryCtxtSwitches uint64
}
func ReadProcessStatus(path string) (*ProcessStatus, error) {
b, err := ioutil.ReadFile(path)
if err != nil {
return nil, err
}
status := ProcessStatus{}
lines := strings.Split(string(b), "\n")
for _, line := range lines {
if strings.Index(line, ":") == -1 {
continue
}
l := strings.Split(line, ":")
k := strings.TrimSpace(l[0])
v := strings.TrimSpace(l[1])
switch k {
case "Name":
status.Name = v
case "State":
status.State = v
case "Tgid":
if status.Tgid, err = strconv.ParseUint(v, 10, 64); err != nil {
return nil, err
}
case "Pid":
if status.Pid, err = strconv.ParseUint(v, 10, 64); err != nil {
return nil, err
}
case "PPid":
if status.PPid, err = strconv.ParseInt(v, 10, 64); err != nil {
return nil, err
}
case "TracerPid":
if status.TracerPid, err = strconv.ParseUint(v, 10, 64); err != nil {
return nil, err
}
case "Uid":
if f := strings.Fields(v); len(f) == 4 {
if status.RealUid, err = strconv.ParseUint(f[0], 10, 64); err != nil {
return nil, err
}
if status.EffectiveUid, err = strconv.ParseUint(f[1], 10, 64); err != nil {
return nil, err
}
if status.SavedSetUid, err = strconv.ParseUint(f[2], 10, 64); err != nil {
return nil, err
}
if status.FilesystemUid, err = strconv.ParseUint(f[3], 10, 64); err != nil {
return nil, err
}
}
case "Gid":
if f := strings.Fields(v); len(f) == 4 {
if status.RealGid, err = strconv.ParseUint(f[0], 10, 64); err != nil {
return nil, err
}
if status.EffectiveGid, err = strconv.ParseUint(f[1], 10, 64); err != nil {
return nil, err
}
if status.SavedSetGid, err = strconv.ParseUint(f[2], 10, 64); err != nil {
return nil, err
}
if status.FilesystemGid, err = strconv.ParseUint(f[3], 10, 64); err != nil {
return nil, err
}
}
case "FDSize":
if status.FDSize, err = strconv.ParseUint(v, 10, 64); err != nil {
return nil, err
}
case "Groups":
{
f := strings.Fields(v)
status.Groups = make([]int64, len(f))
for i := range status.Groups {
if status.Groups[i], err = strconv.ParseInt(f[i], 10, 64); err != nil {
return nil, err
}
}
}
case "VmPeak":
{
f := strings.Fields(v)
if status.VmPeak, err = strconv.ParseUint(f[0], 10, 64); err != nil {
return nil, err
}
}
case "VmSize":
{
f := strings.Fields(v)
if status.VmSize, err = strconv.ParseUint(f[0], 10, 64); err != nil {
return nil, err
}
}
case "VmLck":
{
f := strings.Fields(v)
if status.VmLck, err = strconv.ParseUint(f[0], 10, 64); err != nil {
return nil, err
}
}
case "VmHWM":
{
f := strings.Fields(v)
if status.VmHWM, err = strconv.ParseUint(f[0], 10, 64); err != nil {
return nil, err
}
}
case "VmRSS":
{
f := strings.Fields(v)
if status.VmRSS, err = strconv.ParseUint(f[0], 10, 64); err != nil {
return nil, err
}
}
case "VmData":
{
f := strings.Fields(v)
if status.VmData, err = strconv.ParseUint(f[0], 10, 64); err != nil {
return nil, err
}
}
case "VmStk":
{
f := strings.Fields(v)
if status.VmStk, err = strconv.ParseUint(f[0], 10, 64); err != nil {
return nil, err
}
}
case "VmExe":
{
f := strings.Fields(v)
if status.VmExe, err = strconv.ParseUint(f[0], 10, 64); err != nil {
return nil, err
}
}
case "VmLib":
{
f := strings.Fields(v)
if status.VmLib, err = strconv.ParseUint(f[0], 10, 64); err != nil {
return nil, err
}
}
case "VmPTE":
{
f := strings.Fields(v)
if status.VmPTE, err = strconv.ParseUint(f[0], 10, 64); err != nil {
return nil, err
}
}
case "VmSwap":
{
f := strings.Fields(v)
if status.VmSwap, err = strconv.ParseUint(f[0], 10, 64); err != nil {
return nil, err
}
}
case "Threads":
if status.Threads, err = strconv.ParseUint(v, 10, 64); err != nil {
return nil, err
}
case "SigQ":
{
if f := strings.Split(v, "/"); len(f) == 2 {
if status.SigQLength, err = strconv.ParseUint(f[0], 10, 64); err != nil {
return nil, err
}
if status.SigQLimit, err = strconv.ParseUint(f[1], 10, 64); err != nil {
return nil, err
}
}
}
case "SigPnd":
if status.SigPnd, err = strconv.ParseUint(v, 16, 64); err != nil {
return nil, err
}
case "ShdPnd":
if status.ShdPnd, err = strconv.ParseUint(v, 16, 64); err != nil {
return nil, err
}
case "SigBlk":
if status.SigBlk, err = strconv.ParseUint(v, 16, 64); err != nil {
return nil, err
}
case "SigIgn":
if status.SigIgn, err = strconv.ParseUint(v, 16, 64); err != nil {
return nil, err
}
case "SigCgt":
if status.SigCgt, err = strconv.ParseUint(v, 16, 64); err != nil {
return nil, err
}
case "CapInh":
if status.CapInh, err = strconv.ParseUint(v, 16, 64); err != nil {
return nil, err
}
case "CapPrm":
if status.CapPrm, err = strconv.ParseUint(v, 16, 64); err != nil {
return nil, err
}
case "CapEff":
if status.CapEff, err = strconv.ParseUint(v, 16, 64); err != nil {
return nil, err
}
case "CapBnd":
if status.CapBnd, err = strconv.ParseUint(v, 16, 64); err != nil {
return nil, err
}
case "Seccomp":
{
var n uint64
if n, err = strconv.ParseUint(v, 10, 8); err != nil {
return nil, err
}
status.Seccomp = uint8(n)
}
case "Cpus_allowed":
{
var n uint64
f := strings.Split(v, ",")
status.CpusAllowed = make([]uint32, len(f))
for i := range status.CpusAllowed {
if n, err = strconv.ParseUint(f[i], 16, 32); err != nil {
return nil, err
}
status.CpusAllowed[i] = uint32(n)
}
}
case "Mems_allowed":
{
var n uint64
f := strings.Split(v, ",")
status.MemsAllowed = make([]uint32, len(f))
for i := range status.MemsAllowed {
if n, err = strconv.ParseUint(f[i], 16, 32); err != nil {
return nil, err
}
status.MemsAllowed[i] = uint32(n)
}
}
case "voluntary_ctxt_switches":
if status.VoluntaryCtxtSwitches, err = strconv.ParseUint(v, 10, 64); err != nil {
return nil, err
}
case "nonvoluntary_ctxt_switches":
if status.NonvoluntaryCtxtSwitches, err = strconv.ParseUint(v, 10, 64); err != nil {
return nil, err
}
}
}
return &status, nil
}

144
vendor/github.com/c9s/goprocinfo/linux/snmp.go generated vendored
View File

@ -1,144 +0,0 @@
package linux
import (
"io/ioutil"
"reflect"
"strconv"
"strings"
)
type Snmp struct {
// Ip
IpForwarding uint64 `json:"ip_forwarding"`
IpDefaultTTL uint64 `json:"ip_default_ttl"`
IpInReceives uint64 `json:"ip_in_receives"`
IpInHdrErrors uint64 `json:"ip_in_hdr_errors"`
IpInAddrErrors uint64 `json:"ip_in_addr_errors"`
IpForwDatagrams uint64 `json:"ip_forw_datagrams"`
IpInUnknownProtos uint64 `json:"ip_in_unknown_protos"`
IpInDiscards uint64 `json:"ip_in_discards"`
IpInDelivers uint64 `json:"ip_in_delivers"`
IpOutRequests uint64 `json:"ip_out_requests"`
IpOutDiscards uint64 `json:"ip_out_discards"`
IpOutNoRoutes uint64 `json:"ip_out_no_routes"`
IpReasmTimeout uint64 `json:"ip_reasm_timeout"`
IpReasmReqds uint64 `json:"ip_reasm_reqds"`
IpReasmOKs uint64 `json:"ip_reasm_oks"`
IpReasmFails uint64 `json:"ip_reasm_fails"`
IpFragOKs uint64 `json:"ip_frag_oks"`
IpFragFails uint64 `json:"ip_frag_fails"`
IpFragCreates uint64 `json:"ip_frag_creates"`
// Icmp
IcmpInMsgs uint64 `json:"icmp_in_msgs"`
IcmpInErrors uint64 `json:"icmp_in_errors"`
IcmpInCsumErrors uint64 `json:"icmp_in_csum_errors"`
IcmpInDestUnreachs uint64 `json:"icmp_in_dest_unreachs"`
IcmpInTimeExcds uint64 `json:"icmp_in_time_excds"`
IcmpInParmProbs uint64 `json:"icmp_in_parm_probs"`
IcmpInSrcQuenchs uint64 `json:"icmp_in_src_quenchs"`
IcmpInRedirects uint64 `json:"icmp_in_redirects"`
IcmpInEchos uint64 `json:"icmp_in_echos"`
IcmpInEchoReps uint64 `json:"icmp_in_echo_reps"`
IcmpInTimestamps uint64 `json:"icmp_in_timestamps"`
IcmpInTimestampReps uint64 `json:"icmp_in_timestamp_reps"`
IcmpInAddrMasks uint64 `json:"icmp_in_addr_masks"`
IcmpInAddrMaskReps uint64 `json:"icmp_in_addr_mask_reps"`
IcmpOutMsgs uint64 `json:"icmp_out_msgs"`
IcmpOutErrors uint64 `json:"icmp_out_errors"`
IcmpOutDestUnreachs uint64 `json:"icmp_out_dest_unreachs"`
IcmpOutTimeExcds uint64 `json:"icmp_out_time_excds"`
IcmpOutParmProbs uint64 `json:"icmp_out_parm_probs"`
IcmpOutSrcQuenchs uint64 `json:"icmp_out_src_quenchs"`
IcmpOutRedirects uint64 `json:"icmp_out_redirects"`
IcmpOutEchos uint64 `json:"icmp_out_echos"`
IcmpOutEchoReps uint64 `json:"icmp_out_echo_reps"`
IcmpOutTimestamps uint64 `json:"icmp_out_timestamps"`
IcmpOutTimestampReps uint64 `json:"icmp_out_timestamp_reps"`
IcmpOutAddrMasks uint64 `json:"icmp_out_addr_masks"`
IcmpOutAddrMaskReps uint64 `json:"icmp_out_addr_mask_reps"`
// IcmpMsg
IcmpMsgInType0 uint64 `json:"icmpmsg_in_type0"`
IcmpMsgInType3 uint64 `json:"icmpmsg_in_type3"`
IcmpMsgInType5 uint64 `json:"icmpmsg_in_type5"`
IcmpMsgInType8 uint64 `json:"icmpmsg_in_type8"`
IcmpMsgInType11 uint64 `json:"icmpmsg_in_type11"`
IcmpMsgInType13 uint64 `json:"icmpmsg_in_type13"`
IcmpMsgOutType0 uint64 `json:"icmpmsg_out_type0"`
IcmpMsgOutType3 uint64 `json:"icmpmsg_out_type3"`
IcmpMsgOutType8 uint64 `json:"icmpmsg_out_type8"`
IcmpMsgOutType14 uint64 `json:"icmpmsg_out_type14"`
IcmpMsgOutType69 uint64 `json:"icmpmsg_out_type69"`
// TCP
TcpRtoAlgorithm uint64 `json:"tcp_rto_algorithm"`
TcpRtoMin uint64 `json:"tcp_rto_min"`
TcpRtoMax uint64 `json:"tcp_rto_max"`
TcpMaxConn uint64 `json:"tcp_max_conn"`
TcpActiveOpens uint64 `json:"tcp_active_opens"`
TcpPassiveOpens uint64 `json:"tcp_passive_opens"`
TcpAttemptFails uint64 `json:"tcp_attempt_fails"`
TcpEstabResets uint64 `json:"tcp_estab_resets"`
TcpCurrEstab uint64 `json:"tcp_curr_estab"`
TcpInSegs uint64 `json:"tcp_in_segs"`
TcpOutSegs uint64 `json:"tcp_out_segs"`
TcpRetransSegs uint64 `json:"tcp_retrans_segs"`
TcpInErrs uint64 `json:"tcp_in_errs"`
TcpOutRsts uint64 `json:"tcp_out_rsts"`
TcpInCsumErrors uint64 `json:"tcp_in_csum_errors"`
// UDP
UdpInDatagrams uint64 `json:"udp_in_datagrams"`
UdpNoPorts uint64 `json:"udp_no_ports"`
UdpInErrors uint64 `json:"udp_in_errors"`
UdpOutDatagrams uint64 `json:"udp_out_datagrams"`
UdpRcvbufErrors uint64 `json:"udp_rcvbuf_errors"`
UdpSndbufErrors uint64 `json:"udp_sndbuf_errors"`
UdpInCsumErrors uint64 `json:"udp_in_csum_errors"`
// UDPLite
UdpLiteInDatagrams uint64 `json:"udp_lite_in_datagrams"`
UdpLiteNoPorts uint64 `json:"udp_lite_no_ports"`
UdpLiteInErrors uint64 `json:"udp_lite_in_errors"`
UdpLiteOutDatagrams uint64 `json:"udp_lite_out_datagrams"`
UdpLiteRcvbufErrors uint64 `json:"udp_lite_rcvbuf_errors"`
UdpLiteSndbufErrors uint64 `json:"udp_lite_sndbuf_errors"`
UdpLiteInCsumErrors uint64 `json:"udp_lite_in_csum_errors"`
}
func ReadSnmp(path string) (*Snmp, error) {
data, err := ioutil.ReadFile(path)
if err != nil {
return nil, err
}
lines := strings.Split(string(data), "\n")
// Maps an SNMP metric to its value (i.e. SyncookiesSent --> 0)
statMap := make(map[string]string)
// patterns
// Ip: Forwarding DefaultTTL InReceives InHdrErrors... <-- header
// Ip: 2 64 9305753793 0 0 0 0 0... <-- values
for i := 1; i < len(lines); i = i + 2 {
headers := strings.Fields(lines[i-1][strings.Index(lines[i-1], ":")+1:])
values := strings.Fields(lines[i][strings.Index(lines[i], ":")+1:])
protocol := strings.Replace(strings.Fields(lines[i-1])[0], ":", "", -1)
for j, header := range headers {
statMap[protocol+header] = values[j]
}
}
var snmp Snmp = Snmp{}
elem := reflect.ValueOf(&snmp).Elem()
typeOfElem := elem.Type()
for i := 0; i < elem.NumField(); i++ {
if val, ok := statMap[typeOfElem.Field(i).Name]; ok {
parsedVal, _ := strconv.ParseUint(val, 10, 64)
elem.Field(i).SetUint(parsedVal)
}
}
return &snmp, nil
}

82
vendor/github.com/c9s/goprocinfo/linux/sockstat.go generated vendored
View File

@ -1,82 +0,0 @@
package linux
import (
"io/ioutil"
"reflect"
"strconv"
"strings"
)
type SockStat struct {
// sockets:
SocketsUsed uint64 `json:"sockets_used" field:"sockets.used"`
// TCP:
TCPInUse uint64 `json:"tcp_in_use" field:"TCP.inuse"`
TCPOrphan uint64 `json:"tcp_orphan" field:"TCP.orphan"`
TCPTimeWait uint64 `json:"tcp_time_wait" field:"TCP.tw"`
TCPAllocated uint64 `json:"tcp_allocated" field:"TCP.alloc"`
TCPMemory uint64 `json:"tcp_memory" field:"TCP.mem"`
// UDP:
UDPInUse uint64 `json:"udp_in_use" field:"UDP.inuse"`
UDPMemory uint64 `json:"udp_memory" field:"UDP.mem"`
// UDPLITE:
UDPLITEInUse uint64 `json:"udplite_in_use" field:"UDPLITE.inuse"`
// RAW:
RAWInUse uint64 `json:"raw_in_use" field:"RAW.inuse"`
// FRAG:
FRAGInUse uint64 `json:"frag_in_use" field:"FRAG.inuse"`
FRAGMemory uint64 `json:"frag_memory" field:"FRAG.memory"`
}
func ReadSockStat(path string) (*SockStat, error) {
data, err := ioutil.ReadFile(path)
if err != nil {
return nil, err
}
lines := strings.Split(string(data), "\n")
// Maps a meminfo metric to its value (i.e. MemTotal --> 100000)
statMap := map[string]uint64{}
var sockStat SockStat = SockStat{}
for _, line := range lines {
if strings.Index(line, ":") == -1 {
continue
}
statType := line[0:strings.Index(line, ":")] + "."
// The fields have this pattern: inuse 27 orphan 1 tw 23 alloc 31 mem 3
// The stats are grouped into pairs and need to be parsed and placed into the stat map.
key := ""
for k, v := range strings.Fields(line[strings.Index(line, ":")+1:]) {
// Every second field is a value.
if (k+1)%2 != 0 {
key = v
continue
}
val, _ := strconv.ParseUint(v, 10, 64)
statMap[statType+key] = val
}
}
elem := reflect.ValueOf(&sockStat).Elem()
typeOfElem := elem.Type()
for i := 0; i < elem.NumField(); i++ {
val, ok := statMap[typeOfElem.Field(i).Tag.Get("field")]
if ok {
elem.Field(i).SetUint(val)
}
}
return &sockStat, nil
}

106
vendor/github.com/c9s/goprocinfo/linux/stat.go generated vendored
View File

@ -1,106 +0,0 @@
package linux
import (
"io/ioutil"
"strconv"
"strings"
"time"
)
type Stat struct {
CPUStatAll CPUStat `json:"cpu_all"`
CPUStats []CPUStat `json:"cpus"`
Interrupts uint64 `json:"intr"`
ContextSwitches uint64 `json:"ctxt"`
BootTime time.Time `json:"btime"`
Processes uint64 `json:"processes"`
ProcsRunning uint64 `json:"procs_running"`
ProcsBlocked uint64 `json:"procs_blocked"`
}
type CPUStat struct {
Id string `json:"id"`
User uint64 `json:"user"`
Nice uint64 `json:"nice"`
System uint64 `json:"system"`
Idle uint64 `json:"idle"`
IOWait uint64 `json:"iowait"`
IRQ uint64 `json:"irq"`
SoftIRQ uint64 `json:"softirq"`
Steal uint64 `json:"steal"`
Guest uint64 `json:"guest"`
GuestNice uint64 `json:"guest_nice"`
}
func createCPUStat(fields []string) *CPUStat {
s := CPUStat{}
s.Id = fields[0]
for i := 1; i < len(fields); i++ {
v, _ := strconv.ParseUint(fields[i], 10, 64)
switch i {
case 1:
s.User = v
case 2:
s.Nice = v
case 3:
s.System = v
case 4:
s.Idle = v
case 5:
s.IOWait = v
case 6:
s.IRQ = v
case 7:
s.SoftIRQ = v
case 8:
s.Steal = v
case 9:
s.Guest = v
case 10:
s.GuestNice = v
}
}
return &s
}
func ReadStat(path string) (*Stat, error) {
b, err := ioutil.ReadFile(path)
if err != nil {
return nil, err
}
content := string(b)
lines := strings.Split(content, "\n")
var stat Stat = Stat{}
for i, line := range lines {
fields := strings.Fields(line)
if len(fields) == 0 {
continue
}
if fields[0][:3] == "cpu" {
if cpuStat := createCPUStat(fields); cpuStat != nil {
if i == 0 {
stat.CPUStatAll = *cpuStat
} else {
stat.CPUStats = append(stat.CPUStats, *cpuStat)
}
}
} else if fields[0] == "intr" {
stat.Interrupts, _ = strconv.ParseUint(fields[1], 10, 64)
} else if fields[0] == "ctxt" {
stat.ContextSwitches, _ = strconv.ParseUint(fields[1], 10, 64)
} else if fields[0] == "btime" {
seconds, _ := strconv.ParseInt(fields[1], 10, 64)
stat.BootTime = time.Unix(seconds, 0)
} else if fields[0] == "processes" {
stat.Processes, _ = strconv.ParseUint(fields[1], 10, 64)
} else if fields[0] == "procs_running" {
stat.ProcsRunning, _ = strconv.ParseUint(fields[1], 10, 64)
} else if fields[0] == "procs_blocked" {
stat.ProcsBlocked, _ = strconv.ParseUint(fields[1], 10, 64)
}
}
return &stat, nil
}

43
vendor/github.com/c9s/goprocinfo/linux/uptime.go generated vendored
View File

@ -1,43 +0,0 @@
package linux
import (
"io/ioutil"
"strconv"
"strings"
"time"
)
type Uptime struct {
Total float64 `json:"total"`
Idle float64 `json:"idle"`
}
func (self *Uptime) GetTotalDuration() time.Duration {
return time.Duration(self.Total) * time.Second
}
func (self *Uptime) GetIdleDuration() time.Duration {
return time.Duration(self.Idle) * time.Second
}
func (self *Uptime) CalculateIdle() float64 {
// XXX
// num2/(num1*N) # N = SMP CPU numbers
return 0
}
func ReadUptime(path string) (*Uptime, error) {
b, err := ioutil.ReadFile(path)
if err != nil {
return nil, err
}
fields := strings.Fields(string(b))
uptime := Uptime{}
if uptime.Total, err = strconv.ParseFloat(fields[0], 64); err != nil {
return nil, err
}
if uptime.Idle, err = strconv.ParseFloat(fields[1], 64); err != nil {
return nil, err
}
return &uptime, nil
}

373
vendor/github.com/c9s/goprocinfo/linux/vmstat.go generated vendored
View File

@ -1,373 +0,0 @@
package linux
import (
"io/ioutil"
"strconv"
"strings"
)
type VMStat struct {
NrFreePages uint64 `json:"nr_free_pages"`
NrAllocBatch uint64 `json:"nr_alloc_batch"`
NrInactiveAnon uint64 `json:"nr_inactive_anon"`
NrActiveAnon uint64 `json:"nr_active_anon"`
NrInactiveFile uint64 `json:"nr_inactive_file"`
NrActiveFile uint64 `json:"nr_active_file"`
NrUnevictable uint64 `json:"nr_unevictable"`
NrMlock uint64 `json:"nr_mlock"`
NrAnonPages uint64 `json:"nr_anon_pages"`
NrMapped uint64 `json:"nr_mapped"`
NrFilePages uint64 `json:"nr_file_pages"`
NrDirty uint64 `json:"nr_dirty"`
NrWriteback uint64 `json:"nr_writeback"`
NrSlabReclaimable uint64 `json:"nr_slab_reclaimable"`
NrSlabUnreclaimable uint64 `json:"nr_slab_unreclaimable"`
NrPageTablePages uint64 `json:"nr_page_table_pages"`
NrKernelStack uint64 `json:"nr_kernel_stack"`
NrUnstable uint64 `json:"nr_unstable"`
NrBounce uint64 `json:"nr_bounce"`
NrVmscanWrite uint64 `json:"nr_vmscan_write"`
NrVmscanImmediateReclaim uint64 `json:"nr_vmscan_immediate_reclaim"`
NrWritebackTemp uint64 `json:"nr_writeback_temp"`
NrIsolatedAnon uint64 `json:"nr_isolated_anon"`
NrIsolatedFile uint64 `json:"nr_isolated_file"`
NrShmem uint64 `json:"nr_shmem"`
NrDirtied uint64 `json:"nr_dirtied"`
NrWritten uint64 `json:"nr_written"`
NumaHit uint64 `json:"numa_hit"`
NumaMiss uint64 `json:"numa_miss"`
NumaForeign uint64 `json:"numa_foreign"`
NumaInterleave uint64 `json:"numa_interleave"`
NumaLocal uint64 `json:"numa_local"`
NumaOther uint64 `json:"numa_other"`
WorkingsetRefault uint64 `json:"workingset_refault"`
WorkingsetActivate uint64 `json:"workingset_activate"`
WorkingsetNodereclaim uint64 `json:"workingset_nodereclaim"`
NrAnonTransparentHugepages uint64 `json:"nr_anon_transparent_hugepages"`
NrFreeCma uint64 `json:"nr_free_cma"`
NrDirtyThreshold uint64 `json:"nr_dirty_threshold"`
NrDirtyBackgroundThreshold uint64 `json:"nr_dirty_background_threshold"`
PagePagein uint64 `json:"pgpgin"`
PagePageout uint64 `json:"pgpgout"`
PageSwapin uint64 `json:"pswpin"`
PageSwapout uint64 `json:"pswpout"`
PageAllocDMA uint64 `json:"pgalloc_dma"`
PageAllocDMA32 uint64 `json:"pgalloc_dma32"`
PageAllocNormal uint64 `json:"pgalloc_normal"`
PageAllocMovable uint64 `json:"pgalloc_movable"`
PageFree uint64 `json:"pgfree"`
PageActivate uint64 `json:"pgactivate"`
PageDeactivate uint64 `json:"pgdeactivate"`
PageFault uint64 `json:"pgfault"`
PageMajorFault uint64 `json:"pgmajfault"`
PageRefillDMA uint64 `json:"pgrefill_dma"`
PageRefillDMA32 uint64 `json:"pgrefill_dma32"`
PageRefillMormal uint64 `json:"pgrefill_normal"`
PageRefillMovable uint64 `json:"pgrefill_movable"`
PageStealKswapdDMA uint64 `json:"pgsteal_kswapd_dma"`
PageStealKswapdDMA32 uint64 `json:"pgsteal_kswapd_dma32"`
PageStealKswapdNormal uint64 `json:"pgsteal_kswapd_normal"`
PageStealKswapdMovable uint64 `json:"pgsteal_kswapd_movable"`
PageStealDirectDMA uint64 `json:"pgsteal_direct_dma"`
PageStealDirectDMA32 uint64 `json:"pgsteal_direct_dma32"`
PageStealDirectNormal uint64 `json:"pgsteal_direct_normal"`
PageStealDirectMovable uint64 `json:"pgsteal_direct_movable"`
PageScanKswapdDMA uint64 `json:"pgscan_kswapd_dma"`
PageScanKswapdDMA32 uint64 `json:"pgscan_kswapd_dma32"`
PageScanKswapdNormal uint64 `json:"pgscan_kswapd_normal"`
PageScanKswapdMovable uint64 `json:"pgscan_kswapd_movable"`
PageScanDirectDMA uint64 `json:"pgscan_direct_dma"`
PageScanDirectDMA32 uint64 `json:"pgscan_direct_dma32"`
PageScanDirectNormal uint64 `json:"pgscan_direct_normal"`
PageScanDirectMovable uint64 `json:"pgscan_direct_movable"`
PageScanDirectThrottle uint64 `json:"pgscan_direct_throttle"`
ZoneReclaimFailed uint64 `json:"zone_reclaim_failed"`
PageInodeSteal uint64 `json:"pginodesteal"`
SlabsScanned uint64 `json:"slabs_scanned"`
KswapdInodesteal uint64 `json:"kswapd_inodesteal"`
KswapdLowWatermarkHitQuickly uint64 `json:"kswapd_low_wmark_hit_quickly"`
KswapdHighWatermarkHitQuickly uint64 `json:"kswapd_high_wmark_hit_quickly"`
PageoutRun uint64 `json:"pageoutrun"`
AllocStall uint64 `json:"allocstall"`
PageRotated uint64 `json:"pgrotated"`
DropPagecache uint64 `json:"drop_pagecache"`
DropSlab uint64 `json:"drop_slab"`
NumaPteUpdates uint64 `json:"numa_pte_updates"`
NumaHugePteUpdates uint64 `json:"numa_huge_pte_updates"`
NumaHintFaults uint64 `json:"numa_hint_faults"`
NumaHintFaults_local uint64 `json:"numa_hint_faults_local"`
NumaPagesMigrated uint64 `json:"numa_pages_migrated"`
PageMigrateSuccess uint64 `json:"pgmigrate_success"`
PageMigrateFail uint64 `json:"pgmigrate_fail"`
CompactMigrateScanned uint64 `json:"compact_migrate_scanned"`
CompactFreeScanned uint64 `json:"compact_free_scanned"`
CompactIsolated uint64 `json:"compact_isolated"`
CompactStall uint64 `json:"compact_stall"`
CompactFail uint64 `json:"compact_fail"`
CompactSuccess uint64 `json:"compact_success"`
HtlbBuddyAllocSuccess uint64 `json:"htlb_buddy_alloc_success"`
HtlbBuddyAllocFail uint64 `json:"htlb_buddy_alloc_fail"`
UnevictablePagesCulled uint64 `json:"unevictable_pgs_culled"`
UnevictablePagesScanned uint64 `json:"unevictable_pgs_scanned"`
UnevictablePagesRescued uint64 `json:"unevictable_pgs_rescued"`
UnevictablePagesMlocked uint64 `json:"unevictable_pgs_mlocked"`
UnevictablePagesMunlocked uint64 `json:"unevictable_pgs_munlocked"`
UnevictablePagesCleared uint64 `json:"unevictable_pgs_cleared"`
UnevictablePagesStranded uint64 `json:"unevictable_pgs_stranded"`
THPFaultAlloc uint64 `json:"thp_fault_alloc"`
THPFaultFallback uint64 `json:"thp_fault_fallback"`
THPCollapseAlloc uint64 `json:"thp_collapse_alloc"`
THPCollapseAllocFailed uint64 `json:"thp_collapse_alloc_failed"`
THPSplit uint64 `json:"thp_split"`
THPZeroPageAlloc uint64 `json:"thp_zero_page_alloc"`
THPZeroPageAllocFailed uint64 `json:"thp_zero_page_alloc_failed"`
}
func ReadVMStat(path string) (*VMStat, error) {
b, err := ioutil.ReadFile(path)
if err != nil {
return nil, err
}
content := string(b)
lines := strings.Split(content, "\n")
vmstat := VMStat{}
for _, line := range lines {
fields := strings.Fields(line)
if len(fields) != 2 {
continue
}
name := fields[0]
value, _ := strconv.ParseUint(fields[1], 10, 64)
switch name {
case "nr_free_pages":
vmstat.NrFreePages = value
case "nr_alloc_batch":
vmstat.NrAllocBatch = value
case "nr_inactive_anon":
vmstat.NrInactiveAnon = value
case "nr_active_anon":
vmstat.NrActiveAnon = value
case "nr_inactive_file":
vmstat.NrInactiveFile = value
case "nr_active_file":
vmstat.NrActiveFile = value
case "nr_unevictable":
vmstat.NrUnevictable = value
case "nr_mlock":
vmstat.NrMlock = value
case "nr_anon_pages":
vmstat.NrAnonPages = value
case "nr_mapped":
vmstat.NrMapped = value
case "nr_file_pages":
vmstat.NrFilePages = value
case "nr_dirty":
vmstat.NrDirty = value
case "nr_writeback":
vmstat.NrWriteback = value
case "nr_slab_reclaimable":
vmstat.NrSlabReclaimable = value
case "nr_slab_unreclaimable":
vmstat.NrSlabUnreclaimable = value
case "nr_page_table_pages":
vmstat.NrPageTablePages = value
case "nr_kernel_stack":
vmstat.NrKernelStack = value
case "nr_unstable":
vmstat.NrUnstable = value
case "nr_bounce":
vmstat.NrBounce = value
case "nr_vmscan_write":
vmstat.NrVmscanWrite = value
case "nr_vmscan_immediate_reclaim":
vmstat.NrVmscanImmediateReclaim = value
case "nr_writeback_temp":
vmstat.NrWritebackTemp = value
case "nr_isolated_anon":
vmstat.NrIsolatedAnon = value
case "nr_isolated_file":
vmstat.NrIsolatedFile = value
case "nr_shmem":
vmstat.NrShmem = value
case "nr_dirtied":
vmstat.NrDirtied = value
case "nr_written":
vmstat.NrWritten = value
case "numa_hit":
vmstat.NumaHit = value
case "numa_miss":
vmstat.NumaMiss = value
case "numa_foreign":
vmstat.NumaForeign = value
case "numa_interleave":
vmstat.NumaInterleave = value
case "numa_local":
vmstat.NumaLocal = value
case "numa_other":
vmstat.NumaOther = value
case "workingset_refault":
vmstat.WorkingsetRefault = value
case "workingset_activate":
vmstat.WorkingsetActivate = value
case "workingset_nodereclaim":
vmstat.WorkingsetNodereclaim = value
case "nr_anon_transparent_hugepages":
vmstat.NrAnonTransparentHugepages = value
case "nr_free_cma":
vmstat.NrFreeCma = value
case "nr_dirty_threshold":
vmstat.NrDirtyThreshold = value
case "nr_dirty_background_threshold":
vmstat.NrDirtyBackgroundThreshold = value
case "pgpgin":
vmstat.PagePagein = value
case "pgpgout":
vmstat.PagePageout = value
case "pswpin":
vmstat.PageSwapin = value
case "pswpout":
vmstat.PageSwapout = value
case "pgalloc_dma":
vmstat.PageAllocDMA = value
case "pgalloc_dma32":
vmstat.PageAllocDMA32 = value
case "pgalloc_normal":
vmstat.PageAllocNormal = value
case "pgalloc_movable":
vmstat.PageAllocMovable = value
case "pgfree":
vmstat.PageFree = value
case "pgactivate":
vmstat.PageActivate = value
case "pgdeactivate":
vmstat.PageDeactivate = value
case "pgfault":
vmstat.PageFault = value
case "pgmajfault":
vmstat.PageMajorFault = value
case "pgrefill_dma":
vmstat.PageRefillDMA = value
case "pgrefill_dma32":
vmstat.PageRefillDMA32 = value
case "pgrefill_normal":
vmstat.PageRefillMormal = value
case "pgrefill_movable":
vmstat.PageRefillMovable = value
case "pgsteal_kswapd_dma":
vmstat.PageStealKswapdDMA = value
case "pgsteal_kswapd_dma32":
vmstat.PageStealKswapdDMA32 = value
case "pgsteal_kswapd_normal":
vmstat.PageStealKswapdNormal = value
case "pgsteal_kswapd_movable":
vmstat.PageStealKswapdMovable = value
case "pgsteal_direct_dma":
vmstat.PageStealDirectDMA = value
case "pgsteal_direct_dma32":
vmstat.PageStealDirectDMA32 = value
case "pgsteal_direct_normal":
vmstat.PageStealDirectNormal = value
case "pgsteal_direct_movable":
vmstat.PageStealDirectMovable = value
case "pgscan_kswapd_dma":
vmstat.PageScanKswapdDMA = value
case "pgscan_kswapd_dma32":
vmstat.PageScanKswapdDMA32 = value
case "pgscan_kswapd_normal":
vmstat.PageScanKswapdNormal = value
case "pgscan_kswapd_movable":
vmstat.PageScanKswapdMovable = value
case "pgscan_direct_dma":
vmstat.PageScanDirectDMA = value
case "pgscan_direct_dma32":
vmstat.PageScanDirectDMA32 = value
case "pgscan_direct_normal":
vmstat.PageScanDirectNormal = value
case "pgscan_direct_movable":
vmstat.PageScanDirectMovable = value
case "pgscan_direct_throttle":
vmstat.PageScanDirectThrottle = value
case "zone_reclaim_failed":
vmstat.ZoneReclaimFailed = value
case "pginodesteal":
vmstat.PageInodeSteal = value
case "slabs_scanned":
vmstat.SlabsScanned = value
case "kswapd_inodesteal":
vmstat.KswapdInodesteal = value
case "kswapd_low_wmark_hit_quickly":
vmstat.KswapdLowWatermarkHitQuickly = value
case "kswapd_high_wmark_hit_quickly":
vmstat.KswapdHighWatermarkHitQuickly = value
case "pageoutrun":
vmstat.PageoutRun = value
case "allocstall":
vmstat.AllocStall = value
case "pgrotated":
vmstat.PageRotated = value
case "drop_pagecache":
vmstat.DropPagecache = value
case "drop_slab":
vmstat.DropSlab = value
case "numa_pte_updates":
vmstat.NumaPteUpdates = value
case "numa_huge_pte_updates":
vmstat.NumaHugePteUpdates = value
case "numa_hint_faults":
vmstat.NumaHintFaults = value
case "numa_hint_faults_local":
vmstat.NumaHintFaults_local = value
case "numa_pages_migrated":
vmstat.NumaPagesMigrated = value
case "pgmigrate_success":
vmstat.PageMigrateSuccess = value
case "pgmigrate_fail":
vmstat.PageMigrateFail = value
case "compact_migrate_scanned":
vmstat.CompactMigrateScanned = value
case "compact_free_scanned":
vmstat.CompactFreeScanned = value
case "compact_isolated":
vmstat.CompactIsolated = value
case "compact_stall":
vmstat.CompactStall = value
case "compact_fail":
vmstat.CompactFail = value
case "compact_success":
vmstat.CompactSuccess = value
case "htlb_buddy_alloc_success":
vmstat.HtlbBuddyAllocSuccess = value
case "htlb_buddy_alloc_fail":
vmstat.HtlbBuddyAllocFail = value
case "unevictable_pgs_culled":
vmstat.UnevictablePagesCulled = value
case "unevictable_pgs_scanned":
vmstat.UnevictablePagesScanned = value
case "unevictable_pgs_rescued":
vmstat.UnevictablePagesRescued = value
case "unevictable_pgs_mlocked":
vmstat.UnevictablePagesMlocked = value
case "unevictable_pgs_munlocked":
vmstat.UnevictablePagesMunlocked = value
case "unevictable_pgs_cleared":
vmstat.UnevictablePagesCleared = value
case "unevictable_pgs_stranded":
vmstat.UnevictablePagesStranded = value
case "thp_fault_alloc":
vmstat.THPFaultAlloc = value
case "thp_fault_fallback":
vmstat.THPFaultFallback = value
case "thp_collapse_alloc":
vmstat.THPCollapseAlloc = value
case "thp_collapse_alloc_failed":
vmstat.THPCollapseAllocFailed = value
case "thp_split":
vmstat.THPSplit = value
case "thp_zero_page_alloc":
vmstat.THPZeroPageAlloc = value
case "thp_zero_page_alloc_failed":
vmstat.THPZeroPageAllocFailed = value
}
}
return &vmstat, nil
}

9
vendor/github.com/go-ole/go-ole/.travis.yml generated vendored Normal file
View File

@ -0,0 +1,9 @@
language: go
sudo: false
go:
- 1.1
- 1.2
- 1.3
- 1.4
- tip

49
vendor/github.com/go-ole/go-ole/ChangeLog.md generated vendored Normal file
View File

@ -0,0 +1,49 @@
# Version 1.x.x
* **Add more test cases and reference new test COM server project.** (Placeholder for future additions)
# Version 1.2.0-alphaX
**Minimum supported version is now Go 1.4. Go 1.1 support is deprecated, but should still build.**
* Added CI configuration for Travis-CI and AppVeyor.
* Added test InterfaceID and ClassID for the COM Test Server project.
* Added more inline documentation (#83).
* Added IEnumVARIANT implementation (#88).
* Added IEnumVARIANT test cases (#99, #100, #101).
* Added support for retrieving `time.Time` from VARIANT (#92).
* Added test case for IUnknown (#64).
* Added test case for IDispatch (#64).
* Added test cases for scalar variants (#64, #76).
# Version 1.1.1
* Fixes for Linux build.
* Fixes for Windows build.
# Version 1.1.0
The change to provide building on all platforms is a new feature. The increase in minor version reflects that and allows those who wish to stay on 1.0.x to continue to do so. Support for 1.0.x will be limited to bug fixes.
* Move GUID out of variables.go into its own file to make new documentation available.
* Move OleError out of ole.go into its own file to make new documentation available.
* Add documentation to utility functions.
* Add documentation to variant receiver functions.
* Add documentation to ole structures.
* Make variant available to other systems outside of Windows.
* Make OLE structures available to other systems outside of Windows.
## New Features
* Library should now be built on all platforms supported by Go. Library will NOOP on any platform that is not Windows.
* More functions are now documented and available on godoc.org.
# Version 1.0.1
1. Fix package references from repository location change.
# Version 1.0.0
This version is stable enough for use. The COM API is still incomplete, but provides enough functionality for accessing COM servers using IDispatch interface.
There is no changelog for this version. Check commits for history.

21
vendor/github.com/go-ole/go-ole/LICENSE generated vendored Normal file
View File

@ -0,0 +1,21 @@
The MIT License (MIT)
Copyright © 2013-2017 Yasuhiro Matsumoto, <mattn.jp@gmail.com>
Permission is hereby granted, free of charge, to any person obtaining a copy of
this software and associated documentation files (the “Software”), to deal in
the Software without restriction, including without limitation the rights to
use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
of the Software, and to permit persons to whom the Software is furnished to do
so, subject to the following conditions:
The above copyright notice and this permission notice shall be included in all
copies or substantial portions of the Software.
THE SOFTWARE IS PROVIDED “AS IS”, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
SOFTWARE.

46
vendor/github.com/go-ole/go-ole/README.md generated vendored Normal file
View File

@ -0,0 +1,46 @@
#Go OLE
[![Build status](https://ci.appveyor.com/api/projects/status/qr0u2sf7q43us9fj?svg=true)](https://ci.appveyor.com/project/jacobsantos/go-ole-jgs28)
[![Build Status](https://travis-ci.org/go-ole/go-ole.svg?branch=master)](https://travis-ci.org/go-ole/go-ole)
[![GoDoc](https://godoc.org/github.com/go-ole/go-ole?status.svg)](https://godoc.org/github.com/go-ole/go-ole)
Go bindings for Windows COM using shared libraries instead of cgo.
By Yasuhiro Matsumoto.
## Install
To experiment with go-ole, you can just compile and run the example program:
```
go get github.com/go-ole/go-ole
cd /path/to/go-ole/
go test
cd /path/to/go-ole/example/excel
go run excel.go
```
## Continuous Integration
Continuous integration configuration has been added for both Travis-CI and AppVeyor. You will have to add these to your own account for your fork in order for it to run.
**Travis-CI**
Travis-CI was added to check builds on Linux to ensure that `go get` works when cross building. Currently, Travis-CI is not used to test cross-building, but this may be changed in the future. It is also not currently possible to test the library on Linux, since COM API is specific to Windows and it is not currently possible to run a COM server on Linux or even connect to a remote COM server.
**AppVeyor**
AppVeyor is used to build on Windows using the (in-development) test COM server. It is currently only used to test the build and ensure that the code works on Windows. It will be used to register a COM server and then run the test cases based on the test COM server.
The tests currently do run and do pass and this should be maintained with commits.
##Versioning
Go OLE uses [semantic versioning](http://semver.org) for version numbers, which is similar to the version contract of the Go language. Which means that the major version will always maintain backwards compatibility with minor versions. Minor versions will only add new additions and changes. Fixes will always be in patch.
This contract should allow you to upgrade to new minor and patch versions without breakage or modifications to your existing code. Leave a ticket, if there is breakage, so that it could be fixed.
##LICENSE
Under the MIT License: http://mattn.mit-license.org/2013

54
vendor/github.com/go-ole/go-ole/appveyor.yml generated vendored Normal file
View File

@ -0,0 +1,54 @@
# Notes:
# - Minimal appveyor.yml file is an empty file. All sections are optional.
# - Indent each level of configuration with 2 spaces. Do not use tabs!
# - All section names are case-sensitive.
# - Section names should be unique on each level.
version: "1.3.0.{build}-alpha-{branch}"
os: Windows Server 2012 R2
branches:
only:
- master
- v1.2
- v1.1
- v1.0
skip_tags: true
clone_folder: c:\gopath\src\github.com\go-ole\go-ole
environment:
GOPATH: c:\gopath
matrix:
- GOARCH: amd64
GOVERSION: 1.5
GOROOT: c:\go
DOWNLOADPLATFORM: "x64"
install:
- choco install mingw
- SET PATH=c:\tools\mingw64\bin;%PATH%
# - Download COM Server
- ps: Start-FileDownload "https://github.com/go-ole/test-com-server/releases/download/v1.0.2/test-com-server-${env:DOWNLOADPLATFORM}.zip"
- 7z e test-com-server-%DOWNLOADPLATFORM%.zip -oc:\gopath\src\github.com\go-ole\go-ole > NUL
- c:\gopath\src\github.com\go-ole\go-ole\build\register-assembly.bat
# - set
- go version
- go env
- go get -u golang.org/x/tools/cmd/cover
- go get -u golang.org/x/tools/cmd/godoc
- go get -u golang.org/x/tools/cmd/stringer
build_script:
- cd c:\gopath\src\github.com\go-ole\go-ole
- go get -v -t ./...
- go build
- go test -v -cover ./...
# disable automatic tests
test: off
# disable deployment
deploy: off

329
vendor/github.com/go-ole/go-ole/com.go generated vendored Normal file
View File

@ -0,0 +1,329 @@
// +build windows
package ole
import (
"errors"
"syscall"
"time"
"unicode/utf16"
"unsafe"
)
var (
procCoInitialize, _ = modole32.FindProc("CoInitialize")
procCoInitializeEx, _ = modole32.FindProc("CoInitializeEx")
procCoUninitialize, _ = modole32.FindProc("CoUninitialize")
procCoCreateInstance, _ = modole32.FindProc("CoCreateInstance")
procCoTaskMemFree, _ = modole32.FindProc("CoTaskMemFree")
procCLSIDFromProgID, _ = modole32.FindProc("CLSIDFromProgID")
procCLSIDFromString, _ = modole32.FindProc("CLSIDFromString")
procStringFromCLSID, _ = modole32.FindProc("StringFromCLSID")
procStringFromIID, _ = modole32.FindProc("StringFromIID")
procIIDFromString, _ = modole32.FindProc("IIDFromString")
procGetUserDefaultLCID, _ = modkernel32.FindProc("GetUserDefaultLCID")
procCopyMemory, _ = modkernel32.FindProc("RtlMoveMemory")
procVariantInit, _ = modoleaut32.FindProc("VariantInit")
procVariantClear, _ = modoleaut32.FindProc("VariantClear")
procVariantTimeToSystemTime, _ = modoleaut32.FindProc("VariantTimeToSystemTime")
procSysAllocString, _ = modoleaut32.FindProc("SysAllocString")
procSysAllocStringLen, _ = modoleaut32.FindProc("SysAllocStringLen")
procSysFreeString, _ = modoleaut32.FindProc("SysFreeString")
procSysStringLen, _ = modoleaut32.FindProc("SysStringLen")
procCreateDispTypeInfo, _ = modoleaut32.FindProc("CreateDispTypeInfo")
procCreateStdDispatch, _ = modoleaut32.FindProc("CreateStdDispatch")
procGetActiveObject, _ = modoleaut32.FindProc("GetActiveObject")
procGetMessageW, _ = moduser32.FindProc("GetMessageW")
procDispatchMessageW, _ = moduser32.FindProc("DispatchMessageW")
)
// coInitialize initializes COM library on current thread.
//
// MSDN documentation suggests that this function should not be called. Call
// CoInitializeEx() instead. The reason has to do with threading and this
// function is only for single-threaded apartments.
//
// That said, most users of the library have gotten away with just this
// function. If you are experiencing threading issues, then use
// CoInitializeEx().
func coInitialize() (err error) {
// http://msdn.microsoft.com/en-us/library/windows/desktop/ms678543(v=vs.85).aspx
// Suggests that no value should be passed to CoInitialized.
// Could just be Call() since the parameter is optional. <-- Needs testing to be sure.
hr, _, _ := procCoInitialize.Call(uintptr(0))
if hr != 0 {
err = NewError(hr)
}
return
}
// coInitializeEx initializes COM library with concurrency model.
func coInitializeEx(coinit uint32) (err error) {
// http://msdn.microsoft.com/en-us/library/windows/desktop/ms695279(v=vs.85).aspx
// Suggests that the first parameter is not only optional but should always be NULL.
hr, _, _ := procCoInitializeEx.Call(uintptr(0), uintptr(coinit))
if hr != 0 {
err = NewError(hr)
}
return
}
// CoInitialize initializes COM library on current thread.
//
// MSDN documentation suggests that this function should not be called. Call
// CoInitializeEx() instead. The reason has to do with threading and this
// function is only for single-threaded apartments.
//
// That said, most users of the library have gotten away with just this
// function. If you are experiencing threading issues, then use
// CoInitializeEx().
func CoInitialize(p uintptr) (err error) {
// p is ignored and won't be used.
// Avoid any variable not used errors.
p = uintptr(0)
return coInitialize()
}
// CoInitializeEx initializes COM library with concurrency model.
func CoInitializeEx(p uintptr, coinit uint32) (err error) {
// Avoid any variable not used errors.
p = uintptr(0)
return coInitializeEx(coinit)
}
// CoUninitialize uninitializes COM Library.
func CoUninitialize() {
procCoUninitialize.Call()
}
// CoTaskMemFree frees memory pointer.
func CoTaskMemFree(memptr uintptr) {
procCoTaskMemFree.Call(memptr)
}
// CLSIDFromProgID retrieves Class Identifier with the given Program Identifier.
//
// The Programmatic Identifier must be registered, because it will be looked up
// in the Windows Registry. The registry entry has the following keys: CLSID,
// Insertable, Protocol and Shell
// (https://msdn.microsoft.com/en-us/library/dd542719(v=vs.85).aspx).
//
// programID identifies the class id with less precision and is not guaranteed
// to be unique. These are usually found in the registry under
// HKEY_LOCAL_MACHINE\SOFTWARE\Classes, usually with the format of
// "Program.Component.Version" with version being optional.
//
// CLSIDFromProgID in Windows API.
func CLSIDFromProgID(progId string) (clsid *GUID, err error) {
var guid GUID
lpszProgID := uintptr(unsafe.Pointer(syscall.StringToUTF16Ptr(progId)))
hr, _, _ := procCLSIDFromProgID.Call(lpszProgID, uintptr(unsafe.Pointer(&guid)))
if hr != 0 {
err = NewError(hr)
}
clsid = &guid
return
}
// CLSIDFromString retrieves Class ID from string representation.
//
// This is technically the string version of the GUID and will convert the
// string to object.
//
// CLSIDFromString in Windows API.
func CLSIDFromString(str string) (clsid *GUID, err error) {
var guid GUID
lpsz := uintptr(unsafe.Pointer(syscall.StringToUTF16Ptr(str)))
hr, _, _ := procCLSIDFromString.Call(lpsz, uintptr(unsafe.Pointer(&guid)))
if hr != 0 {
err = NewError(hr)
}
clsid = &guid
return
}
// StringFromCLSID returns GUID formated string from GUID object.
func StringFromCLSID(clsid *GUID) (str string, err error) {
var p *uint16
hr, _, _ := procStringFromCLSID.Call(uintptr(unsafe.Pointer(clsid)), uintptr(unsafe.Pointer(&p)))
if hr != 0 {
err = NewError(hr)
}
str = LpOleStrToString(p)
return
}
// IIDFromString returns GUID from program ID.
func IIDFromString(progId string) (clsid *GUID, err error) {
var guid GUID
lpsz := uintptr(unsafe.Pointer(syscall.StringToUTF16Ptr(progId)))
hr, _, _ := procIIDFromString.Call(lpsz, uintptr(unsafe.Pointer(&guid)))
if hr != 0 {
err = NewError(hr)
}
clsid = &guid
return
}
// StringFromIID returns GUID formatted string from GUID object.
func StringFromIID(iid *GUID) (str string, err error) {
var p *uint16
hr, _, _ := procStringFromIID.Call(uintptr(unsafe.Pointer(iid)), uintptr(unsafe.Pointer(&p)))
if hr != 0 {
err = NewError(hr)
}
str = LpOleStrToString(p)
return
}
// CreateInstance of single uninitialized object with GUID.
func CreateInstance(clsid *GUID, iid *GUID) (unk *IUnknown, err error) {
if iid == nil {
iid = IID_IUnknown
}
hr, _, _ := procCoCreateInstance.Call(
uintptr(unsafe.Pointer(clsid)),
0,
CLSCTX_SERVER,
uintptr(unsafe.Pointer(iid)),
uintptr(unsafe.Pointer(&unk)))
if hr != 0 {
err = NewError(hr)
}
return
}
// GetActiveObject retrieves pointer to active object.
func GetActiveObject(clsid *GUID, iid *GUID) (unk *IUnknown, err error) {
if iid == nil {
iid = IID_IUnknown
}
hr, _, _ := procGetActiveObject.Call(
uintptr(unsafe.Pointer(clsid)),
uintptr(unsafe.Pointer(iid)),
uintptr(unsafe.Pointer(&unk)))
if hr != 0 {
err = NewError(hr)
}
return
}
// VariantInit initializes variant.
func VariantInit(v *VARIANT) (err error) {
hr, _, _ := procVariantInit.Call(uintptr(unsafe.Pointer(v)))
if hr != 0 {
err = NewError(hr)
}
return
}
// VariantClear clears value in Variant settings to VT_EMPTY.
func VariantClear(v *VARIANT) (err error) {
hr, _, _ := procVariantClear.Call(uintptr(unsafe.Pointer(v)))
if hr != 0 {
err = NewError(hr)
}
return
}
// SysAllocString allocates memory for string and copies string into memory.
func SysAllocString(v string) (ss *int16) {
pss, _, _ := procSysAllocString.Call(uintptr(unsafe.Pointer(syscall.StringToUTF16Ptr(v))))
ss = (*int16)(unsafe.Pointer(pss))
return
}
// SysAllocStringLen copies up to length of given string returning pointer.
func SysAllocStringLen(v string) (ss *int16) {
utf16 := utf16.Encode([]rune(v + "\x00"))
ptr := &utf16[0]
pss, _, _ := procSysAllocStringLen.Call(uintptr(unsafe.Pointer(ptr)), uintptr(len(utf16)-1))
ss = (*int16)(unsafe.Pointer(pss))
return
}
// SysFreeString frees string system memory. This must be called with SysAllocString.
func SysFreeString(v *int16) (err error) {
hr, _, _ := procSysFreeString.Call(uintptr(unsafe.Pointer(v)))
if hr != 0 {
err = NewError(hr)
}
return
}
// SysStringLen is the length of the system allocated string.
func SysStringLen(v *int16) uint32 {
l, _, _ := procSysStringLen.Call(uintptr(unsafe.Pointer(v)))
return uint32(l)
}
// CreateStdDispatch provides default IDispatch implementation for IUnknown.
//
// This handles default IDispatch implementation for objects. It haves a few
// limitations with only supporting one language. It will also only return
// default exception codes.
func CreateStdDispatch(unk *IUnknown, v uintptr, ptinfo *IUnknown) (disp *IDispatch, err error) {
hr, _, _ := procCreateStdDispatch.Call(
uintptr(unsafe.Pointer(unk)),
v,
uintptr(unsafe.Pointer(ptinfo)),
uintptr(unsafe.Pointer(&disp)))
if hr != 0 {
err = NewError(hr)
}
return
}
// CreateDispTypeInfo provides default ITypeInfo implementation for IDispatch.
//
// This will not handle the full implementation of the interface.
func CreateDispTypeInfo(idata *INTERFACEDATA) (pptinfo *IUnknown, err error) {
hr, _, _ := procCreateDispTypeInfo.Call(
uintptr(unsafe.Pointer(idata)),
uintptr(GetUserDefaultLCID()),
uintptr(unsafe.Pointer(&pptinfo)))
if hr != 0 {
err = NewError(hr)
}
return
}
// copyMemory moves location of a block of memory.
func copyMemory(dest unsafe.Pointer, src unsafe.Pointer, length uint32) {
procCopyMemory.Call(uintptr(dest), uintptr(src), uintptr(length))
}
// GetUserDefaultLCID retrieves current user default locale.
func GetUserDefaultLCID() (lcid uint32) {
ret, _, _ := procGetUserDefaultLCID.Call()
lcid = uint32(ret)
return
}
// GetMessage in message queue from runtime.
//
// This function appears to block. PeekMessage does not block.
func GetMessage(msg *Msg, hwnd uint32, MsgFilterMin uint32, MsgFilterMax uint32) (ret int32, err error) {
r0, _, err := procGetMessageW.Call(uintptr(unsafe.Pointer(msg)), uintptr(hwnd), uintptr(MsgFilterMin), uintptr(MsgFilterMax))
ret = int32(r0)
return
}
// DispatchMessage to window procedure.
func DispatchMessage(msg *Msg) (ret int32) {
r0, _, _ := procDispatchMessageW.Call(uintptr(unsafe.Pointer(msg)))
ret = int32(r0)
return
}
// GetVariantDate converts COM Variant Time value to Go time.Time.
func GetVariantDate(value float64) (time.Time, error) {
var st syscall.Systemtime
r, _, _ := procVariantTimeToSystemTime.Call(uintptr(value), uintptr(unsafe.Pointer(&st)))
if r != 0 {
return time.Date(int(st.Year), time.Month(st.Month), int(st.Day), int(st.Hour), int(st.Minute), int(st.Second), int(st.Milliseconds/1000), time.UTC), nil
}
return time.Now(), errors.New("Could not convert to time, passing current time.")
}

174
vendor/github.com/go-ole/go-ole/com_func.go generated vendored Normal file
View File

@ -0,0 +1,174 @@
// +build !windows
package ole
import (
"time"
"unsafe"
)
// coInitialize initializes COM library on current thread.
//
// MSDN documentation suggests that this function should not be called. Call
// CoInitializeEx() instead. The reason has to do with threading and this
// function is only for single-threaded apartments.
//
// That said, most users of the library have gotten away with just this
// function. If you are experiencing threading issues, then use
// CoInitializeEx().
func coInitialize() error {
return NewError(E_NOTIMPL)
}
// coInitializeEx initializes COM library with concurrency model.
func coInitializeEx(coinit uint32) error {
return NewError(E_NOTIMPL)
}
// CoInitialize initializes COM library on current thread.
//
// MSDN documentation suggests that this function should not be called. Call
// CoInitializeEx() instead. The reason has to do with threading and this
// function is only for single-threaded apartments.
//
// That said, most users of the library have gotten away with just this
// function. If you are experiencing threading issues, then use
// CoInitializeEx().
func CoInitialize(p uintptr) error {
return NewError(E_NOTIMPL)
}
// CoInitializeEx initializes COM library with concurrency model.
func CoInitializeEx(p uintptr, coinit uint32) error {
return NewError(E_NOTIMPL)
}
// CoUninitialize uninitializes COM Library.
func CoUninitialize() {}
// CoTaskMemFree frees memory pointer.
func CoTaskMemFree(memptr uintptr) {}
// CLSIDFromProgID retrieves Class Identifier with the given Program Identifier.
//
// The Programmatic Identifier must be registered, because it will be looked up
// in the Windows Registry. The registry entry has the following keys: CLSID,
// Insertable, Protocol and Shell
// (https://msdn.microsoft.com/en-us/library/dd542719(v=vs.85).aspx).
//
// programID identifies the class id with less precision and is not guaranteed
// to be unique. These are usually found in the registry under
// HKEY_LOCAL_MACHINE\SOFTWARE\Classes, usually with the format of
// "Program.Component.Version" with version being optional.
//
// CLSIDFromProgID in Windows API.
func CLSIDFromProgID(progId string) (*GUID, error) {
return nil, NewError(E_NOTIMPL)
}
// CLSIDFromString retrieves Class ID from string representation.
//
// This is technically the string version of the GUID and will convert the
// string to object.
//
// CLSIDFromString in Windows API.
func CLSIDFromString(str string) (*GUID, error) {
return nil, NewError(E_NOTIMPL)
}
// StringFromCLSID returns GUID formated string from GUID object.
func StringFromCLSID(clsid *GUID) (string, error) {
return "", NewError(E_NOTIMPL)
}
// IIDFromString returns GUID from program ID.
func IIDFromString(progId string) (*GUID, error) {
return nil, NewError(E_NOTIMPL)
}
// StringFromIID returns GUID formatted string from GUID object.
func StringFromIID(iid *GUID) (string, error) {
return "", NewError(E_NOTIMPL)
}
// CreateInstance of single uninitialized object with GUID.
func CreateInstance(clsid *GUID, iid *GUID) (*IUnknown, error) {
return nil, NewError(E_NOTIMPL)
}
// GetActiveObject retrieves pointer to active object.
func GetActiveObject(clsid *GUID, iid *GUID) (*IUnknown, error) {
return nil, NewError(E_NOTIMPL)
}
// VariantInit initializes variant.
func VariantInit(v *VARIANT) error {
return NewError(E_NOTIMPL)
}
// VariantClear clears value in Variant settings to VT_EMPTY.
func VariantClear(v *VARIANT) error {
return NewError(E_NOTIMPL)
}
// SysAllocString allocates memory for string and copies string into memory.
func SysAllocString(v string) *int16 {
u := int16(0)
return &u
}
// SysAllocStringLen copies up to length of given string returning pointer.
func SysAllocStringLen(v string) *int16 {
u := int16(0)
return &u
}
// SysFreeString frees string system memory. This must be called with SysAllocString.
func SysFreeString(v *int16) error {
return NewError(E_NOTIMPL)
}
// SysStringLen is the length of the system allocated string.
func SysStringLen(v *int16) uint32 {
return uint32(0)
}
// CreateStdDispatch provides default IDispatch implementation for IUnknown.
//
// This handles default IDispatch implementation for objects. It haves a few
// limitations with only supporting one language. It will also only return
// default exception codes.
func CreateStdDispatch(unk *IUnknown, v uintptr, ptinfo *IUnknown) (*IDispatch, error) {
return nil, NewError(E_NOTIMPL)
}
// CreateDispTypeInfo provides default ITypeInfo implementation for IDispatch.
//
// This will not handle the full implementation of the interface.
func CreateDispTypeInfo(idata *INTERFACEDATA) (*IUnknown, error) {
return nil, NewError(E_NOTIMPL)
}
// copyMemory moves location of a block of memory.
func copyMemory(dest unsafe.Pointer, src unsafe.Pointer, length uint32) {}
// GetUserDefaultLCID retrieves current user default locale.
func GetUserDefaultLCID() uint32 {
return uint32(0)
}
// GetMessage in message queue from runtime.
//
// This function appears to block. PeekMessage does not block.
func GetMessage(msg *Msg, hwnd uint32, MsgFilterMin uint32, MsgFilterMax uint32) (int32, error) {
return int32(0), NewError(E_NOTIMPL)
}
// DispatchMessage to window procedure.
func DispatchMessage(msg *Msg) int32 {
return int32(0)
}
func GetVariantDate(value float64) (time.Time, error) {
return time.Now(), NewError(E_NOTIMPL)
}

192
vendor/github.com/go-ole/go-ole/connect.go generated vendored Normal file
View File

@ -0,0 +1,192 @@
package ole
// Connection contains IUnknown for fluent interface interaction.
//
// Deprecated. Use oleutil package instead.
type Connection struct {
Object *IUnknown // Access COM
}
// Initialize COM.
func (*Connection) Initialize() (err error) {
return coInitialize()
}
// Uninitialize COM.
func (*Connection) Uninitialize() {
CoUninitialize()
}
// Create IUnknown object based first on ProgId and then from String.
func (c *Connection) Create(progId string) (err error) {
var clsid *GUID
clsid, err = CLSIDFromProgID(progId)
if err != nil {
clsid, err = CLSIDFromString(progId)
if err != nil {
return
}
}
unknown, err := CreateInstance(clsid, IID_IUnknown)
if err != nil {
return
}
c.Object = unknown
return
}
// Release IUnknown object.
func (c *Connection) Release() {
c.Object.Release()
}
// Load COM object from list of programIDs or strings.
func (c *Connection) Load(names ...string) (errors []error) {
var tempErrors []error = make([]error, len(names))
var numErrors int = 0
for _, name := range names {
err := c.Create(name)
if err != nil {
tempErrors = append(tempErrors, err)
numErrors += 1
continue
}
break
}
copy(errors, tempErrors[0:numErrors])
return
}
// Dispatch returns Dispatch object.
func (c *Connection) Dispatch() (object *Dispatch, err error) {
dispatch, err := c.Object.QueryInterface(IID_IDispatch)
if err != nil {
return
}
object = &Dispatch{dispatch}
return
}
// Dispatch stores IDispatch object.
type Dispatch struct {
Object *IDispatch // Dispatch object.
}
// Call method on IDispatch with parameters.
func (d *Dispatch) Call(method string, params ...interface{}) (result *VARIANT, err error) {
id, err := d.GetId(method)
if err != nil {
return
}
result, err = d.Invoke(id, DISPATCH_METHOD, params)
return
}
// MustCall method on IDispatch with parameters.
func (d *Dispatch) MustCall(method string, params ...interface{}) (result *VARIANT) {
id, err := d.GetId(method)
if err != nil {
panic(err)
}
result, err = d.Invoke(id, DISPATCH_METHOD, params)
if err != nil {
panic(err)
}
return
}
// Get property on IDispatch with parameters.
func (d *Dispatch) Get(name string, params ...interface{}) (result *VARIANT, err error) {
id, err := d.GetId(name)
if err != nil {
return
}
result, err = d.Invoke(id, DISPATCH_PROPERTYGET, params)
return
}
// MustGet property on IDispatch with parameters.
func (d *Dispatch) MustGet(name string, params ...interface{}) (result *VARIANT) {
id, err := d.GetId(name)
if err != nil {
panic(err)
}
result, err = d.Invoke(id, DISPATCH_PROPERTYGET, params)
if err != nil {
panic(err)
}
return
}
// Set property on IDispatch with parameters.
func (d *Dispatch) Set(name string, params ...interface{}) (result *VARIANT, err error) {
id, err := d.GetId(name)
if err != nil {
return
}
result, err = d.Invoke(id, DISPATCH_PROPERTYPUT, params)
return
}
// MustSet property on IDispatch with parameters.
func (d *Dispatch) MustSet(name string, params ...interface{}) (result *VARIANT) {
id, err := d.GetId(name)
if err != nil {
panic(err)
}
result, err = d.Invoke(id, DISPATCH_PROPERTYPUT, params)
if err != nil {
panic(err)
}
return
}
// GetId retrieves ID of name on IDispatch.
func (d *Dispatch) GetId(name string) (id int32, err error) {
var dispid []int32
dispid, err = d.Object.GetIDsOfName([]string{name})
if err != nil {
return
}
id = dispid[0]
return
}
// GetIds retrieves all IDs of names on IDispatch.
func (d *Dispatch) GetIds(names ...string) (dispid []int32, err error) {
dispid, err = d.Object.GetIDsOfName(names)
return
}
// Invoke IDispatch on DisplayID of dispatch type with parameters.
//
// There have been problems where if send cascading params..., it would error
// out because the parameters would be empty.
func (d *Dispatch) Invoke(id int32, dispatch int16, params []interface{}) (result *VARIANT, err error) {
if len(params) < 1 {
result, err = d.Object.Invoke(id, dispatch)
} else {
result, err = d.Object.Invoke(id, dispatch, params...)
}
return
}
// Release IDispatch object.
func (d *Dispatch) Release() {
d.Object.Release()
}
// Connect initializes COM and attempts to load IUnknown based on given names.
func Connect(names ...string) (connection *Connection) {
connection.Initialize()
connection.Load(names...)
return
}

153
vendor/github.com/go-ole/go-ole/constants.go generated vendored Normal file
View File

@ -0,0 +1,153 @@
package ole
const (
CLSCTX_INPROC_SERVER = 1
CLSCTX_INPROC_HANDLER = 2
CLSCTX_LOCAL_SERVER = 4
CLSCTX_INPROC_SERVER16 = 8
CLSCTX_REMOTE_SERVER = 16
CLSCTX_ALL = CLSCTX_INPROC_SERVER | CLSCTX_INPROC_HANDLER | CLSCTX_LOCAL_SERVER
CLSCTX_INPROC = CLSCTX_INPROC_SERVER | CLSCTX_INPROC_HANDLER
CLSCTX_SERVER = CLSCTX_INPROC_SERVER | CLSCTX_LOCAL_SERVER | CLSCTX_REMOTE_SERVER
)
const (
COINIT_APARTMENTTHREADED = 0x2
COINIT_MULTITHREADED = 0x0
COINIT_DISABLE_OLE1DDE = 0x4
COINIT_SPEED_OVER_MEMORY = 0x8
)
const (
DISPATCH_METHOD = 1
DISPATCH_PROPERTYGET = 2
DISPATCH_PROPERTYPUT = 4
DISPATCH_PROPERTYPUTREF = 8
)
const (
S_OK = 0x00000000
E_UNEXPECTED = 0x8000FFFF
E_NOTIMPL = 0x80004001
E_OUTOFMEMORY = 0x8007000E
E_INVALIDARG = 0x80070057
E_NOINTERFACE = 0x80004002
E_POINTER = 0x80004003
E_HANDLE = 0x80070006
E_ABORT = 0x80004004
E_FAIL = 0x80004005
E_ACCESSDENIED = 0x80070005
E_PENDING = 0x8000000A
CO_E_CLASSSTRING = 0x800401F3
)
const (
CC_FASTCALL = iota
CC_CDECL
CC_MSCPASCAL
CC_PASCAL = CC_MSCPASCAL
CC_MACPASCAL
CC_STDCALL
CC_FPFASTCALL
CC_SYSCALL
CC_MPWCDECL
CC_MPWPASCAL
CC_MAX = CC_MPWPASCAL
)
type VT uint16
const (
VT_EMPTY VT = 0x0
VT_NULL VT = 0x1
VT_I2 VT = 0x2
VT_I4 VT = 0x3
VT_R4 VT = 0x4
VT_R8 VT = 0x5
VT_CY VT = 0x6
VT_DATE VT = 0x7
VT_BSTR VT = 0x8
VT_DISPATCH VT = 0x9
VT_ERROR VT = 0xa
VT_BOOL VT = 0xb
VT_VARIANT VT = 0xc
VT_UNKNOWN VT = 0xd
VT_DECIMAL VT = 0xe
VT_I1 VT = 0x10
VT_UI1 VT = 0x11
VT_UI2 VT = 0x12
VT_UI4 VT = 0x13
VT_I8 VT = 0x14
VT_UI8 VT = 0x15
VT_INT VT = 0x16
VT_UINT VT = 0x17
VT_VOID VT = 0x18
VT_HRESULT VT = 0x19
VT_PTR VT = 0x1a
VT_SAFEARRAY VT = 0x1b
VT_CARRAY VT = 0x1c
VT_USERDEFINED VT = 0x1d
VT_LPSTR VT = 0x1e
VT_LPWSTR VT = 0x1f
VT_RECORD VT = 0x24
VT_INT_PTR VT = 0x25
VT_UINT_PTR VT = 0x26
VT_FILETIME VT = 0x40
VT_BLOB VT = 0x41
VT_STREAM VT = 0x42
VT_STORAGE VT = 0x43
VT_STREAMED_OBJECT VT = 0x44
VT_STORED_OBJECT VT = 0x45
VT_BLOB_OBJECT VT = 0x46
VT_CF VT = 0x47
VT_CLSID VT = 0x48
VT_BSTR_BLOB VT = 0xfff
VT_VECTOR VT = 0x1000
VT_ARRAY VT = 0x2000
VT_BYREF VT = 0x4000
VT_RESERVED VT = 0x8000
VT_ILLEGAL VT = 0xffff
VT_ILLEGALMASKED VT = 0xfff
VT_TYPEMASK VT = 0xfff
)
const (
DISPID_UNKNOWN = -1
DISPID_VALUE = 0
DISPID_PROPERTYPUT = -3
DISPID_NEWENUM = -4
DISPID_EVALUATE = -5
DISPID_CONSTRUCTOR = -6
DISPID_DESTRUCTOR = -7
DISPID_COLLECT = -8
)
const (
TKIND_ENUM = 1
TKIND_RECORD = 2
TKIND_MODULE = 3
TKIND_INTERFACE = 4
TKIND_DISPATCH = 5
TKIND_COCLASS = 6
TKIND_ALIAS = 7
TKIND_UNION = 8
TKIND_MAX = 9
)
// Safe Array Feature Flags
const (
FADF_AUTO = 0x0001
FADF_STATIC = 0x0002
FADF_EMBEDDED = 0x0004
FADF_FIXEDSIZE = 0x0010
FADF_RECORD = 0x0020
FADF_HAVEIID = 0x0040
FADF_HAVEVARTYPE = 0x0080
FADF_BSTR = 0x0100
FADF_UNKNOWN = 0x0200
FADF_DISPATCH = 0x0400
FADF_VARIANT = 0x0800
FADF_RESERVED = 0xF008
)

51
vendor/github.com/go-ole/go-ole/error.go generated vendored Normal file
View File

@ -0,0 +1,51 @@
package ole
// OleError stores COM errors.
type OleError struct {
hr uintptr
description string
subError error
}
// NewError creates new error with HResult.
func NewError(hr uintptr) *OleError {
return &OleError{hr: hr}
}
// NewErrorWithDescription creates new COM error with HResult and description.
func NewErrorWithDescription(hr uintptr, description string) *OleError {
return &OleError{hr: hr, description: description}
}
// NewErrorWithSubError creates new COM error with parent error.
func NewErrorWithSubError(hr uintptr, description string, err error) *OleError {
return &OleError{hr: hr, description: description, subError: err}
}
// Code is the HResult.
func (v *OleError) Code() uintptr {
return uintptr(v.hr)
}
// String description, either manually set or format message with error code.
func (v *OleError) String() string {
if v.description != "" {
return errstr(int(v.hr)) + " (" + v.description + ")"
}
return errstr(int(v.hr))
}
// Error implements error interface.
func (v *OleError) Error() string {
return v.String()
}
// Description retrieves error summary, if there is one.
func (v *OleError) Description() string {
return v.description
}
// SubError returns parent error, if there is one.
func (v *OleError) SubError() error {
return v.subError
}

8
vendor/github.com/go-ole/go-ole/error_func.go generated vendored Normal file
View File

@ -0,0 +1,8 @@
// +build !windows
package ole
// errstr converts error code to string.
func errstr(errno int) string {
return ""
}

24
vendor/github.com/go-ole/go-ole/error_windows.go generated vendored Normal file
View File

@ -0,0 +1,24 @@
// +build windows
package ole
import (
"fmt"
"syscall"
"unicode/utf16"
)
// errstr converts error code to string.
func errstr(errno int) string {
// ask windows for the remaining errors
var flags uint32 = syscall.FORMAT_MESSAGE_FROM_SYSTEM | syscall.FORMAT_MESSAGE_ARGUMENT_ARRAY | syscall.FORMAT_MESSAGE_IGNORE_INSERTS
b := make([]uint16, 300)
n, err := syscall.FormatMessage(flags, 0, uint32(errno), 0, b, nil)
if err != nil {
return fmt.Sprintf("error %d (FormatMessage failed with: %v)", errno, err)
}
// trim terminating \r and \n
for ; n > 0 && (b[n-1] == '\n' || b[n-1] == '\r'); n-- {
}
return string(utf16.Decode(b[:n]))
}

284
vendor/github.com/go-ole/go-ole/guid.go generated vendored Normal file
View File

@ -0,0 +1,284 @@
package ole
var (
// IID_NULL is null Interface ID, used when no other Interface ID is known.
IID_NULL = NewGUID("{00000000-0000-0000-0000-000000000000}")
// IID_IUnknown is for IUnknown interfaces.
IID_IUnknown = NewGUID("{00000000-0000-0000-C000-000000000046}")
// IID_IDispatch is for IDispatch interfaces.
IID_IDispatch = NewGUID("{00020400-0000-0000-C000-000000000046}")
// IID_IEnumVariant is for IEnumVariant interfaces
IID_IEnumVariant = NewGUID("{00020404-0000-0000-C000-000000000046}")
// IID_IConnectionPointContainer is for IConnectionPointContainer interfaces.
IID_IConnectionPointContainer = NewGUID("{B196B284-BAB4-101A-B69C-00AA00341D07}")
// IID_IConnectionPoint is for IConnectionPoint interfaces.
IID_IConnectionPoint = NewGUID("{B196B286-BAB4-101A-B69C-00AA00341D07}")
// IID_IInspectable is for IInspectable interfaces.
IID_IInspectable = NewGUID("{AF86E2E0-B12D-4C6A-9C5A-D7AA65101E90}")
// IID_IProvideClassInfo is for IProvideClassInfo interfaces.
IID_IProvideClassInfo = NewGUID("{B196B283-BAB4-101A-B69C-00AA00341D07}")
)
// These are for testing and not part of any library.
var (
// IID_ICOMTestString is for ICOMTestString interfaces.
//
// {E0133EB4-C36F-469A-9D3D-C66B84BE19ED}
IID_ICOMTestString = NewGUID("{E0133EB4-C36F-469A-9D3D-C66B84BE19ED}")
// IID_ICOMTestInt8 is for ICOMTestInt8 interfaces.
//
// {BEB06610-EB84-4155-AF58-E2BFF53680B4}
IID_ICOMTestInt8 = NewGUID("{BEB06610-EB84-4155-AF58-E2BFF53680B4}")
// IID_ICOMTestInt16 is for ICOMTestInt16 interfaces.
//
// {DAA3F9FA-761E-4976-A860-8364CE55F6FC}
IID_ICOMTestInt16 = NewGUID("{DAA3F9FA-761E-4976-A860-8364CE55F6FC}")
// IID_ICOMTestInt32 is for ICOMTestInt32 interfaces.
//
// {E3DEDEE7-38A2-4540-91D1-2EEF1D8891B0}
IID_ICOMTestInt32 = NewGUID("{E3DEDEE7-38A2-4540-91D1-2EEF1D8891B0}")
// IID_ICOMTestInt64 is for ICOMTestInt64 interfaces.
//
// {8D437CBC-B3ED-485C-BC32-C336432A1623}
IID_ICOMTestInt64 = NewGUID("{8D437CBC-B3ED-485C-BC32-C336432A1623}")
// IID_ICOMTestFloat is for ICOMTestFloat interfaces.
//
// {BF1ED004-EA02-456A-AA55-2AC8AC6B054C}
IID_ICOMTestFloat = NewGUID("{BF1ED004-EA02-456A-AA55-2AC8AC6B054C}")
// IID_ICOMTestDouble is for ICOMTestDouble interfaces.
//
// {BF908A81-8687-4E93-999F-D86FAB284BA0}
IID_ICOMTestDouble = NewGUID("{BF908A81-8687-4E93-999F-D86FAB284BA0}")
// IID_ICOMTestBoolean is for ICOMTestBoolean interfaces.
//
// {D530E7A6-4EE8-40D1-8931-3D63B8605010}
IID_ICOMTestBoolean = NewGUID("{D530E7A6-4EE8-40D1-8931-3D63B8605010}")
// IID_ICOMEchoTestObject is for ICOMEchoTestObject interfaces.
//
// {6485B1EF-D780-4834-A4FE-1EBB51746CA3}
IID_ICOMEchoTestObject = NewGUID("{6485B1EF-D780-4834-A4FE-1EBB51746CA3}")
// IID_ICOMTestTypes is for ICOMTestTypes interfaces.
//
// {CCA8D7AE-91C0-4277-A8B3-FF4EDF28D3C0}
IID_ICOMTestTypes = NewGUID("{CCA8D7AE-91C0-4277-A8B3-FF4EDF28D3C0}")
// CLSID_COMEchoTestObject is for COMEchoTestObject class.
//
// {3C24506A-AE9E-4D50-9157-EF317281F1B0}
CLSID_COMEchoTestObject = NewGUID("{3C24506A-AE9E-4D50-9157-EF317281F1B0}")
// CLSID_COMTestScalarClass is for COMTestScalarClass class.
//
// {865B85C5-0334-4AC6-9EF6-AACEC8FC5E86}
CLSID_COMTestScalarClass = NewGUID("{865B85C5-0334-4AC6-9EF6-AACEC8FC5E86}")
)
const hextable = "0123456789ABCDEF"
const emptyGUID = "{00000000-0000-0000-0000-000000000000}"
// GUID is Windows API specific GUID type.
//
// This exists to match Windows GUID type for direct passing for COM.
// Format is in xxxxxxxx-xxxx-xxxx-xxxxxxxxxxxxxxxx.
type GUID struct {
Data1 uint32
Data2 uint16
Data3 uint16
Data4 [8]byte
}
// NewGUID converts the given string into a globally unique identifier that is
// compliant with the Windows API.
//
// The supplied string may be in any of these formats:
//
// XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
// XXXXXXXX-XXXX-XXXX-XXXX-XXXXXXXXXXXX
// {XXXXXXXX-XXXX-XXXX-XXXX-XXXXXXXXXXXX}
//
// The conversion of the supplied string is not case-sensitive.
func NewGUID(guid string) *GUID {
d := []byte(guid)
var d1, d2, d3, d4a, d4b []byte
switch len(d) {
case 38:
if d[0] != '{' || d[37] != '}' {
return nil
}
d = d[1:37]
fallthrough
case 36:
if d[8] != '-' || d[13] != '-' || d[18] != '-' || d[23] != '-' {
return nil
}
d1 = d[0:8]
d2 = d[9:13]
d3 = d[14:18]
d4a = d[19:23]
d4b = d[24:36]
case 32:
d1 = d[0:8]
d2 = d[8:12]
d3 = d[12:16]
d4a = d[16:20]
d4b = d[20:32]
default:
return nil
}
var g GUID
var ok1, ok2, ok3, ok4 bool
g.Data1, ok1 = decodeHexUint32(d1)
g.Data2, ok2 = decodeHexUint16(d2)
g.Data3, ok3 = decodeHexUint16(d3)
g.Data4, ok4 = decodeHexByte64(d4a, d4b)
if ok1 && ok2 && ok3 && ok4 {
return &g
}
return nil
}
func decodeHexUint32(src []byte) (value uint32, ok bool) {
var b1, b2, b3, b4 byte
var ok1, ok2, ok3, ok4 bool
b1, ok1 = decodeHexByte(src[0], src[1])
b2, ok2 = decodeHexByte(src[2], src[3])
b3, ok3 = decodeHexByte(src[4], src[5])
b4, ok4 = decodeHexByte(src[6], src[7])
value = (uint32(b1) << 24) | (uint32(b2) << 16) | (uint32(b3) << 8) | uint32(b4)
ok = ok1 && ok2 && ok3 && ok4
return
}
func decodeHexUint16(src []byte) (value uint16, ok bool) {
var b1, b2 byte
var ok1, ok2 bool
b1, ok1 = decodeHexByte(src[0], src[1])
b2, ok2 = decodeHexByte(src[2], src[3])
value = (uint16(b1) << 8) | uint16(b2)
ok = ok1 && ok2
return
}
func decodeHexByte64(s1 []byte, s2 []byte) (value [8]byte, ok bool) {
var ok1, ok2, ok3, ok4, ok5, ok6, ok7, ok8 bool
value[0], ok1 = decodeHexByte(s1[0], s1[1])
value[1], ok2 = decodeHexByte(s1[2], s1[3])
value[2], ok3 = decodeHexByte(s2[0], s2[1])
value[3], ok4 = decodeHexByte(s2[2], s2[3])
value[4], ok5 = decodeHexByte(s2[4], s2[5])
value[5], ok6 = decodeHexByte(s2[6], s2[7])
value[6], ok7 = decodeHexByte(s2[8], s2[9])
value[7], ok8 = decodeHexByte(s2[10], s2[11])
ok = ok1 && ok2 && ok3 && ok4 && ok5 && ok6 && ok7 && ok8
return
}
func decodeHexByte(c1, c2 byte) (value byte, ok bool) {
var n1, n2 byte
var ok1, ok2 bool
n1, ok1 = decodeHexChar(c1)
n2, ok2 = decodeHexChar(c2)
value = (n1 << 4) | n2
ok = ok1 && ok2
return
}
func decodeHexChar(c byte) (byte, bool) {
switch {
case '0' <= c && c <= '9':
return c - '0', true
case 'a' <= c && c <= 'f':
return c - 'a' + 10, true
case 'A' <= c && c <= 'F':
return c - 'A' + 10, true
}
return 0, false
}
// String converts the GUID to string form. It will adhere to this pattern:
//
// {XXXXXXXX-XXXX-XXXX-XXXX-XXXXXXXXXXXX}
//
// If the GUID is nil, the string representation of an empty GUID is returned:
//
// {00000000-0000-0000-0000-000000000000}
func (guid *GUID) String() string {
if guid == nil {
return emptyGUID
}
var c [38]byte
c[0] = '{'
putUint32Hex(c[1:9], guid.Data1)
c[9] = '-'
putUint16Hex(c[10:14], guid.Data2)
c[14] = '-'
putUint16Hex(c[15:19], guid.Data3)
c[19] = '-'
putByteHex(c[20:24], guid.Data4[0:2])
c[24] = '-'
putByteHex(c[25:37], guid.Data4[2:8])
c[37] = '}'
return string(c[:])
}
func putUint32Hex(b []byte, v uint32) {
b[0] = hextable[byte(v>>24)>>4]
b[1] = hextable[byte(v>>24)&0x0f]
b[2] = hextable[byte(v>>16)>>4]
b[3] = hextable[byte(v>>16)&0x0f]
b[4] = hextable[byte(v>>8)>>4]
b[5] = hextable[byte(v>>8)&0x0f]
b[6] = hextable[byte(v)>>4]
b[7] = hextable[byte(v)&0x0f]
}
func putUint16Hex(b []byte, v uint16) {
b[0] = hextable[byte(v>>8)>>4]
b[1] = hextable[byte(v>>8)&0x0f]
b[2] = hextable[byte(v)>>4]
b[3] = hextable[byte(v)&0x0f]
}
func putByteHex(dst, src []byte) {
for i := 0; i < len(src); i++ {
dst[i*2] = hextable[src[i]>>4]
dst[i*2+1] = hextable[src[i]&0x0f]
}
}
// IsEqualGUID compares two GUID.
//
// Not constant time comparison.
func IsEqualGUID(guid1 *GUID, guid2 *GUID) bool {
return guid1.Data1 == guid2.Data1 &&
guid1.Data2 == guid2.Data2 &&
guid1.Data3 == guid2.Data3 &&
guid1.Data4[0] == guid2.Data4[0] &&
guid1.Data4[1] == guid2.Data4[1] &&
guid1.Data4[2] == guid2.Data4[2] &&
guid1.Data4[3] == guid2.Data4[3] &&
guid1.Data4[4] == guid2.Data4[4] &&
guid1.Data4[5] == guid2.Data4[5] &&
guid1.Data4[6] == guid2.Data4[6] &&
guid1.Data4[7] == guid2.Data4[7]
}

20
vendor/github.com/go-ole/go-ole/iconnectionpoint.go generated vendored Normal file
View File

@ -0,0 +1,20 @@
package ole
import "unsafe"
type IConnectionPoint struct {
IUnknown
}
type IConnectionPointVtbl struct {
IUnknownVtbl
GetConnectionInterface uintptr
GetConnectionPointContainer uintptr
Advise uintptr
Unadvise uintptr
EnumConnections uintptr
}
func (v *IConnectionPoint) VTable() *IConnectionPointVtbl {
return (*IConnectionPointVtbl)(unsafe.Pointer(v.RawVTable))
}

21
vendor/github.com/go-ole/go-ole/iconnectionpoint_func.go generated vendored Normal file
View File

@ -0,0 +1,21 @@
// +build !windows
package ole
import "unsafe"
func (v *IConnectionPoint) GetConnectionInterface(piid **GUID) int32 {
return int32(0)
}
func (v *IConnectionPoint) Advise(unknown *IUnknown) (uint32, error) {
return uint32(0), NewError(E_NOTIMPL)
}
func (v *IConnectionPoint) Unadvise(cookie uint32) error {
return NewError(E_NOTIMPL)
}
func (v *IConnectionPoint) EnumConnections(p *unsafe.Pointer) (err error) {
return NewError(E_NOTIMPL)
}

43
vendor/github.com/go-ole/go-ole/iconnectionpoint_windows.go generated vendored Normal file
View File

@ -0,0 +1,43 @@
// +build windows
package ole
import (
"syscall"
"unsafe"
)
func (v *IConnectionPoint) GetConnectionInterface(piid **GUID) int32 {
// XXX: This doesn't look like it does what it's supposed to
return release((*IUnknown)(unsafe.Pointer(v)))
}
func (v *IConnectionPoint) Advise(unknown *IUnknown) (cookie uint32, err error) {
hr, _, _ := syscall.Syscall(
v.VTable().Advise,
3,
uintptr(unsafe.Pointer(v)),
uintptr(unsafe.Pointer(unknown)),
uintptr(unsafe.Pointer(&cookie)))
if hr != 0 {
err = NewError(hr)
}
return
}
func (v *IConnectionPoint) Unadvise(cookie uint32) (err error) {
hr, _, _ := syscall.Syscall(
v.VTable().Unadvise,
2,
uintptr(unsafe.Pointer(v)),
uintptr(cookie),
0)
if hr != 0 {
err = NewError(hr)
}
return
}
func (v *IConnectionPoint) EnumConnections(p *unsafe.Pointer) error {
return NewError(E_NOTIMPL)
}

17
vendor/github.com/go-ole/go-ole/iconnectionpointcontainer.go generated vendored Normal file
View File

@ -0,0 +1,17 @@
package ole
import "unsafe"
type IConnectionPointContainer struct {
IUnknown
}
type IConnectionPointContainerVtbl struct {
IUnknownVtbl
EnumConnectionPoints uintptr
FindConnectionPoint uintptr
}
func (v *IConnectionPointContainer) VTable() *IConnectionPointContainerVtbl {
return (*IConnectionPointContainerVtbl)(unsafe.Pointer(v.RawVTable))
}

11
vendor/github.com/go-ole/go-ole/iconnectionpointcontainer_func.go generated vendored Normal file
View File

@ -0,0 +1,11 @@
// +build !windows
package ole
func (v *IConnectionPointContainer) EnumConnectionPoints(points interface{}) error {
return NewError(E_NOTIMPL)
}
func (v *IConnectionPointContainer) FindConnectionPoint(iid *GUID, point **IConnectionPoint) error {
return NewError(E_NOTIMPL)
}

25
vendor/github.com/go-ole/go-ole/iconnectionpointcontainer_windows.go generated vendored Normal file
View File

@ -0,0 +1,25 @@
// +build windows
package ole
import (
"syscall"
"unsafe"
)
func (v *IConnectionPointContainer) EnumConnectionPoints(points interface{}) error {
return NewError(E_NOTIMPL)
}
func (v *IConnectionPointContainer) FindConnectionPoint(iid *GUID, point **IConnectionPoint) (err error) {
hr, _, _ := syscall.Syscall(
v.VTable().FindConnectionPoint,
3,
uintptr(unsafe.Pointer(v)),
uintptr(unsafe.Pointer(iid)),
uintptr(unsafe.Pointer(point)))
if hr != 0 {
err = NewError(hr)
}
return
}

94
vendor/github.com/go-ole/go-ole/idispatch.go generated vendored Normal file
View File

@ -0,0 +1,94 @@
package ole
import "unsafe"
type IDispatch struct {
IUnknown
}
type IDispatchVtbl struct {
IUnknownVtbl
GetTypeInfoCount uintptr
GetTypeInfo uintptr
GetIDsOfNames uintptr
Invoke uintptr
}
func (v *IDispatch) VTable() *IDispatchVtbl {
return (*IDispatchVtbl)(unsafe.Pointer(v.RawVTable))
}
func (v *IDispatch) GetIDsOfName(names []string) (dispid []int32, err error) {
dispid, err = getIDsOfName(v, names)
return
}
func (v *IDispatch) Invoke(dispid int32, dispatch int16, params ...interface{}) (result *VARIANT, err error) {
result, err = invoke(v, dispid, dispatch, params...)
return
}
func (v *IDispatch) GetTypeInfoCount() (c uint32, err error) {
c, err = getTypeInfoCount(v)
return
}
func (v *IDispatch) GetTypeInfo() (tinfo *ITypeInfo, err error) {
tinfo, err = getTypeInfo(v)
return
}
// GetSingleIDOfName is a helper that returns single display ID for IDispatch name.
//
// This replaces the common pattern of attempting to get a single name from the list of available
// IDs. It gives the first ID, if it is available.
func (v *IDispatch) GetSingleIDOfName(name string) (displayID int32, err error) {
var displayIDs []int32
displayIDs, err = v.GetIDsOfName([]string{name})
if err != nil {
return
}
displayID = displayIDs[0]
return
}
// InvokeWithOptionalArgs accepts arguments as an array, works like Invoke.
//
// Accepts name and will attempt to retrieve Display ID to pass to Invoke.
//
// Passing params as an array is a workaround that could be fixed in later versions of Go that
// prevent passing empty params. During testing it was discovered that this is an acceptable way of
// getting around not being able to pass params normally.
func (v *IDispatch) InvokeWithOptionalArgs(name string, dispatch int16, params []interface{}) (result *VARIANT, err error) {
displayID, err := v.GetSingleIDOfName(name)
if err != nil {
return
}
if len(params) < 1 {
result, err = v.Invoke(displayID, dispatch)
} else {
result, err = v.Invoke(displayID, dispatch, params...)
}
return
}
// CallMethod invokes named function with arguments on object.
func (v *IDispatch) CallMethod(name string, params ...interface{}) (*VARIANT, error) {
return v.InvokeWithOptionalArgs(name, DISPATCH_METHOD, params)
}
// GetProperty retrieves the property with the name with the ability to pass arguments.
//
// Most of the time you will not need to pass arguments as most objects do not allow for this
// feature. Or at least, should not allow for this feature. Some servers don't follow best practices
// and this is provided for those edge cases.
func (v *IDispatch) GetProperty(name string, params ...interface{}) (*VARIANT, error) {
return v.InvokeWithOptionalArgs(name, DISPATCH_PROPERTYGET, params)
}
// PutProperty attempts to mutate a property in the object.
func (v *IDispatch) PutProperty(name string, params ...interface{}) (*VARIANT, error) {
return v.InvokeWithOptionalArgs(name, DISPATCH_PROPERTYPUT, params)
}

19
vendor/github.com/go-ole/go-ole/idispatch_func.go generated vendored Normal file
View File

@ -0,0 +1,19 @@
// +build !windows
package ole
func getIDsOfName(disp *IDispatch, names []string) ([]int32, error) {
return []int32{}, NewError(E_NOTIMPL)
}
func getTypeInfoCount(disp *IDispatch) (uint32, error) {
return uint32(0), NewError(E_NOTIMPL)
}
func getTypeInfo(disp *IDispatch) (*ITypeInfo, error) {
return nil, NewError(E_NOTIMPL)
}
func invoke(disp *IDispatch, dispid int32, dispatch int16, params ...interface{}) (*VARIANT, error) {
return nil, NewError(E_NOTIMPL)
}

197
vendor/github.com/go-ole/go-ole/idispatch_windows.go generated vendored Normal file
View File

@ -0,0 +1,197 @@
// +build windows
package ole
import (
"syscall"
"time"
"unsafe"
)
func getIDsOfName(disp *IDispatch, names []string) (dispid []int32, err error) {
wnames := make([]*uint16, len(names))
for i := 0; i < len(names); i++ {
wnames[i] = syscall.StringToUTF16Ptr(names[i])
}
dispid = make([]int32, len(names))
namelen := uint32(len(names))
hr, _, _ := syscall.Syscall6(
disp.VTable().GetIDsOfNames,
6,
uintptr(unsafe.Pointer(disp)),
uintptr(unsafe.Pointer(IID_NULL)),
uintptr(unsafe.Pointer(&wnames[0])),
uintptr(namelen),
uintptr(GetUserDefaultLCID()),
uintptr(unsafe.Pointer(&dispid[0])))
if hr != 0 {
err = NewError(hr)
}
return
}
func getTypeInfoCount(disp *IDispatch) (c uint32, err error) {
hr, _, _ := syscall.Syscall(
disp.VTable().GetTypeInfoCount,
2,
uintptr(unsafe.Pointer(disp)),
uintptr(unsafe.Pointer(&c)),
0)
if hr != 0 {
err = NewError(hr)
}
return
}
func getTypeInfo(disp *IDispatch) (tinfo *ITypeInfo, err error) {
hr, _, _ := syscall.Syscall(
disp.VTable().GetTypeInfo,
3,
uintptr(unsafe.Pointer(disp)),
uintptr(GetUserDefaultLCID()),
uintptr(unsafe.Pointer(&tinfo)))
if hr != 0 {
err = NewError(hr)
}
return
}
func invoke(disp *IDispatch, dispid int32, dispatch int16, params ...interface{}) (result *VARIANT, err error) {
var dispparams DISPPARAMS
if dispatch&DISPATCH_PROPERTYPUT != 0 {
dispnames := [1]int32{DISPID_PROPERTYPUT}
dispparams.rgdispidNamedArgs = uintptr(unsafe.Pointer(&dispnames[0]))
dispparams.cNamedArgs = 1
} else if dispatch&DISPATCH_PROPERTYPUTREF != 0 {
dispnames := [1]int32{DISPID_PROPERTYPUT}
dispparams.rgdispidNamedArgs = uintptr(unsafe.Pointer(&dispnames[0]))
dispparams.cNamedArgs = 1
}
var vargs []VARIANT
if len(params) > 0 {
vargs = make([]VARIANT, len(params))
for i, v := range params {
//n := len(params)-i-1
n := len(params) - i - 1
VariantInit(&vargs[n])
switch vv := v.(type) {
case bool:
if vv {
vargs[n] = NewVariant(VT_BOOL, 0xffff)
} else {
vargs[n] = NewVariant(VT_BOOL, 0)
}
case *bool:
vargs[n] = NewVariant(VT_BOOL|VT_BYREF, int64(uintptr(unsafe.Pointer(v.(*bool)))))
case uint8:
vargs[n] = NewVariant(VT_I1, int64(v.(uint8)))
case *uint8:
vargs[n] = NewVariant(VT_I1|VT_BYREF, int64(uintptr(unsafe.Pointer(v.(*uint8)))))
case int8:
vargs[n] = NewVariant(VT_I1, int64(v.(int8)))
case *int8:
vargs[n] = NewVariant(VT_I1|VT_BYREF, int64(uintptr(unsafe.Pointer(v.(*uint8)))))
case int16:
vargs[n] = NewVariant(VT_I2, int64(v.(int16)))
case *int16:
vargs[n] = NewVariant(VT_I2|VT_BYREF, int64(uintptr(unsafe.Pointer(v.(*int16)))))
case uint16:
vargs[n] = NewVariant(VT_UI2, int64(v.(uint16)))
case *uint16:
vargs[n] = NewVariant(VT_UI2|VT_BYREF, int64(uintptr(unsafe.Pointer(v.(*uint16)))))
case int32:
vargs[n] = NewVariant(VT_I4, int64(v.(int32)))
case *int32:
vargs[n] = NewVariant(VT_I4|VT_BYREF, int64(uintptr(unsafe.Pointer(v.(*int32)))))
case uint32:
vargs[n] = NewVariant(VT_UI4, int64(v.(uint32)))
case *uint32:
vargs[n] = NewVariant(VT_UI4|VT_BYREF, int64(uintptr(unsafe.Pointer(v.(*uint32)))))
case int64:
vargs[n] = NewVariant(VT_I8, int64(v.(int64)))
case *int64:
vargs[n] = NewVariant(VT_I8|VT_BYREF, int64(uintptr(unsafe.Pointer(v.(*int64)))))
case uint64:
vargs[n] = NewVariant(VT_UI8, int64(uintptr(v.(uint64))))
case *uint64:
vargs[n] = NewVariant(VT_UI8|VT_BYREF, int64(uintptr(unsafe.Pointer(v.(*uint64)))))
case int:
vargs[n] = NewVariant(VT_I4, int64(v.(int)))
case *int:
vargs[n] = NewVariant(VT_I4|VT_BYREF, int64(uintptr(unsafe.Pointer(v.(*int)))))
case uint:
vargs[n] = NewVariant(VT_UI4, int64(v.(uint)))
case *uint:
vargs[n] = NewVariant(VT_UI4|VT_BYREF, int64(uintptr(unsafe.Pointer(v.(*uint)))))
case float32:
vargs[n] = NewVariant(VT_R4, *(*int64)(unsafe.Pointer(&vv)))
case *float32:
vargs[n] = NewVariant(VT_R4|VT_BYREF, int64(uintptr(unsafe.Pointer(v.(*float32)))))
case float64:
vargs[n] = NewVariant(VT_R8, *(*int64)(unsafe.Pointer(&vv)))
case *float64:
vargs[n] = NewVariant(VT_R8|VT_BYREF, int64(uintptr(unsafe.Pointer(v.(*float64)))))
case string:
vargs[n] = NewVariant(VT_BSTR, int64(uintptr(unsafe.Pointer(SysAllocStringLen(v.(string))))))
case *string:
vargs[n] = NewVariant(VT_BSTR|VT_BYREF, int64(uintptr(unsafe.Pointer(v.(*string)))))
case time.Time:
s := vv.Format("2006-01-02 15:04:05")
vargs[n] = NewVariant(VT_BSTR, int64(uintptr(unsafe.Pointer(SysAllocStringLen(s)))))
case *time.Time:
s := vv.Format("2006-01-02 15:04:05")
vargs[n] = NewVariant(VT_BSTR|VT_BYREF, int64(uintptr(unsafe.Pointer(&s))))
case *IDispatch:
vargs[n] = NewVariant(VT_DISPATCH, int64(uintptr(unsafe.Pointer(v.(*IDispatch)))))
case **IDispatch:
vargs[n] = NewVariant(VT_DISPATCH|VT_BYREF, int64(uintptr(unsafe.Pointer(v.(**IDispatch)))))
case nil:
vargs[n] = NewVariant(VT_NULL, 0)
case *VARIANT:
vargs[n] = NewVariant(VT_VARIANT|VT_BYREF, int64(uintptr(unsafe.Pointer(v.(*VARIANT)))))
case []byte:
safeByteArray := safeArrayFromByteSlice(v.([]byte))
vargs[n] = NewVariant(VT_ARRAY|VT_UI1, int64(uintptr(unsafe.Pointer(safeByteArray))))
defer VariantClear(&vargs[n])
case []string:
safeByteArray := safeArrayFromStringSlice(v.([]string))
vargs[n] = NewVariant(VT_ARRAY|VT_BSTR, int64(uintptr(unsafe.Pointer(safeByteArray))))
defer VariantClear(&vargs[n])
default:
panic("unknown type")
}
}
dispparams.rgvarg = uintptr(unsafe.Pointer(&vargs[0]))
dispparams.cArgs = uint32(len(params))
}
result = new(VARIANT)
var excepInfo EXCEPINFO
VariantInit(result)
hr, _, _ := syscall.Syscall9(
disp.VTable().Invoke,
9,
uintptr(unsafe.Pointer(disp)),
uintptr(dispid),
uintptr(unsafe.Pointer(IID_NULL)),
uintptr(GetUserDefaultLCID()),
uintptr(dispatch),
uintptr(unsafe.Pointer(&dispparams)),
uintptr(unsafe.Pointer(result)),
uintptr(unsafe.Pointer(&excepInfo)),
0)
if hr != 0 {
err = NewErrorWithSubError(hr, BstrToString(excepInfo.bstrDescription), excepInfo)
}
for i, varg := range vargs {
n := len(params) - i - 1
if varg.VT == VT_BSTR && varg.Val != 0 {
SysFreeString(((*int16)(unsafe.Pointer(uintptr(varg.Val)))))
}
if varg.VT == (VT_BSTR|VT_BYREF) && varg.Val != 0 {
*(params[n].(*string)) = LpOleStrToString(*(**uint16)(unsafe.Pointer(uintptr(varg.Val))))
}
}
return
}

19
vendor/github.com/go-ole/go-ole/ienumvariant.go generated vendored Normal file
View File

@ -0,0 +1,19 @@
package ole
import "unsafe"
type IEnumVARIANT struct {
IUnknown
}
type IEnumVARIANTVtbl struct {
IUnknownVtbl
Next uintptr
Skip uintptr
Reset uintptr
Clone uintptr
}
func (v *IEnumVARIANT) VTable() *IEnumVARIANTVtbl {
return (*IEnumVARIANTVtbl)(unsafe.Pointer(v.RawVTable))
}

19
vendor/github.com/go-ole/go-ole/ienumvariant_func.go generated vendored Normal file
View File

@ -0,0 +1,19 @@
// +build !windows
package ole
func (enum *IEnumVARIANT) Clone() (*IEnumVARIANT, error) {
return nil, NewError(E_NOTIMPL)
}
func (enum *IEnumVARIANT) Reset() error {
return NewError(E_NOTIMPL)
}
func (enum *IEnumVARIANT) Skip(celt uint) error {
return NewError(E_NOTIMPL)
}
func (enum *IEnumVARIANT) Next(celt uint) (VARIANT, uint, error) {
return NewVariant(VT_NULL, int64(0)), 0, NewError(E_NOTIMPL)
}

63
vendor/github.com/go-ole/go-ole/ienumvariant_windows.go generated vendored Normal file
View File

@ -0,0 +1,63 @@
// +build windows
package ole
import (
"syscall"
"unsafe"
)
func (enum *IEnumVARIANT) Clone() (cloned *IEnumVARIANT, err error) {
hr, _, _ := syscall.Syscall(
enum.VTable().Clone,
2,
uintptr(unsafe.Pointer(enum)),
uintptr(unsafe.Pointer(&cloned)),
0)
if hr != 0 {
err = NewError(hr)
}
return
}
func (enum *IEnumVARIANT) Reset() (err error) {
hr, _, _ := syscall.Syscall(
enum.VTable().Reset,
1,
uintptr(unsafe.Pointer(enum)),
0,
0)
if hr != 0 {
err = NewError(hr)
}
return
}
func (enum *IEnumVARIANT) Skip(celt uint) (err error) {
hr, _, _ := syscall.Syscall(
enum.VTable().Skip,
2,
uintptr(unsafe.Pointer(enum)),
uintptr(celt),
0)
if hr != 0 {
err = NewError(hr)
}
return
}
func (enum *IEnumVARIANT) Next(celt uint) (array VARIANT, length uint, err error) {
hr, _, _ := syscall.Syscall6(
enum.VTable().Next,
4,
uintptr(unsafe.Pointer(enum)),
uintptr(celt),
uintptr(unsafe.Pointer(&array)),
uintptr(unsafe.Pointer(&length)),
0,
0)
if hr != 0 {
err = NewError(hr)
}
return
}

18
vendor/github.com/go-ole/go-ole/iinspectable.go generated vendored Normal file
View File

@ -0,0 +1,18 @@
package ole
import "unsafe"
type IInspectable struct {
IUnknown
}
type IInspectableVtbl struct {
IUnknownVtbl
GetIIds uintptr
GetRuntimeClassName uintptr
GetTrustLevel uintptr
}
func (v *IInspectable) VTable() *IInspectableVtbl {
return (*IInspectableVtbl)(unsafe.Pointer(v.RawVTable))
}

15
vendor/github.com/go-ole/go-ole/iinspectable_func.go generated vendored Normal file
View File

@ -0,0 +1,15 @@
// +build !windows
package ole
func (v *IInspectable) GetIids() ([]*GUID, error) {
return []*GUID{}, NewError(E_NOTIMPL)
}
func (v *IInspectable) GetRuntimeClassName() (string, error) {
return "", NewError(E_NOTIMPL)
}
func (v *IInspectable) GetTrustLevel() (uint32, error) {
return uint32(0), NewError(E_NOTIMPL)
}

72
vendor/github.com/go-ole/go-ole/iinspectable_windows.go generated vendored Normal file
View File

@ -0,0 +1,72 @@
// +build windows
package ole
import (
"bytes"
"encoding/binary"
"reflect"
"syscall"
"unsafe"
)
func (v *IInspectable) GetIids() (iids []*GUID, err error) {
var count uint32
var array uintptr
hr, _, _ := syscall.Syscall(
v.VTable().GetIIds,
3,
uintptr(unsafe.Pointer(v)),
uintptr(unsafe.Pointer(&count)),
uintptr(unsafe.Pointer(&array)))
if hr != 0 {
err = NewError(hr)
return
}
defer CoTaskMemFree(array)
iids = make([]*GUID, count)
byteCount := count * uint32(unsafe.Sizeof(GUID{}))
slicehdr := reflect.SliceHeader{Data: array, Len: int(byteCount), Cap: int(byteCount)}
byteSlice := *(*[]byte)(unsafe.Pointer(&slicehdr))
reader := bytes.NewReader(byteSlice)
for i := range iids {
guid := GUID{}
err = binary.Read(reader, binary.LittleEndian, &guid)
if err != nil {
return
}
iids[i] = &guid
}
return
}
func (v *IInspectable) GetRuntimeClassName() (s string, err error) {
var hstring HString
hr, _, _ := syscall.Syscall(
v.VTable().GetRuntimeClassName,
2,
uintptr(unsafe.Pointer(v)),
uintptr(unsafe.Pointer(&hstring)),
0)
if hr != 0 {
err = NewError(hr)
return
}
s = hstring.String()
DeleteHString(hstring)
return
}
func (v *IInspectable) GetTrustLevel() (level uint32, err error) {
hr, _, _ := syscall.Syscall(
v.VTable().GetTrustLevel,
2,
uintptr(unsafe.Pointer(v)),
uintptr(unsafe.Pointer(&level)),
0)
if hr != 0 {
err = NewError(hr)
}
return
}

21
vendor/github.com/go-ole/go-ole/iprovideclassinfo.go generated vendored Normal file
View File

@ -0,0 +1,21 @@
package ole
import "unsafe"
type IProvideClassInfo struct {
IUnknown
}
type IProvideClassInfoVtbl struct {
IUnknownVtbl
GetClassInfo uintptr
}
func (v *IProvideClassInfo) VTable() *IProvideClassInfoVtbl {
return (*IProvideClassInfoVtbl)(unsafe.Pointer(v.RawVTable))
}
func (v *IProvideClassInfo) GetClassInfo() (cinfo *ITypeInfo, err error) {
cinfo, err = getClassInfo(v)
return
}

7
vendor/github.com/go-ole/go-ole/iprovideclassinfo_func.go generated vendored Normal file
View File

@ -0,0 +1,7 @@
// +build !windows
package ole
func getClassInfo(disp *IProvideClassInfo) (tinfo *ITypeInfo, err error) {
return nil, NewError(E_NOTIMPL)
}

21
vendor/github.com/go-ole/go-ole/iprovideclassinfo_windows.go generated vendored Normal file
View File

@ -0,0 +1,21 @@
// +build windows
package ole
import (
"syscall"
"unsafe"
)
func getClassInfo(disp *IProvideClassInfo) (tinfo *ITypeInfo, err error) {
hr, _, _ := syscall.Syscall(
disp.VTable().GetClassInfo,
2,
uintptr(unsafe.Pointer(disp)),
uintptr(unsafe.Pointer(&tinfo)),
0)
if hr != 0 {
err = NewError(hr)
}
return
}

34
vendor/github.com/go-ole/go-ole/itypeinfo.go generated vendored Normal file
View File

@ -0,0 +1,34 @@
package ole
import "unsafe"
type ITypeInfo struct {
IUnknown
}
type ITypeInfoVtbl struct {
IUnknownVtbl
GetTypeAttr uintptr
GetTypeComp uintptr
GetFuncDesc uintptr
GetVarDesc uintptr
GetNames uintptr
GetRefTypeOfImplType uintptr
GetImplTypeFlags uintptr
GetIDsOfNames uintptr
Invoke uintptr
GetDocumentation uintptr
GetDllEntry uintptr
GetRefTypeInfo uintptr
AddressOfMember uintptr
CreateInstance uintptr
GetMops uintptr
GetContainingTypeLib uintptr
ReleaseTypeAttr uintptr
ReleaseFuncDesc uintptr
ReleaseVarDesc uintptr
}
func (v *ITypeInfo) VTable() *ITypeInfoVtbl {
return (*ITypeInfoVtbl)(unsafe.Pointer(v.RawVTable))
}

7
vendor/github.com/go-ole/go-ole/itypeinfo_func.go generated vendored Normal file
View File

@ -0,0 +1,7 @@
// +build !windows
package ole
func (v *ITypeInfo) GetTypeAttr() (*TYPEATTR, error) {
return nil, NewError(E_NOTIMPL)
}

21
vendor/github.com/go-ole/go-ole/itypeinfo_windows.go generated vendored Normal file
View File

@ -0,0 +1,21 @@
// +build windows
package ole
import (
"syscall"
"unsafe"
)
func (v *ITypeInfo) GetTypeAttr() (tattr *TYPEATTR, err error) {
hr, _, _ := syscall.Syscall(
uintptr(v.VTable().GetTypeAttr),
2,
uintptr(unsafe.Pointer(v)),
uintptr(unsafe.Pointer(&tattr)),
0)
if hr != 0 {
err = NewError(hr)
}
return
}

57
vendor/github.com/go-ole/go-ole/iunknown.go generated vendored Normal file
View File

@ -0,0 +1,57 @@
package ole
import "unsafe"
type IUnknown struct {
RawVTable *interface{}
}
type IUnknownVtbl struct {
QueryInterface uintptr
AddRef uintptr
Release uintptr
}
type UnknownLike interface {
QueryInterface(iid *GUID) (disp *IDispatch, err error)
AddRef() int32
Release() int32
}
func (v *IUnknown) VTable() *IUnknownVtbl {
return (*IUnknownVtbl)(unsafe.Pointer(v.RawVTable))
}
func (v *IUnknown) PutQueryInterface(interfaceID *GUID, obj interface{}) error {
return reflectQueryInterface(v, v.VTable().QueryInterface, interfaceID, obj)
}
func (v *IUnknown) IDispatch(interfaceID *GUID) (dispatch *IDispatch, err error) {
err = v.PutQueryInterface(interfaceID, &dispatch)
return
}
func (v *IUnknown) IEnumVARIANT(interfaceID *GUID) (enum *IEnumVARIANT, err error) {
err = v.PutQueryInterface(interfaceID, &enum)
return
}
func (v *IUnknown) QueryInterface(iid *GUID) (*IDispatch, error) {
return queryInterface(v, iid)
}
func (v *IUnknown) MustQueryInterface(iid *GUID) (disp *IDispatch) {
unk, err := queryInterface(v, iid)
if err != nil {
panic(err)
}
return unk
}
func (v *IUnknown) AddRef() int32 {
return addRef(v)
}
func (v *IUnknown) Release() int32 {
return release(v)
}

19
vendor/github.com/go-ole/go-ole/iunknown_func.go generated vendored Normal file
View File

@ -0,0 +1,19 @@
// +build !windows
package ole
func reflectQueryInterface(self interface{}, method uintptr, interfaceID *GUID, obj interface{}) (err error) {
return NewError(E_NOTIMPL)
}
func queryInterface(unk *IUnknown, iid *GUID) (disp *IDispatch, err error) {
return nil, NewError(E_NOTIMPL)
}
func addRef(unk *IUnknown) int32 {
return 0
}
func release(unk *IUnknown) int32 {
return 0
}

58
vendor/github.com/go-ole/go-ole/iunknown_windows.go generated vendored Normal file
View File

@ -0,0 +1,58 @@
// +build windows
package ole
import (
"reflect"
"syscall"
"unsafe"
)
func reflectQueryInterface(self interface{}, method uintptr, interfaceID *GUID, obj interface{}) (err error) {
selfValue := reflect.ValueOf(self).Elem()
objValue := reflect.ValueOf(obj).Elem()
hr, _, _ := syscall.Syscall(
method,
3,
selfValue.UnsafeAddr(),
uintptr(unsafe.Pointer(interfaceID)),
objValue.Addr().Pointer())
if hr != 0 {
err = NewError(hr)
}
return
}
func queryInterface(unk *IUnknown, iid *GUID) (disp *IDispatch, err error) {
hr, _, _ := syscall.Syscall(
unk.VTable().QueryInterface,
3,
uintptr(unsafe.Pointer(unk)),
uintptr(unsafe.Pointer(iid)),
uintptr(unsafe.Pointer(&disp)))
if hr != 0 {
err = NewError(hr)
}
return
}
func addRef(unk *IUnknown) int32 {
ret, _, _ := syscall.Syscall(
unk.VTable().AddRef,
1,
uintptr(unsafe.Pointer(unk)),
0,
0)
return int32(ret)
}
func release(unk *IUnknown) int32 {
ret, _, _ := syscall.Syscall(
unk.VTable().Release,
1,
uintptr(unsafe.Pointer(unk)),
0,
0)
return int32(ret)
}

157
vendor/github.com/go-ole/go-ole/ole.go generated vendored Normal file
View File

@ -0,0 +1,157 @@
package ole
import (
"fmt"
"strings"
)
// DISPPARAMS are the arguments that passed to methods or property.
type DISPPARAMS struct {
rgvarg uintptr
rgdispidNamedArgs uintptr
cArgs uint32
cNamedArgs uint32
}
// EXCEPINFO defines exception info.
type EXCEPINFO struct {
wCode uint16
wReserved uint16
bstrSource *uint16
bstrDescription *uint16
bstrHelpFile *uint16
dwHelpContext uint32
pvReserved uintptr
pfnDeferredFillIn uintptr
scode uint32
}
// WCode return wCode in EXCEPINFO.
func (e EXCEPINFO) WCode() uint16 {
return e.wCode
}
// SCODE return scode in EXCEPINFO.
func (e EXCEPINFO) SCODE() uint32 {
return e.scode
}
// String convert EXCEPINFO to string.
func (e EXCEPINFO) String() string {
var src, desc, hlp string
if e.bstrSource == nil {
src = "<nil>"
} else {
src = BstrToString(e.bstrSource)
}
if e.bstrDescription == nil {
desc = "<nil>"
} else {
desc = BstrToString(e.bstrDescription)
}
if e.bstrHelpFile == nil {
hlp = "<nil>"
} else {
hlp = BstrToString(e.bstrHelpFile)
}
return fmt.Sprintf(
"wCode: %#x, bstrSource: %v, bstrDescription: %v, bstrHelpFile: %v, dwHelpContext: %#x, scode: %#x",
e.wCode, src, desc, hlp, e.dwHelpContext, e.scode,
)
}
// Error implements error interface and returns error string.
func (e EXCEPINFO) Error() string {
if e.bstrDescription != nil {
return strings.TrimSpace(BstrToString(e.bstrDescription))
}
src := "Unknown"
if e.bstrSource != nil {
src = BstrToString(e.bstrSource)
}
code := e.scode
if e.wCode != 0 {
code = uint32(e.wCode)
}
return fmt.Sprintf("%v: %#x", src, code)
}
// PARAMDATA defines parameter data type.
type PARAMDATA struct {
Name *int16
Vt uint16
}
// METHODDATA defines method info.
type METHODDATA struct {
Name *uint16
Data *PARAMDATA
Dispid int32
Meth uint32
CC int32
CArgs uint32
Flags uint16
VtReturn uint32
}
// INTERFACEDATA defines interface info.
type INTERFACEDATA struct {
MethodData *METHODDATA
CMembers uint32
}
// Point is 2D vector type.
type Point struct {
X int32
Y int32
}
// Msg is message between processes.
type Msg struct {
Hwnd uint32
Message uint32
Wparam int32
Lparam int32
Time uint32
Pt Point
}
// TYPEDESC defines data type.
type TYPEDESC struct {
Hreftype uint32
VT uint16
}
// IDLDESC defines IDL info.
type IDLDESC struct {
DwReserved uint32
WIDLFlags uint16
}
// TYPEATTR defines type info.
type TYPEATTR struct {
Guid GUID
Lcid uint32
dwReserved uint32
MemidConstructor int32
MemidDestructor int32
LpstrSchema *uint16
CbSizeInstance uint32
Typekind int32
CFuncs uint16
CVars uint16
CImplTypes uint16
CbSizeVft uint16
CbAlignment uint16
WTypeFlags uint16
WMajorVerNum uint16
WMinorVerNum uint16
TdescAlias TYPEDESC
IdldescType IDLDESC
}

100
vendor/github.com/go-ole/go-ole/oleutil/connection.go generated vendored Normal file
View File

@ -0,0 +1,100 @@
// +build windows
package oleutil
import (
"reflect"
"unsafe"
ole "github.com/go-ole/go-ole"
)
type stdDispatch struct {
lpVtbl *stdDispatchVtbl
ref int32
iid *ole.GUID
iface interface{}
funcMap map[string]int32
}
type stdDispatchVtbl struct {
pQueryInterface uintptr
pAddRef uintptr
pRelease uintptr
pGetTypeInfoCount uintptr
pGetTypeInfo uintptr
pGetIDsOfNames uintptr
pInvoke uintptr
}
func dispQueryInterface(this *ole.IUnknown, iid *ole.GUID, punk **ole.IUnknown) uint32 {
pthis := (*stdDispatch)(unsafe.Pointer(this))
*punk = nil
if ole.IsEqualGUID(iid, ole.IID_IUnknown) ||
ole.IsEqualGUID(iid, ole.IID_IDispatch) {
dispAddRef(this)
*punk = this
return ole.S_OK
}
if ole.IsEqualGUID(iid, pthis.iid) {
dispAddRef(this)
*punk = this
return ole.S_OK
}
return ole.E_NOINTERFACE
}
func dispAddRef(this *ole.IUnknown) int32 {
pthis := (*stdDispatch)(unsafe.Pointer(this))
pthis.ref++
return pthis.ref
}
func dispRelease(this *ole.IUnknown) int32 {
pthis := (*stdDispatch)(unsafe.Pointer(this))
pthis.ref--
return pthis.ref
}
func dispGetIDsOfNames(this *ole.IUnknown, iid *ole.GUID, wnames []*uint16, namelen int, lcid int, pdisp []int32) uintptr {
pthis := (*stdDispatch)(unsafe.Pointer(this))
names := make([]string, len(wnames))
for i := 0; i < len(names); i++ {
names[i] = ole.LpOleStrToString(wnames[i])
}
for n := 0; n < namelen; n++ {
if id, ok := pthis.funcMap[names[n]]; ok {
pdisp[n] = id
}
}
return ole.S_OK
}
func dispGetTypeInfoCount(pcount *int) uintptr {
if pcount != nil {
*pcount = 0
}
return ole.S_OK
}
func dispGetTypeInfo(ptypeif *uintptr) uintptr {
return ole.E_NOTIMPL
}
func dispInvoke(this *ole.IDispatch, dispid int32, riid *ole.GUID, lcid int, flags int16, dispparams *ole.DISPPARAMS, result *ole.VARIANT, pexcepinfo *ole.EXCEPINFO, nerr *uint) uintptr {
pthis := (*stdDispatch)(unsafe.Pointer(this))
found := ""
for name, id := range pthis.funcMap {
if id == dispid {
found = name
}
}
if found != "" {
rv := reflect.ValueOf(pthis.iface).Elem()
rm := rv.MethodByName(found)
rr := rm.Call([]reflect.Value{})
println(len(rr))
return ole.S_OK
}
return ole.E_NOTIMPL
}

10
vendor/github.com/go-ole/go-ole/oleutil/connection_func.go generated vendored Normal file
View File

@ -0,0 +1,10 @@
// +build !windows
package oleutil
import ole "github.com/go-ole/go-ole"
// ConnectObject creates a connection point between two services for communication.
func ConnectObject(disp *ole.IDispatch, iid *ole.GUID, idisp interface{}) (uint32, error) {
return 0, ole.NewError(ole.E_NOTIMPL)
}

58
vendor/github.com/go-ole/go-ole/oleutil/connection_windows.go generated vendored Normal file
View File

@ -0,0 +1,58 @@
// +build windows
package oleutil
import (
"reflect"
"syscall"
"unsafe"
ole "github.com/go-ole/go-ole"
)
// ConnectObject creates a connection point between two services for communication.
func ConnectObject(disp *ole.IDispatch, iid *ole.GUID, idisp interface{}) (cookie uint32, err error) {
unknown, err := disp.QueryInterface(ole.IID_IConnectionPointContainer)
if err != nil {
return
}
container := (*ole.IConnectionPointContainer)(unsafe.Pointer(unknown))
var point *ole.IConnectionPoint
err = container.FindConnectionPoint(iid, &point)
if err != nil {
return
}
if edisp, ok := idisp.(*ole.IUnknown); ok {
cookie, err = point.Advise(edisp)
container.Release()
if err != nil {
return
}
}
rv := reflect.ValueOf(disp).Elem()
if rv.Type().Kind() == reflect.Struct {
dest := &stdDispatch{}
dest.lpVtbl = &stdDispatchVtbl{}
dest.lpVtbl.pQueryInterface = syscall.NewCallback(dispQueryInterface)
dest.lpVtbl.pAddRef = syscall.NewCallback(dispAddRef)
dest.lpVtbl.pRelease = syscall.NewCallback(dispRelease)
dest.lpVtbl.pGetTypeInfoCount = syscall.NewCallback(dispGetTypeInfoCount)
dest.lpVtbl.pGetTypeInfo = syscall.NewCallback(dispGetTypeInfo)
dest.lpVtbl.pGetIDsOfNames = syscall.NewCallback(dispGetIDsOfNames)
dest.lpVtbl.pInvoke = syscall.NewCallback(dispInvoke)
dest.iface = disp
dest.iid = iid
cookie, err = point.Advise((*ole.IUnknown)(unsafe.Pointer(dest)))
container.Release()
if err != nil {
point.Release()
return
}
return
}
container.Release()
return 0, ole.NewError(ole.E_INVALIDARG)
}

6
vendor/github.com/go-ole/go-ole/oleutil/go-get.go generated vendored Normal file
View File

@ -0,0 +1,6 @@
// This file is here so go get succeeds as without it errors with:
// no buildable Go source files in ...
//
// +build !windows
package oleutil

127
vendor/github.com/go-ole/go-ole/oleutil/oleutil.go generated vendored Normal file
View File

@ -0,0 +1,127 @@
package oleutil
import ole "github.com/go-ole/go-ole"
// ClassIDFrom retrieves class ID whether given is program ID or application string.
func ClassIDFrom(programID string) (classID *ole.GUID, err error) {
return ole.ClassIDFrom(programID)
}
// CreateObject creates object from programID based on interface type.
//
// Only supports IUnknown.
//
// Program ID can be either program ID or application string.
func CreateObject(programID string) (unknown *ole.IUnknown, err error) {
classID, err := ole.ClassIDFrom(programID)
if err != nil {
return
}
unknown, err = ole.CreateInstance(classID, ole.IID_IUnknown)
if err != nil {
return
}
return
}
// GetActiveObject retrieves active object for program ID and interface ID based
// on interface type.
//
// Only supports IUnknown.
//
// Program ID can be either program ID or application string.
func GetActiveObject(programID string) (unknown *ole.IUnknown, err error) {
classID, err := ole.ClassIDFrom(programID)
if err != nil {
return
}
unknown, err = ole.GetActiveObject(classID, ole.IID_IUnknown)
if err != nil {
return
}
return
}
// CallMethod calls method on IDispatch with parameters.
func CallMethod(disp *ole.IDispatch, name string, params ...interface{}) (result *ole.VARIANT, err error) {
return disp.InvokeWithOptionalArgs(name, ole.DISPATCH_METHOD, params)
}
// MustCallMethod calls method on IDispatch with parameters or panics.
func MustCallMethod(disp *ole.IDispatch, name string, params ...interface{}) (result *ole.VARIANT) {
r, err := CallMethod(disp, name, params...)
if err != nil {
panic(err.Error())
}
return r
}
// GetProperty retrieves property from IDispatch.
func GetProperty(disp *ole.IDispatch, name string, params ...interface{}) (result *ole.VARIANT, err error) {
return disp.InvokeWithOptionalArgs(name, ole.DISPATCH_PROPERTYGET, params)
}
// MustGetProperty retrieves property from IDispatch or panics.
func MustGetProperty(disp *ole.IDispatch, name string, params ...interface{}) (result *ole.VARIANT) {
r, err := GetProperty(disp, name, params...)
if err != nil {
panic(err.Error())
}
return r
}
// PutProperty mutates property.
func PutProperty(disp *ole.IDispatch, name string, params ...interface{}) (result *ole.VARIANT, err error) {
return disp.InvokeWithOptionalArgs(name, ole.DISPATCH_PROPERTYPUT, params)
}
// MustPutProperty mutates property or panics.
func MustPutProperty(disp *ole.IDispatch, name string, params ...interface{}) (result *ole.VARIANT) {
r, err := PutProperty(disp, name, params...)
if err != nil {
panic(err.Error())
}
return r
}
// PutPropertyRef mutates property reference.
func PutPropertyRef(disp *ole.IDispatch, name string, params ...interface{}) (result *ole.VARIANT, err error) {
return disp.InvokeWithOptionalArgs(name, ole.DISPATCH_PROPERTYPUTREF, params)
}
// MustPutPropertyRef mutates property reference or panics.
func MustPutPropertyRef(disp *ole.IDispatch, name string, params ...interface{}) (result *ole.VARIANT) {
r, err := PutPropertyRef(disp, name, params...)
if err != nil {
panic(err.Error())
}
return r
}
func ForEach(disp *ole.IDispatch, f func(v *ole.VARIANT) error) error {
newEnum, err := disp.GetProperty("_NewEnum")
if err != nil {
return err
}
defer newEnum.Clear()
enum, err := newEnum.ToIUnknown().IEnumVARIANT(ole.IID_IEnumVariant)
if err != nil {
return err
}
defer enum.Release()
for item, length, err := enum.Next(1); length > 0; item, length, err = enum.Next(1) {
if err != nil {
return err
}
if ferr := f(&item); ferr != nil {
return ferr
}
}
return nil
}

27
vendor/github.com/go-ole/go-ole/safearray.go generated vendored Normal file
View File

@ -0,0 +1,27 @@
// Package is meant to retrieve and process safe array data returned from COM.
package ole
// SafeArrayBound defines the SafeArray boundaries.
type SafeArrayBound struct {
Elements uint32
LowerBound int32
}
// SafeArray is how COM handles arrays.
type SafeArray struct {
Dimensions uint16
FeaturesFlag uint16
ElementsSize uint32
LocksAmount uint32
Data uint32
Bounds [16]byte
}
// SAFEARRAY is obsolete, exists for backwards compatibility.
// Use SafeArray
type SAFEARRAY SafeArray
// SAFEARRAYBOUND is obsolete, exists for backwards compatibility.
// Use SafeArrayBound
type SAFEARRAYBOUND SafeArrayBound

211
vendor/github.com/go-ole/go-ole/safearray_func.go generated vendored Normal file
View File

@ -0,0 +1,211 @@
// +build !windows
package ole
import (
"unsafe"
)
// safeArrayAccessData returns raw array pointer.
//
// AKA: SafeArrayAccessData in Windows API.
func safeArrayAccessData(safearray *SafeArray) (uintptr, error) {
return uintptr(0), NewError(E_NOTIMPL)
}
// safeArrayUnaccessData releases raw array.
//
// AKA: SafeArrayUnaccessData in Windows API.
func safeArrayUnaccessData(safearray *SafeArray) error {
return NewError(E_NOTIMPL)
}
// safeArrayAllocData allocates SafeArray.
//
// AKA: SafeArrayAllocData in Windows API.
func safeArrayAllocData(safearray *SafeArray) error {
return NewError(E_NOTIMPL)
}
// safeArrayAllocDescriptor allocates SafeArray.
//
// AKA: SafeArrayAllocDescriptor in Windows API.
func safeArrayAllocDescriptor(dimensions uint32) (*SafeArray, error) {
return nil, NewError(E_NOTIMPL)
}
// safeArrayAllocDescriptorEx allocates SafeArray.
//
// AKA: SafeArrayAllocDescriptorEx in Windows API.
func safeArrayAllocDescriptorEx(variantType VT, dimensions uint32) (*SafeArray, error) {
return nil, NewError(E_NOTIMPL)
}
// safeArrayCopy returns copy of SafeArray.
//
// AKA: SafeArrayCopy in Windows API.
func safeArrayCopy(original *SafeArray) (*SafeArray, error) {
return nil, NewError(E_NOTIMPL)
}
// safeArrayCopyData duplicates SafeArray into another SafeArray object.
//
// AKA: SafeArrayCopyData in Windows API.
func safeArrayCopyData(original *SafeArray, duplicate *SafeArray) error {
return NewError(E_NOTIMPL)
}
// safeArrayCreate creates SafeArray.
//
// AKA: SafeArrayCreate in Windows API.
func safeArrayCreate(variantType VT, dimensions uint32, bounds *SafeArrayBound) (*SafeArray, error) {
return nil, NewError(E_NOTIMPL)
}
// safeArrayCreateEx creates SafeArray.
//
// AKA: SafeArrayCreateEx in Windows API.
func safeArrayCreateEx(variantType VT, dimensions uint32, bounds *SafeArrayBound, extra uintptr) (*SafeArray, error) {
return nil, NewError(E_NOTIMPL)
}
// safeArrayCreateVector creates SafeArray.
//
// AKA: SafeArrayCreateVector in Windows API.
func safeArrayCreateVector(variantType VT, lowerBound int32, length uint32) (*SafeArray, error) {
return nil, NewError(E_NOTIMPL)
}
// safeArrayCreateVectorEx creates SafeArray.
//
// AKA: SafeArrayCreateVectorEx in Windows API.
func safeArrayCreateVectorEx(variantType VT, lowerBound int32, length uint32, extra uintptr) (*SafeArray, error) {
return nil, NewError(E_NOTIMPL)
}
// safeArrayDestroy destroys SafeArray object.
//
// AKA: SafeArrayDestroy in Windows API.
func safeArrayDestroy(safearray *SafeArray) error {
return NewError(E_NOTIMPL)
}
// safeArrayDestroyData destroys SafeArray object.
//
// AKA: SafeArrayDestroyData in Windows API.
func safeArrayDestroyData(safearray *SafeArray) error {
return NewError(E_NOTIMPL)
}
// safeArrayDestroyDescriptor destroys SafeArray object.
//
// AKA: SafeArrayDestroyDescriptor in Windows API.
func safeArrayDestroyDescriptor(safearray *SafeArray) error {
return NewError(E_NOTIMPL)
}
// safeArrayGetDim is the amount of dimensions in the SafeArray.
//
// SafeArrays may have multiple dimensions. Meaning, it could be
// multidimensional array.
//
// AKA: SafeArrayGetDim in Windows API.
func safeArrayGetDim(safearray *SafeArray) (*uint32, error) {
u := uint32(0)
return &u, NewError(E_NOTIMPL)
}
// safeArrayGetElementSize is the element size in bytes.
//
// AKA: SafeArrayGetElemsize in Windows API.
func safeArrayGetElementSize(safearray *SafeArray) (*uint32, error) {
u := uint32(0)
return &u, NewError(E_NOTIMPL)
}
// safeArrayGetElement retrieves element at given index.
func safeArrayGetElement(safearray *SafeArray, index int64, pv unsafe.Pointer) error {
return NewError(E_NOTIMPL)
}
// safeArrayGetElement retrieves element at given index and converts to string.
func safeArrayGetElementString(safearray *SafeArray, index int64) (string, error) {
return "", NewError(E_NOTIMPL)
}
// safeArrayGetIID is the InterfaceID of the elements in the SafeArray.
//
// AKA: SafeArrayGetIID in Windows API.
func safeArrayGetIID(safearray *SafeArray) (*GUID, error) {
return nil, NewError(E_NOTIMPL)
}
// safeArrayGetLBound returns lower bounds of SafeArray.
//
// SafeArrays may have multiple dimensions. Meaning, it could be
// multidimensional array.
//
// AKA: SafeArrayGetLBound in Windows API.
func safeArrayGetLBound(safearray *SafeArray, dimension uint32) (int64, error) {
return int64(0), NewError(E_NOTIMPL)
}
// safeArrayGetUBound returns upper bounds of SafeArray.
//
// SafeArrays may have multiple dimensions. Meaning, it could be
// multidimensional array.
//
// AKA: SafeArrayGetUBound in Windows API.
func safeArrayGetUBound(safearray *SafeArray, dimension uint32) (int64, error) {
return int64(0), NewError(E_NOTIMPL)
}
// safeArrayGetVartype returns data type of SafeArray.
//
// AKA: SafeArrayGetVartype in Windows API.
func safeArrayGetVartype(safearray *SafeArray) (uint16, error) {
return uint16(0), NewError(E_NOTIMPL)
}
// safeArrayLock locks SafeArray for reading to modify SafeArray.
//
// This must be called during some calls to ensure that another process does not
// read or write to the SafeArray during editing.
//
// AKA: SafeArrayLock in Windows API.
func safeArrayLock(safearray *SafeArray) error {
return NewError(E_NOTIMPL)
}
// safeArrayUnlock unlocks SafeArray for reading.
//
// AKA: SafeArrayUnlock in Windows API.
func safeArrayUnlock(safearray *SafeArray) error {
return NewError(E_NOTIMPL)
}
// safeArrayPutElement stores the data element at the specified location in the
// array.
//
// AKA: SafeArrayPutElement in Windows API.
func safeArrayPutElement(safearray *SafeArray, index int64, element uintptr) error {
return NewError(E_NOTIMPL)
}
// safeArrayGetRecordInfo accesses IRecordInfo info for custom types.
//
// AKA: SafeArrayGetRecordInfo in Windows API.
//
// XXX: Must implement IRecordInfo interface for this to return.
func safeArrayGetRecordInfo(safearray *SafeArray) (interface{}, error) {
return nil, NewError(E_NOTIMPL)
}
// safeArraySetRecordInfo mutates IRecordInfo info for custom types.
//
// AKA: SafeArraySetRecordInfo in Windows API.
//
// XXX: Must implement IRecordInfo interface for this to return.
func safeArraySetRecordInfo(safearray *SafeArray, recordInfo interface{}) error {
return NewError(E_NOTIMPL)
}

337
vendor/github.com/go-ole/go-ole/safearray_windows.go generated vendored Normal file
View File

@ -0,0 +1,337 @@
// +build windows
package ole
import (
"unsafe"
)
var (
procSafeArrayAccessData, _ = modoleaut32.FindProc("SafeArrayAccessData")
procSafeArrayAllocData, _ = modoleaut32.FindProc("SafeArrayAllocData")
procSafeArrayAllocDescriptor, _ = modoleaut32.FindProc("SafeArrayAllocDescriptor")
procSafeArrayAllocDescriptorEx, _ = modoleaut32.FindProc("SafeArrayAllocDescriptorEx")
procSafeArrayCopy, _ = modoleaut32.FindProc("SafeArrayCopy")
procSafeArrayCopyData, _ = modoleaut32.FindProc("SafeArrayCopyData")
procSafeArrayCreate, _ = modoleaut32.FindProc("SafeArrayCreate")
procSafeArrayCreateEx, _ = modoleaut32.FindProc("SafeArrayCreateEx")
procSafeArrayCreateVector, _ = modoleaut32.FindProc("SafeArrayCreateVector")
procSafeArrayCreateVectorEx, _ = modoleaut32.FindProc("SafeArrayCreateVectorEx")
procSafeArrayDestroy, _ = modoleaut32.FindProc("SafeArrayDestroy")
procSafeArrayDestroyData, _ = modoleaut32.FindProc("SafeArrayDestroyData")
procSafeArrayDestroyDescriptor, _ = modoleaut32.FindProc("SafeArrayDestroyDescriptor")
procSafeArrayGetDim, _ = modoleaut32.FindProc("SafeArrayGetDim")
procSafeArrayGetElement, _ = modoleaut32.FindProc("SafeArrayGetElement")
procSafeArrayGetElemsize, _ = modoleaut32.FindProc("SafeArrayGetElemsize")
procSafeArrayGetIID, _ = modoleaut32.FindProc("SafeArrayGetIID")
procSafeArrayGetLBound, _ = modoleaut32.FindProc("SafeArrayGetLBound")
procSafeArrayGetUBound, _ = modoleaut32.FindProc("SafeArrayGetUBound")
procSafeArrayGetVartype, _ = modoleaut32.FindProc("SafeArrayGetVartype")
procSafeArrayLock, _ = modoleaut32.FindProc("SafeArrayLock")
procSafeArrayPtrOfIndex, _ = modoleaut32.FindProc("SafeArrayPtrOfIndex")
procSafeArrayUnaccessData, _ = modoleaut32.FindProc("SafeArrayUnaccessData")
procSafeArrayUnlock, _ = modoleaut32.FindProc("SafeArrayUnlock")
procSafeArrayPutElement, _ = modoleaut32.FindProc("SafeArrayPutElement")
//procSafeArrayRedim, _ = modoleaut32.FindProc("SafeArrayRedim") // TODO
//procSafeArraySetIID, _ = modoleaut32.FindProc("SafeArraySetIID") // TODO
procSafeArrayGetRecordInfo, _ = modoleaut32.FindProc("SafeArrayGetRecordInfo")
procSafeArraySetRecordInfo, _ = modoleaut32.FindProc("SafeArraySetRecordInfo")
)
// safeArrayAccessData returns raw array pointer.
//
// AKA: SafeArrayAccessData in Windows API.
// Todo: Test
func safeArrayAccessData(safearray *SafeArray) (element uintptr, err error) {
err = convertHresultToError(
procSafeArrayAccessData.Call(
uintptr(unsafe.Pointer(safearray)),
uintptr(unsafe.Pointer(&element))))
return
}
// safeArrayUnaccessData releases raw array.
//
// AKA: SafeArrayUnaccessData in Windows API.
func safeArrayUnaccessData(safearray *SafeArray) (err error) {
err = convertHresultToError(procSafeArrayUnaccessData.Call(uintptr(unsafe.Pointer(safearray))))
return
}
// safeArrayAllocData allocates SafeArray.
//
// AKA: SafeArrayAllocData in Windows API.
func safeArrayAllocData(safearray *SafeArray) (err error) {
err = convertHresultToError(procSafeArrayAllocData.Call(uintptr(unsafe.Pointer(safearray))))
return
}
// safeArrayAllocDescriptor allocates SafeArray.
//
// AKA: SafeArrayAllocDescriptor in Windows API.
func safeArrayAllocDescriptor(dimensions uint32) (safearray *SafeArray, err error) {
err = convertHresultToError(
procSafeArrayAllocDescriptor.Call(uintptr(dimensions), uintptr(unsafe.Pointer(&safearray))))
return
}
// safeArrayAllocDescriptorEx allocates SafeArray.
//
// AKA: SafeArrayAllocDescriptorEx in Windows API.
func safeArrayAllocDescriptorEx(variantType VT, dimensions uint32) (safearray *SafeArray, err error) {
err = convertHresultToError(
procSafeArrayAllocDescriptorEx.Call(
uintptr(variantType),
uintptr(dimensions),
uintptr(unsafe.Pointer(&safearray))))
return
}
// safeArrayCopy returns copy of SafeArray.
//
// AKA: SafeArrayCopy in Windows API.
func safeArrayCopy(original *SafeArray) (safearray *SafeArray, err error) {
err = convertHresultToError(
procSafeArrayCopy.Call(
uintptr(unsafe.Pointer(original)),
uintptr(unsafe.Pointer(&safearray))))
return
}
// safeArrayCopyData duplicates SafeArray into another SafeArray object.
//
// AKA: SafeArrayCopyData in Windows API.
func safeArrayCopyData(original *SafeArray, duplicate *SafeArray) (err error) {
err = convertHresultToError(
procSafeArrayCopyData.Call(
uintptr(unsafe.Pointer(original)),
uintptr(unsafe.Pointer(duplicate))))
return
}
// safeArrayCreate creates SafeArray.
//
// AKA: SafeArrayCreate in Windows API.
func safeArrayCreate(variantType VT, dimensions uint32, bounds *SafeArrayBound) (safearray *SafeArray, err error) {
sa, _, err := procSafeArrayCreate.Call(
uintptr(variantType),
uintptr(dimensions),
uintptr(unsafe.Pointer(bounds)))
safearray = (*SafeArray)(unsafe.Pointer(&sa))
return
}
// safeArrayCreateEx creates SafeArray.
//
// AKA: SafeArrayCreateEx in Windows API.
func safeArrayCreateEx(variantType VT, dimensions uint32, bounds *SafeArrayBound, extra uintptr) (safearray *SafeArray, err error) {
sa, _, err := procSafeArrayCreateEx.Call(
uintptr(variantType),
uintptr(dimensions),
uintptr(unsafe.Pointer(bounds)),
extra)
safearray = (*SafeArray)(unsafe.Pointer(sa))
return
}
// safeArrayCreateVector creates SafeArray.
//
// AKA: SafeArrayCreateVector in Windows API.
func safeArrayCreateVector(variantType VT, lowerBound int32, length uint32) (safearray *SafeArray, err error) {
sa, _, err := procSafeArrayCreateVector.Call(
uintptr(variantType),
uintptr(lowerBound),
uintptr(length))
safearray = (*SafeArray)(unsafe.Pointer(sa))
return
}
// safeArrayCreateVectorEx creates SafeArray.
//
// AKA: SafeArrayCreateVectorEx in Windows API.
func safeArrayCreateVectorEx(variantType VT, lowerBound int32, length uint32, extra uintptr) (safearray *SafeArray, err error) {
sa, _, err := procSafeArrayCreateVectorEx.Call(
uintptr(variantType),
uintptr(lowerBound),
uintptr(length),
extra)
safearray = (*SafeArray)(unsafe.Pointer(sa))
return
}
// safeArrayDestroy destroys SafeArray object.
//
// AKA: SafeArrayDestroy in Windows API.
func safeArrayDestroy(safearray *SafeArray) (err error) {
err = convertHresultToError(procSafeArrayDestroy.Call(uintptr(unsafe.Pointer(safearray))))
return
}
// safeArrayDestroyData destroys SafeArray object.
//
// AKA: SafeArrayDestroyData in Windows API.
func safeArrayDestroyData(safearray *SafeArray) (err error) {
err = convertHresultToError(procSafeArrayDestroyData.Call(uintptr(unsafe.Pointer(safearray))))
return
}
// safeArrayDestroyDescriptor destroys SafeArray object.
//
// AKA: SafeArrayDestroyDescriptor in Windows API.
func safeArrayDestroyDescriptor(safearray *SafeArray) (err error) {
err = convertHresultToError(procSafeArrayDestroyDescriptor.Call(uintptr(unsafe.Pointer(safearray))))
return
}
// safeArrayGetDim is the amount of dimensions in the SafeArray.
//
// SafeArrays may have multiple dimensions. Meaning, it could be
// multidimensional array.
//
// AKA: SafeArrayGetDim in Windows API.
func safeArrayGetDim(safearray *SafeArray) (dimensions *uint32, err error) {
l, _, err := procSafeArrayGetDim.Call(uintptr(unsafe.Pointer(safearray)))
dimensions = (*uint32)(unsafe.Pointer(l))
return
}
// safeArrayGetElementSize is the element size in bytes.
//
// AKA: SafeArrayGetElemsize in Windows API.
func safeArrayGetElementSize(safearray *SafeArray) (length *uint32, err error) {
l, _, err := procSafeArrayGetElemsize.Call(uintptr(unsafe.Pointer(safearray)))
length = (*uint32)(unsafe.Pointer(l))
return
}
// safeArrayGetElement retrieves element at given index.
func safeArrayGetElement(safearray *SafeArray, index int64, pv unsafe.Pointer) error {
return convertHresultToError(
procSafeArrayGetElement.Call(
uintptr(unsafe.Pointer(safearray)),
uintptr(unsafe.Pointer(&index)),
uintptr(pv)))
}
// safeArrayGetElementString retrieves element at given index and converts to string.
func safeArrayGetElementString(safearray *SafeArray, index int64) (str string, err error) {
var element *int16
err = convertHresultToError(
procSafeArrayGetElement.Call(
uintptr(unsafe.Pointer(safearray)),
uintptr(unsafe.Pointer(&index)),
uintptr(unsafe.Pointer(&element))))
str = BstrToString(*(**uint16)(unsafe.Pointer(&element)))
SysFreeString(element)
return
}
// safeArrayGetIID is the InterfaceID of the elements in the SafeArray.
//
// AKA: SafeArrayGetIID in Windows API.
func safeArrayGetIID(safearray *SafeArray) (guid *GUID, err error) {
err = convertHresultToError(
procSafeArrayGetIID.Call(
uintptr(unsafe.Pointer(safearray)),
uintptr(unsafe.Pointer(&guid))))
return
}
// safeArrayGetLBound returns lower bounds of SafeArray.
//
// SafeArrays may have multiple dimensions. Meaning, it could be
// multidimensional array.
//
// AKA: SafeArrayGetLBound in Windows API.
func safeArrayGetLBound(safearray *SafeArray, dimension uint32) (lowerBound int64, err error) {
err = convertHresultToError(
procSafeArrayGetLBound.Call(
uintptr(unsafe.Pointer(safearray)),
uintptr(dimension),
uintptr(unsafe.Pointer(&lowerBound))))
return
}
// safeArrayGetUBound returns upper bounds of SafeArray.
//
// SafeArrays may have multiple dimensions. Meaning, it could be
// multidimensional array.
//
// AKA: SafeArrayGetUBound in Windows API.
func safeArrayGetUBound(safearray *SafeArray, dimension uint32) (upperBound int64, err error) {
err = convertHresultToError(
procSafeArrayGetUBound.Call(
uintptr(unsafe.Pointer(safearray)),
uintptr(dimension),
uintptr(unsafe.Pointer(&upperBound))))
return
}
// safeArrayGetVartype returns data type of SafeArray.
//
// AKA: SafeArrayGetVartype in Windows API.
func safeArrayGetVartype(safearray *SafeArray) (varType uint16, err error) {
err = convertHresultToError(
procSafeArrayGetVartype.Call(
uintptr(unsafe.Pointer(safearray)),
uintptr(unsafe.Pointer(&varType))))
return
}
// safeArrayLock locks SafeArray for reading to modify SafeArray.
//
// This must be called during some calls to ensure that another process does not
// read or write to the SafeArray during editing.
//
// AKA: SafeArrayLock in Windows API.
func safeArrayLock(safearray *SafeArray) (err error) {
err = convertHresultToError(procSafeArrayLock.Call(uintptr(unsafe.Pointer(safearray))))
return
}
// safeArrayUnlock unlocks SafeArray for reading.
//
// AKA: SafeArrayUnlock in Windows API.
func safeArrayUnlock(safearray *SafeArray) (err error) {
err = convertHresultToError(procSafeArrayUnlock.Call(uintptr(unsafe.Pointer(safearray))))
return
}
// safeArrayPutElement stores the data element at the specified location in the
// array.
//
// AKA: SafeArrayPutElement in Windows API.
func safeArrayPutElement(safearray *SafeArray, index int64, element uintptr) (err error) {
err = convertHresultToError(
procSafeArrayPutElement.Call(
uintptr(unsafe.Pointer(safearray)),
uintptr(unsafe.Pointer(&index)),
uintptr(unsafe.Pointer(element))))
return
}
// safeArrayGetRecordInfo accesses IRecordInfo info for custom types.
//
// AKA: SafeArrayGetRecordInfo in Windows API.
//
// XXX: Must implement IRecordInfo interface for this to return.
func safeArrayGetRecordInfo(safearray *SafeArray) (recordInfo interface{}, err error) {
err = convertHresultToError(
procSafeArrayGetRecordInfo.Call(
uintptr(unsafe.Pointer(safearray)),
uintptr(unsafe.Pointer(&recordInfo))))
return
}
// safeArraySetRecordInfo mutates IRecordInfo info for custom types.
//
// AKA: SafeArraySetRecordInfo in Windows API.
//
// XXX: Must implement IRecordInfo interface for this to return.
func safeArraySetRecordInfo(safearray *SafeArray, recordInfo interface{}) (err error) {
err = convertHresultToError(
procSafeArraySetRecordInfo.Call(
uintptr(unsafe.Pointer(safearray)),
uintptr(unsafe.Pointer(&recordInfo))))
return
}

140
vendor/github.com/go-ole/go-ole/safearrayconversion.go generated vendored Normal file
View File

@ -0,0 +1,140 @@
// Helper for converting SafeArray to array of objects.
package ole
import (
"unsafe"
)
type SafeArrayConversion struct {
Array *SafeArray
}
func (sac *SafeArrayConversion) ToStringArray() (strings []string) {
totalElements, _ := sac.TotalElements(0)
strings = make([]string, totalElements)
for i := int64(0); i < totalElements; i++ {
strings[int32(i)], _ = safeArrayGetElementString(sac.Array, i)
}
return
}
func (sac *SafeArrayConversion) ToByteArray() (bytes []byte) {
totalElements, _ := sac.TotalElements(0)
bytes = make([]byte, totalElements)
for i := int64(0); i < totalElements; i++ {
safeArrayGetElement(sac.Array, i, unsafe.Pointer(&bytes[int32(i)]))
}
return
}
func (sac *SafeArrayConversion) ToValueArray() (values []interface{}) {
totalElements, _ := sac.TotalElements(0)
values = make([]interface{}, totalElements)
vt, _ := safeArrayGetVartype(sac.Array)
for i := 0; i < int(totalElements); i++ {
switch VT(vt) {
case VT_BOOL:
var v bool
safeArrayGetElement(sac.Array, int64(i), unsafe.Pointer(&v))
values[i] = v
case VT_I1:
var v int8
safeArrayGetElement(sac.Array, int64(i), unsafe.Pointer(&v))
values[i] = v
case VT_I2:
var v int16
safeArrayGetElement(sac.Array, int64(i), unsafe.Pointer(&v))
values[i] = v
case VT_I4:
var v int32
safeArrayGetElement(sac.Array, int64(i), unsafe.Pointer(&v))
values[i] = v
case VT_I8:
var v int64
safeArrayGetElement(sac.Array, int64(i), unsafe.Pointer(&v))
values[i] = v
case VT_UI1:
var v uint8
safeArrayGetElement(sac.Array, int64(i), unsafe.Pointer(&v))
values[i] = v
case VT_UI2:
var v uint16
safeArrayGetElement(sac.Array, int64(i), unsafe.Pointer(&v))
values[i] = v
case VT_UI4:
var v uint32
safeArrayGetElement(sac.Array, int64(i), unsafe.Pointer(&v))
values[i] = v
case VT_UI8:
var v uint64
safeArrayGetElement(sac.Array, int64(i), unsafe.Pointer(&v))
values[i] = v
case VT_R4:
var v float32
safeArrayGetElement(sac.Array, int64(i), unsafe.Pointer(&v))
values[i] = v
case VT_R8:
var v float64
safeArrayGetElement(sac.Array, int64(i), unsafe.Pointer(&v))
values[i] = v
case VT_BSTR:
var v string
safeArrayGetElement(sac.Array, int64(i), unsafe.Pointer(&v))
values[i] = v
case VT_VARIANT:
var v VARIANT
safeArrayGetElement(sac.Array, int64(i), unsafe.Pointer(&v))
values[i] = v.Value()
default:
// TODO
}
}
return
}
func (sac *SafeArrayConversion) GetType() (varType uint16, err error) {
return safeArrayGetVartype(sac.Array)
}
func (sac *SafeArrayConversion) GetDimensions() (dimensions *uint32, err error) {
return safeArrayGetDim(sac.Array)
}
func (sac *SafeArrayConversion) GetSize() (length *uint32, err error) {
return safeArrayGetElementSize(sac.Array)
}
func (sac *SafeArrayConversion) TotalElements(index uint32) (totalElements int64, err error) {
if index < 1 {
index = 1
}
// Get array bounds
var LowerBounds int64
var UpperBounds int64
LowerBounds, err = safeArrayGetLBound(sac.Array, index)
if err != nil {
return
}
UpperBounds, err = safeArrayGetUBound(sac.Array, index)
if err != nil {
return
}
totalElements = UpperBounds - LowerBounds + 1
return
}
// Release Safe Array memory
func (sac *SafeArrayConversion) Release() {
safeArrayDestroy(sac.Array)
}

33
vendor/github.com/go-ole/go-ole/safearrayslices.go generated vendored Normal file
View File

@ -0,0 +1,33 @@
// +build windows
package ole
import (
"unsafe"
)
func safeArrayFromByteSlice(slice []byte) *SafeArray {
array, _ := safeArrayCreateVector(VT_UI1, 0, uint32(len(slice)))
if array == nil {
panic("Could not convert []byte to SAFEARRAY")
}
for i, v := range slice {
safeArrayPutElement(array, int64(i), uintptr(unsafe.Pointer(&v)))
}
return array
}
func safeArrayFromStringSlice(slice []string) *SafeArray {
array, _ := safeArrayCreateVector(VT_BSTR, 0, uint32(len(slice)))
if array == nil {
panic("Could not convert []string to SAFEARRAY")
}
// SysAllocStringLen(s)
for i, v := range slice {
safeArrayPutElement(array, int64(i), uintptr(unsafe.Pointer(SysAllocStringLen(v))))
}
return array
}

101
vendor/github.com/go-ole/go-ole/utility.go generated vendored Normal file
View File

@ -0,0 +1,101 @@
package ole
import (
"unicode/utf16"
"unsafe"
)
// ClassIDFrom retrieves class ID whether given is program ID or application string.
//
// Helper that provides check against both Class ID from Program ID and Class ID from string. It is
// faster, if you know which you are using, to use the individual functions, but this will check
// against available functions for you.
func ClassIDFrom(programID string) (classID *GUID, err error) {
classID, err = CLSIDFromProgID(programID)
if err != nil {
classID, err = CLSIDFromString(programID)
if err != nil {
return
}
}
return
}
// BytePtrToString converts byte pointer to a Go string.
func BytePtrToString(p *byte) string {
a := (*[10000]uint8)(unsafe.Pointer(p))
i := 0
for a[i] != 0 {
i++
}
return string(a[:i])
}
// UTF16PtrToString is alias for LpOleStrToString.
//
// Kept for compatibility reasons.
func UTF16PtrToString(p *uint16) string {
return LpOleStrToString(p)
}
// LpOleStrToString converts COM Unicode to Go string.
func LpOleStrToString(p *uint16) string {
if p == nil {
return ""
}
length := lpOleStrLen(p)
a := make([]uint16, length)
ptr := unsafe.Pointer(p)
for i := 0; i < int(length); i++ {
a[i] = *(*uint16)(ptr)
ptr = unsafe.Pointer(uintptr(ptr) + 2)
}
return string(utf16.Decode(a))
}
// BstrToString converts COM binary string to Go string.
func BstrToString(p *uint16) string {
if p == nil {
return ""
}
length := SysStringLen((*int16)(unsafe.Pointer(p)))
a := make([]uint16, length)
ptr := unsafe.Pointer(p)
for i := 0; i < int(length); i++ {
a[i] = *(*uint16)(ptr)
ptr = unsafe.Pointer(uintptr(ptr) + 2)
}
return string(utf16.Decode(a))
}
// lpOleStrLen returns the length of Unicode string.
func lpOleStrLen(p *uint16) (length int64) {
if p == nil {
return 0
}
ptr := unsafe.Pointer(p)
for i := 0; ; i++ {
if 0 == *(*uint16)(ptr) {
length = int64(i)
break
}
ptr = unsafe.Pointer(uintptr(ptr) + 2)
}
return
}
// convertHresultToError converts syscall to error, if call is unsuccessful.
func convertHresultToError(hr uintptr, r2 uintptr, ignore error) (err error) {
if hr != 0 {
err = NewError(hr)
}
return
}

16
vendor/github.com/go-ole/go-ole/variables.go generated vendored Normal file
View File

@ -0,0 +1,16 @@
// +build windows
package ole
import (
"syscall"
)
var (
modcombase = syscall.NewLazyDLL("combase.dll")
modkernel32, _ = syscall.LoadDLL("kernel32.dll")
modole32, _ = syscall.LoadDLL("ole32.dll")
modoleaut32, _ = syscall.LoadDLL("oleaut32.dll")
modmsvcrt, _ = syscall.LoadDLL("msvcrt.dll")
moduser32, _ = syscall.LoadDLL("user32.dll")
)

105
vendor/github.com/go-ole/go-ole/variant.go generated vendored Normal file
View File

@ -0,0 +1,105 @@
package ole
import "unsafe"
// NewVariant returns new variant based on type and value.
func NewVariant(vt VT, val int64) VARIANT {
return VARIANT{VT: vt, Val: val}
}
// ToIUnknown converts Variant to Unknown object.
func (v *VARIANT) ToIUnknown() *IUnknown {
if v.VT != VT_UNKNOWN {
return nil
}
return (*IUnknown)(unsafe.Pointer(uintptr(v.Val)))
}
// ToIDispatch converts variant to dispatch object.
func (v *VARIANT) ToIDispatch() *IDispatch {
if v.VT != VT_DISPATCH {
return nil
}
return (*IDispatch)(unsafe.Pointer(uintptr(v.Val)))
}
// ToArray converts variant to SafeArray helper.
func (v *VARIANT) ToArray() *SafeArrayConversion {
if v.VT != VT_SAFEARRAY {
if v.VT&VT_ARRAY == 0 {
return nil
}
}
var safeArray *SafeArray = (*SafeArray)(unsafe.Pointer(uintptr(v.Val)))
return &SafeArrayConversion{safeArray}
}
// ToString converts variant to Go string.
func (v *VARIANT) ToString() string {
if v.VT != VT_BSTR {
return ""
}
return BstrToString(*(**uint16)(unsafe.Pointer(&v.Val)))
}
// Clear the memory of variant object.
func (v *VARIANT) Clear() error {
return VariantClear(v)
}
// Value returns variant value based on its type.
//
// Currently supported types: 2- and 4-byte integers, strings, bools.
// Note that 64-bit integers, datetimes, and other types are stored as strings
// and will be returned as strings.
//
// Needs to be further converted, because this returns an interface{}.
func (v *VARIANT) Value() interface{} {
switch v.VT {
case VT_I1:
return int8(v.Val)
case VT_UI1:
return uint8(v.Val)
case VT_I2:
return int16(v.Val)
case VT_UI2:
return uint16(v.Val)
case VT_I4:
return int32(v.Val)
case VT_UI4:
return uint32(v.Val)
case VT_I8:
return int64(v.Val)
case VT_UI8:
return uint64(v.Val)
case VT_INT:
return int(v.Val)
case VT_UINT:
return uint(v.Val)
case VT_INT_PTR:
return uintptr(v.Val) // TODO
case VT_UINT_PTR:
return uintptr(v.Val)
case VT_R4:
return *(*float32)(unsafe.Pointer(&v.Val))
case VT_R8:
return *(*float64)(unsafe.Pointer(&v.Val))
case VT_BSTR:
return v.ToString()
case VT_DATE:
// VT_DATE type will either return float64 or time.Time.
d := float64(v.Val)
date, err := GetVariantDate(d)
if err != nil {
return d
}
return date
case VT_UNKNOWN:
return v.ToIUnknown()
case VT_DISPATCH:
return v.ToIDispatch()
case VT_BOOL:
return v.Val != 0
}
return nil
}

11
vendor/github.com/go-ole/go-ole/variant_386.go generated vendored Normal file
View File

@ -0,0 +1,11 @@
// +build 386
package ole
type VARIANT struct {
VT VT // 2
wReserved1 uint16 // 4
wReserved2 uint16 // 6
wReserved3 uint16 // 8
Val int64 // 16
}

12
vendor/github.com/go-ole/go-ole/variant_amd64.go generated vendored Normal file
View File

@ -0,0 +1,12 @@
// +build amd64
package ole
type VARIANT struct {
VT VT // 2
wReserved1 uint16 // 4
wReserved2 uint16 // 6
wReserved3 uint16 // 8
Val int64 // 16
_ [8]byte // 24
}

12
vendor/github.com/go-ole/go-ole/variant_s390x.go generated vendored Normal file
View File

@ -0,0 +1,12 @@
// +build s390x
package ole
type VARIANT struct {
VT VT // 2
wReserved1 uint16 // 4
wReserved2 uint16 // 6
wReserved3 uint16 // 8
Val int64 // 16
_ [8]byte // 24
}

58
vendor/github.com/go-ole/go-ole/vt_string.go generated vendored Normal file
View File

@ -0,0 +1,58 @@
// generated by stringer -output vt_string.go -type VT; DO NOT EDIT
package ole
import "fmt"
const (
_VT_name_0 = "VT_EMPTYVT_NULLVT_I2VT_I4VT_R4VT_R8VT_CYVT_DATEVT_BSTRVT_DISPATCHVT_ERRORVT_BOOLVT_VARIANTVT_UNKNOWNVT_DECIMAL"
_VT_name_1 = "VT_I1VT_UI1VT_UI2VT_UI4VT_I8VT_UI8VT_INTVT_UINTVT_VOIDVT_HRESULTVT_PTRVT_SAFEARRAYVT_CARRAYVT_USERDEFINEDVT_LPSTRVT_LPWSTR"
_VT_name_2 = "VT_RECORDVT_INT_PTRVT_UINT_PTR"
_VT_name_3 = "VT_FILETIMEVT_BLOBVT_STREAMVT_STORAGEVT_STREAMED_OBJECTVT_STORED_OBJECTVT_BLOB_OBJECTVT_CFVT_CLSID"
_VT_name_4 = "VT_BSTR_BLOBVT_VECTOR"
_VT_name_5 = "VT_ARRAY"
_VT_name_6 = "VT_BYREF"
_VT_name_7 = "VT_RESERVED"
_VT_name_8 = "VT_ILLEGAL"
)
var (
_VT_index_0 = [...]uint8{0, 8, 15, 20, 25, 30, 35, 40, 47, 54, 65, 73, 80, 90, 100, 110}
_VT_index_1 = [...]uint8{0, 5, 11, 17, 23, 28, 34, 40, 47, 54, 64, 70, 82, 91, 105, 113, 122}
_VT_index_2 = [...]uint8{0, 9, 19, 30}
_VT_index_3 = [...]uint8{0, 11, 18, 27, 37, 55, 71, 85, 90, 98}
_VT_index_4 = [...]uint8{0, 12, 21}
_VT_index_5 = [...]uint8{0, 8}
_VT_index_6 = [...]uint8{0, 8}
_VT_index_7 = [...]uint8{0, 11}
_VT_index_8 = [...]uint8{0, 10}
)
func (i VT) String() string {
switch {
case 0 <= i && i <= 14:
return _VT_name_0[_VT_index_0[i]:_VT_index_0[i+1]]
case 16 <= i && i <= 31:
i -= 16
return _VT_name_1[_VT_index_1[i]:_VT_index_1[i+1]]
case 36 <= i && i <= 38:
i -= 36
return _VT_name_2[_VT_index_2[i]:_VT_index_2[i+1]]
case 64 <= i && i <= 72:
i -= 64
return _VT_name_3[_VT_index_3[i]:_VT_index_3[i+1]]
case 4095 <= i && i <= 4096:
i -= 4095
return _VT_name_4[_VT_index_4[i]:_VT_index_4[i+1]]
case i == 8192:
return _VT_name_5
case i == 16384:
return _VT_name_6
case i == 32768:
return _VT_name_7
case i == 65535:
return _VT_name_8
default:
return fmt.Sprintf("VT(%d)", i)
}
}

99
vendor/github.com/go-ole/go-ole/winrt.go generated vendored Normal file
View File

@ -0,0 +1,99 @@
// +build windows
package ole
import (
"reflect"
"syscall"
"unicode/utf8"
"unsafe"
)
var (
procRoInitialize = modcombase.NewProc("RoInitialize")
procRoActivateInstance = modcombase.NewProc("RoActivateInstance")
procRoGetActivationFactory = modcombase.NewProc("RoGetActivationFactory")
procWindowsCreateString = modcombase.NewProc("WindowsCreateString")
procWindowsDeleteString = modcombase.NewProc("WindowsDeleteString")
procWindowsGetStringRawBuffer = modcombase.NewProc("WindowsGetStringRawBuffer")
)
func RoInitialize(thread_type uint32) (err error) {
hr, _, _ := procRoInitialize.Call(uintptr(thread_type))
if hr != 0 {
err = NewError(hr)
}
return
}
func RoActivateInstance(clsid string) (ins *IInspectable, err error) {
hClsid, err := NewHString(clsid)
if err != nil {
return nil, err
}
defer DeleteHString(hClsid)
hr, _, _ := procRoActivateInstance.Call(
uintptr(unsafe.Pointer(hClsid)),
uintptr(unsafe.Pointer(&ins)))
if hr != 0 {
err = NewError(hr)
}
return
}
func RoGetActivationFactory(clsid string, iid *GUID) (ins *IInspectable, err error) {
hClsid, err := NewHString(clsid)
if err != nil {
return nil, err
}
defer DeleteHString(hClsid)
hr, _, _ := procRoGetActivationFactory.Call(
uintptr(unsafe.Pointer(hClsid)),
uintptr(unsafe.Pointer(iid)),
uintptr(unsafe.Pointer(&ins)))
if hr != 0 {
err = NewError(hr)
}
return
}
// HString is handle string for pointers.
type HString uintptr
// NewHString returns a new HString for Go string.
func NewHString(s string) (hstring HString, err error) {
u16 := syscall.StringToUTF16Ptr(s)
len := uint32(utf8.RuneCountInString(s))
hr, _, _ := procWindowsCreateString.Call(
uintptr(unsafe.Pointer(u16)),
uintptr(len),
uintptr(unsafe.Pointer(&hstring)))
if hr != 0 {
err = NewError(hr)
}
return
}
// DeleteHString deletes HString.
func DeleteHString(hstring HString) (err error) {
hr, _, _ := procWindowsDeleteString.Call(uintptr(hstring))
if hr != 0 {
err = NewError(hr)
}
return
}
// String returns Go string value of HString.
func (h HString) String() string {
var u16buf uintptr
var u16len uint32
u16buf, _, _ = procWindowsGetStringRawBuffer.Call(
uintptr(h),
uintptr(unsafe.Pointer(&u16len)))
u16hdr := reflect.SliceHeader{Data: u16buf, Len: int(u16len), Cap: int(u16len)}
u16 := *(*[]uint16)(unsafe.Pointer(&u16hdr))
return syscall.UTF16ToString(u16)
}

36
vendor/github.com/go-ole/go-ole/winrt_doc.go generated vendored Normal file
View File

@ -0,0 +1,36 @@
// +build !windows
package ole
// RoInitialize
func RoInitialize(thread_type uint32) (err error) {
return NewError(E_NOTIMPL)
}
// RoActivateInstance
func RoActivateInstance(clsid string) (ins *IInspectable, err error) {
return nil, NewError(E_NOTIMPL)
}
// RoGetActivationFactory
func RoGetActivationFactory(clsid string, iid *GUID) (ins *IInspectable, err error) {
return nil, NewError(E_NOTIMPL)
}
// HString is handle string for pointers.
type HString uintptr
// NewHString returns a new HString for Go string.
func NewHString(s string) (hstring HString, err error) {
return HString(uintptr(0)), NewError(E_NOTIMPL)
}
// DeleteHString deletes HString.
func DeleteHString(hstring HString) (err error) {
return NewError(E_NOTIMPL)
}
// String returns Go string value of HString.
func (h HString) String() string {
return ""
}

61
vendor/github.com/shirou/gopsutil/LICENSE generated vendored Normal file
View File

@ -0,0 +1,61 @@
gopsutil is distributed under BSD license reproduced below.
Copyright (c) 2014, WAKAYAMA Shirou
All rights reserved.
Redistribution and use in source and binary forms, with or without modification,
are permitted provided that the following conditions are met:
* Redistributions of source code must retain the above copyright notice, this
list of conditions and the following disclaimer.
* Redistributions in binary form must reproduce the above copyright notice,
this list of conditions and the following disclaimer in the documentation
and/or other materials provided with the distribution.
* Neither the name of the gopsutil authors nor the names of its contributors
may be used to endorse or promote products derived from this software without
specific prior written permission.
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
(INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
-------
internal/common/binary.go in the gopsutil is copied and modifid from golang/encoding/binary.go.
Copyright (c) 2009 The Go Authors. All rights reserved.
Redistribution and use in source and binary forms, with or without
modification, are permitted provided that the following conditions are
met:
* Redistributions of source code must retain the above copyright
notice, this list of conditions and the following disclaimer.
* Redistributions in binary form must reproduce the above
copyright notice, this list of conditions and the following disclaimer
in the documentation and/or other materials provided with the
distribution.
* Neither the name of Google Inc. nor the names of its
contributors may be used to endorse or promote products derived from
this software without specific prior written permission.
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

189
vendor/github.com/shirou/gopsutil/cpu/cpu.go generated vendored Normal file
View File

@ -0,0 +1,189 @@
package cpu
import (
"context"
"encoding/json"
"fmt"
"runtime"
"strconv"
"strings"
"sync"
"time"
"github.com/shirou/gopsutil/internal/common"
)
// TimesStat contains the amounts of time the CPU has spent performing different
// kinds of work. Time units are in USER_HZ or Jiffies (typically hundredths of
// a second). It is based on linux /proc/stat file.
type TimesStat struct {
CPU string `json:"cpu"`
User float64 `json:"user"`
System float64 `json:"system"`
Idle float64 `json:"idle"`
Nice float64 `json:"nice"`
Iowait float64 `json:"iowait"`
Irq float64 `json:"irq"`
Softirq float64 `json:"softirq"`
Steal float64 `json:"steal"`
Guest float64 `json:"guest"`
GuestNice float64 `json:"guestNice"`
Stolen float64 `json:"stolen"`
}
type InfoStat struct {
CPU int32 `json:"cpu"`
VendorID string `json:"vendorId"`
Family string `json:"family"`
Model string `json:"model"`
Stepping int32 `json:"stepping"`
PhysicalID string `json:"physicalId"`
CoreID string `json:"coreId"`
Cores int32 `json:"cores"`
ModelName string `json:"modelName"`
Mhz float64 `json:"mhz"`
CacheSize int32 `json:"cacheSize"`
Flags []string `json:"flags"`
Microcode string `json:"microcode"`
}
type lastPercent struct {
sync.Mutex
lastCPUTimes []TimesStat
lastPerCPUTimes []TimesStat
}
var lastCPUPercent lastPercent
var invoke common.Invoker = common.Invoke{}
func init() {
lastCPUPercent.Lock()
lastCPUPercent.lastCPUTimes, _ = Times(false)
lastCPUPercent.lastPerCPUTimes, _ = Times(true)
lastCPUPercent.Unlock()
}
func Counts(logical bool) (int, error) {
return CountsWithContext(context.Background(), logical)
}
func CountsWithContext(ctx context.Context, logical bool) (int, error) {
return runtime.NumCPU(), nil
}
func (c TimesStat) String() string {
v := []string{
`"cpu":"` + c.CPU + `"`,
`"user":` + strconv.FormatFloat(c.User, 'f', 1, 64),
`"system":` + strconv.FormatFloat(c.System, 'f', 1, 64),
`"idle":` + strconv.FormatFloat(c.Idle, 'f', 1, 64),
`"nice":` + strconv.FormatFloat(c.Nice, 'f', 1, 64),
`"iowait":` + strconv.FormatFloat(c.Iowait, 'f', 1, 64),
`"irq":` + strconv.FormatFloat(c.Irq, 'f', 1, 64),
`"softirq":` + strconv.FormatFloat(c.Softirq, 'f', 1, 64),
`"steal":` + strconv.FormatFloat(c.Steal, 'f', 1, 64),
`"guest":` + strconv.FormatFloat(c.Guest, 'f', 1, 64),
`"guestNice":` + strconv.FormatFloat(c.GuestNice, 'f', 1, 64),
`"stolen":` + strconv.FormatFloat(c.Stolen, 'f', 1, 64),
}
return `{` + strings.Join(v, ",") + `}`
}
// Total returns the total number of seconds in a CPUTimesStat
func (c TimesStat) Total() float64 {
total := c.User + c.System + c.Nice + c.Iowait + c.Irq + c.Softirq + c.Steal +
c.Guest + c.GuestNice + c.Idle + c.Stolen
return total
}
func (c InfoStat) String() string {
s, _ := json.Marshal(c)
return string(s)
}
func getAllBusy(t TimesStat) (float64, float64) {
busy := t.User + t.System + t.Nice + t.Iowait + t.Irq +
t.Softirq + t.Steal + t.Guest + t.GuestNice + t.Stolen
return busy + t.Idle, busy
}
func calculateBusy(t1, t2 TimesStat) float64 {
t1All, t1Busy := getAllBusy(t1)
t2All, t2Busy := getAllBusy(t2)
if t2Busy <= t1Busy {
return 0
}
if t2All <= t1All {
return 1
}
return (t2Busy - t1Busy) / (t2All - t1All) * 100
}
func calculateAllBusy(t1, t2 []TimesStat) ([]float64, error) {
// Make sure the CPU measurements have the same length.
if len(t1) != len(t2) {
return nil, fmt.Errorf(
"received two CPU counts: %d != %d",
len(t1), len(t2),
)
}
ret := make([]float64, len(t1))
for i, t := range t2 {
ret[i] = calculateBusy(t1[i], t)
}
return ret, nil
}
// Percent calculates the percentage of cpu used either per CPU or combined.
// If an interval of 0 is given it will compare the current cpu times against the last call.
// Returns one value per cpu, or a single value if percpu is set to false.
func Percent(interval time.Duration, percpu bool) ([]float64, error) {
return PercentWithContext(context.Background(), interval, percpu)
}
func PercentWithContext(ctx context.Context, interval time.Duration, percpu bool) ([]float64, error) {
if interval <= 0 {
return percentUsedFromLastCall(percpu)
}
// Get CPU usage at the start of the interval.
cpuTimes1, err := Times(percpu)
if err != nil {
return nil, err
}
time.Sleep(interval)
// And at the end of the interval.
cpuTimes2, err := Times(percpu)
if err != nil {
return nil, err
}
return calculateAllBusy(cpuTimes1, cpuTimes2)
}
func percentUsedFromLastCall(percpu bool) ([]float64, error) {
cpuTimes, err := Times(percpu)
if err != nil {
return nil, err
}
lastCPUPercent.Lock()
defer lastCPUPercent.Unlock()
var lastTimes []TimesStat
if percpu {
lastTimes = lastCPUPercent.lastPerCPUTimes
lastCPUPercent.lastPerCPUTimes = cpuTimes
} else {
lastTimes = lastCPUPercent.lastCPUTimes
lastCPUPercent.lastCPUTimes = cpuTimes
}
if lastTimes == nil {
return nil, fmt.Errorf("error getting times for cpu percent. lastTimes was nil")
}
return calculateAllBusy(lastTimes, cpuTimes)
}

115
vendor/github.com/shirou/gopsutil/cpu/cpu_darwin.go generated vendored Normal file
View File

@ -0,0 +1,115 @@
// +build darwin
package cpu
import (
"context"
"os/exec"
"strconv"
"strings"
)
// sys/resource.h
const (
CPUser = 0
CPNice = 1
CPSys = 2
CPIntr = 3
CPIdle = 4
CPUStates = 5
)
// default value. from time.h
var ClocksPerSec = float64(128)
func Times(percpu bool) ([]TimesStat, error) {
return TimesWithContext(context.Background(), percpu)
}
func TimesWithContext(ctx context.Context, percpu bool) ([]TimesStat, error) {
if percpu {
return perCPUTimes()
}
return allCPUTimes()
}
// Returns only one CPUInfoStat on FreeBSD
func Info() ([]InfoStat, error) {
return InfoWithContext(context.Background())
}
func InfoWithContext(ctx context.Context) ([]InfoStat, error) {
var ret []InfoStat
sysctl, err := exec.LookPath("/usr/sbin/sysctl")
if err != nil {
return ret, err
}
out, err := invoke.CommandWithContext(ctx, sysctl, "machdep.cpu")
if err != nil {
return ret, err
}
c := InfoStat{}
for _, line := range strings.Split(string(out), "\n") {
values := strings.Fields(line)
if len(values) < 1 {
continue
}
t, err := strconv.ParseInt(values[1], 10, 64)
// err is not checked here because some value is string.
if strings.HasPrefix(line, "machdep.cpu.brand_string") {
c.ModelName = strings.Join(values[1:], " ")
} else if strings.HasPrefix(line, "machdep.cpu.family") {
c.Family = values[1]
} else if strings.HasPrefix(line, "machdep.cpu.model") {
c.Model = values[1]
} else if strings.HasPrefix(line, "machdep.cpu.stepping") {
if err != nil {
return ret, err
}
c.Stepping = int32(t)
} else if strings.HasPrefix(line, "machdep.cpu.features") {
for _, v := range values[1:] {
c.Flags = append(c.Flags, strings.ToLower(v))
}
} else if strings.HasPrefix(line, "machdep.cpu.leaf7_features") {
for _, v := range values[1:] {
c.Flags = append(c.Flags, strings.ToLower(v))
}
} else if strings.HasPrefix(line, "machdep.cpu.extfeatures") {
for _, v := range values[1:] {
c.Flags = append(c.Flags, strings.ToLower(v))
}
} else if strings.HasPrefix(line, "machdep.cpu.core_count") {
if err != nil {
return ret, err
}
c.Cores = int32(t)
} else if strings.HasPrefix(line, "machdep.cpu.cache.size") {
if err != nil {
return ret, err
}
c.CacheSize = int32(t)
} else if strings.HasPrefix(line, "machdep.cpu.vendor") {
c.VendorID = values[1]
}
}
// Use the rated frequency of the CPU. This is a static value and does not
// account for low power or Turbo Boost modes.
out, err = invoke.CommandWithContext(ctx, sysctl, "hw.cpufrequency")
if err != nil {
return ret, err
}
values := strings.Fields(string(out))
hz, err := strconv.ParseFloat(values[1], 64)
if err != nil {
return ret, err
}
c.Mhz = hz / 1000000.0
return append(ret, c), nil
}

111
vendor/github.com/shirou/gopsutil/cpu/cpu_darwin_cgo.go generated vendored Normal file
View File

@ -0,0 +1,111 @@
// +build darwin
// +build cgo
package cpu
/*
#include <stdlib.h>
#include <sys/sysctl.h>
#include <sys/mount.h>
#include <mach/mach_init.h>
#include <mach/mach_host.h>
#include <mach/host_info.h>
#if TARGET_OS_MAC
#include <libproc.h>
#endif
#include <mach/processor_info.h>
#include <mach/vm_map.h>
*/
import "C"
import (
"bytes"
"encoding/binary"
"fmt"
"unsafe"
)
// these CPU times for darwin is borrowed from influxdb/telegraf.
func perCPUTimes() ([]TimesStat, error) {
var (
count C.mach_msg_type_number_t
cpuload *C.processor_cpu_load_info_data_t
ncpu C.natural_t
)
status := C.host_processor_info(C.host_t(C.mach_host_self()),
C.PROCESSOR_CPU_LOAD_INFO,
&ncpu,
(*C.processor_info_array_t)(unsafe.Pointer(&cpuload)),
&count)
if status != C.KERN_SUCCESS {
return nil, fmt.Errorf("host_processor_info error=%d", status)
}
// jump through some cgo casting hoops and ensure we properly free
// the memory that cpuload points to
target := C.vm_map_t(C.mach_task_self_)
address := C.vm_address_t(uintptr(unsafe.Pointer(cpuload)))
defer C.vm_deallocate(target, address, C.vm_size_t(ncpu))
// the body of struct processor_cpu_load_info
// aka processor_cpu_load_info_data_t
var cpu_ticks [C.CPU_STATE_MAX]uint32
// copy the cpuload array to a []byte buffer
// where we can binary.Read the data
size := int(ncpu) * binary.Size(cpu_ticks)
buf := (*[1 << 30]byte)(unsafe.Pointer(cpuload))[:size:size]
bbuf := bytes.NewBuffer(buf)
var ret []TimesStat
for i := 0; i < int(ncpu); i++ {
err := binary.Read(bbuf, binary.LittleEndian, &cpu_ticks)
if err != nil {
return nil, err
}
c := TimesStat{
CPU: fmt.Sprintf("cpu%d", i),
User: float64(cpu_ticks[C.CPU_STATE_USER]) / ClocksPerSec,
System: float64(cpu_ticks[C.CPU_STATE_SYSTEM]) / ClocksPerSec,
Nice: float64(cpu_ticks[C.CPU_STATE_NICE]) / ClocksPerSec,
Idle: float64(cpu_ticks[C.CPU_STATE_IDLE]) / ClocksPerSec,
}
ret = append(ret, c)
}
return ret, nil
}
func allCPUTimes() ([]TimesStat, error) {
var count C.mach_msg_type_number_t
var cpuload C.host_cpu_load_info_data_t
count = C.HOST_CPU_LOAD_INFO_COUNT
status := C.host_statistics(C.host_t(C.mach_host_self()),
C.HOST_CPU_LOAD_INFO,
C.host_info_t(unsafe.Pointer(&cpuload)),
&count)
if status != C.KERN_SUCCESS {
return nil, fmt.Errorf("host_statistics error=%d", status)
}
c := TimesStat{
CPU: "cpu-total",
User: float64(cpuload.cpu_ticks[C.CPU_STATE_USER]) / ClocksPerSec,
System: float64(cpuload.cpu_ticks[C.CPU_STATE_SYSTEM]) / ClocksPerSec,
Nice: float64(cpuload.cpu_ticks[C.CPU_STATE_NICE]) / ClocksPerSec,
Idle: float64(cpuload.cpu_ticks[C.CPU_STATE_IDLE]) / ClocksPerSec,
}
return []TimesStat{c}, nil
}

14
vendor/github.com/shirou/gopsutil/cpu/cpu_darwin_nocgo.go generated vendored Normal file
View File

@ -0,0 +1,14 @@
// +build darwin
// +build !cgo
package cpu
import "github.com/shirou/gopsutil/internal/common"
func perCPUTimes() ([]TimesStat, error) {
return []TimesStat{}, common.ErrNotImplementedError
}
func allCPUTimes() ([]TimesStat, error) {
return []TimesStat{}, common.ErrNotImplementedError
}

25
vendor/github.com/shirou/gopsutil/cpu/cpu_fallback.go generated vendored Normal file
View File

@ -0,0 +1,25 @@
// +build !darwin,!linux,!freebsd,!openbsd,!solaris,!windows
package cpu
import (
"context"
"github.com/shirou/gopsutil/internal/common"
)
func Times(percpu bool) ([]TimesStat, error) {
return TimesWithContext(context.Background(), percpu)
}
func TimesWithContext(ctx context.Context, percpu bool) ([]TimesStat, error) {
return []TimesStat{}, common.ErrNotImplementedError
}
func Info() ([]InfoStat, error) {
return InfoWithContext(context.Background())
}
func InfoWithContext(ctx context.Context) ([]InfoStat, error) {
return []InfoStat{}, common.ErrNotImplementedError
}

168
vendor/github.com/shirou/gopsutil/cpu/cpu_freebsd.go generated vendored Normal file
View File

@ -0,0 +1,168 @@
package cpu
import (
"context"
"fmt"
"os/exec"
"reflect"
"regexp"
"strconv"
"strings"
"unsafe"
"github.com/shirou/gopsutil/internal/common"
"golang.org/x/sys/unix"
)
var ClocksPerSec = float64(128)
var cpuMatch = regexp.MustCompile(`^CPU:`)
var originMatch = regexp.MustCompile(`Origin\s*=\s*"(.+)"\s+Id\s*=\s*(.+)\s+Family\s*=\s*(.+)\s+Model\s*=\s*(.+)\s+Stepping\s*=\s*(.+)`)
var featuresMatch = regexp.MustCompile(`Features=.+<(.+)>`)
var featuresMatch2 = regexp.MustCompile(`Features2=[a-f\dx]+<(.+)>`)
var cpuEnd = regexp.MustCompile(`^Trying to mount root`)
var cpuCores = regexp.MustCompile(`FreeBSD/SMP: (\d*) package\(s\) x (\d*) core\(s\)`)
var cpuTimesSize int
var emptyTimes cpuTimes
func init() {
getconf, err := exec.LookPath("/usr/bin/getconf")
if err != nil {
return
}
out, err := invoke.Command(getconf, "CLK_TCK")
// ignore errors
if err == nil {
i, err := strconv.ParseFloat(strings.TrimSpace(string(out)), 64)
if err == nil {
ClocksPerSec = float64(i)
}
}
}
func timeStat(name string, t *cpuTimes) *TimesStat {
return &TimesStat{
User: float64(t.User) / ClocksPerSec,
Nice: float64(t.Nice) / ClocksPerSec,
System: float64(t.Sys) / ClocksPerSec,
Idle: float64(t.Idle) / ClocksPerSec,
Irq: float64(t.Intr) / ClocksPerSec,
CPU: name,
}
}
func Times(percpu bool) ([]TimesStat, error) {
return TimesWithContext(context.Background(), percpu)
}
func TimesWithContext(ctx context.Context, percpu bool) ([]TimesStat, error) {
if percpu {
buf, err := unix.SysctlRaw("kern.cp_times")
if err != nil {
return nil, err
}
// We can't do this in init due to the conflict with cpu.init()
if cpuTimesSize == 0 {
cpuTimesSize = int(reflect.TypeOf(cpuTimes{}).Size())
}
ncpus := len(buf) / cpuTimesSize
ret := make([]TimesStat, 0, ncpus)
for i := 0; i < ncpus; i++ {
times := (*cpuTimes)(unsafe.Pointer(&buf[i*cpuTimesSize]))
if *times == emptyTimes {
// CPU not present
continue
}
ret = append(ret, *timeStat(fmt.Sprintf("cpu%d", len(ret)), times))
}
return ret, nil
}
buf, err := unix.SysctlRaw("kern.cp_time")
if err != nil {
return nil, err
}
times := (*cpuTimes)(unsafe.Pointer(&buf[0]))
return []TimesStat{*timeStat("cpu-total", times)}, nil
}
// Returns only one InfoStat on FreeBSD. The information regarding core
// count, however is accurate and it is assumed that all InfoStat attributes
// are the same across CPUs.
func Info() ([]InfoStat, error) {
return InfoWithContext(context.Background())
}
func InfoWithContext(ctx context.Context) ([]InfoStat, error) {
const dmesgBoot = "/var/run/dmesg.boot"
c, num, err := parseDmesgBoot(dmesgBoot)
if err != nil {
return nil, err
}
var u32 uint32
if u32, err = unix.SysctlUint32("hw.clockrate"); err != nil {
return nil, err
}
c.Mhz = float64(u32)
if u32, err = unix.SysctlUint32("hw.ncpu"); err != nil {
return nil, err
}
c.Cores = int32(u32)
if c.ModelName, err = unix.Sysctl("hw.model"); err != nil {
return nil, err
}
ret := make([]InfoStat, num)
for i := 0; i < num; i++ {
ret[i] = c
}
return ret, nil
}
func parseDmesgBoot(fileName string) (InfoStat, int, error) {
c := InfoStat{}
lines, _ := common.ReadLines(fileName)
cpuNum := 1 // default cpu num is 1
for _, line := range lines {
if matches := cpuEnd.FindStringSubmatch(line); matches != nil {
break
} else if matches := originMatch.FindStringSubmatch(line); matches != nil {
c.VendorID = matches[1]
c.Family = matches[3]
c.Model = matches[4]
t, err := strconv.ParseInt(matches[5], 10, 32)
if err != nil {
return c, 0, fmt.Errorf("unable to parse FreeBSD CPU stepping information from %q: %v", line, err)
}
c.Stepping = int32(t)
} else if matches := featuresMatch.FindStringSubmatch(line); matches != nil {
for _, v := range strings.Split(matches[1], ",") {
c.Flags = append(c.Flags, strings.ToLower(v))
}
} else if matches := featuresMatch2.FindStringSubmatch(line); matches != nil {
for _, v := range strings.Split(matches[1], ",") {
c.Flags = append(c.Flags, strings.ToLower(v))
}
} else if matches := cpuCores.FindStringSubmatch(line); matches != nil {
t, err := strconv.ParseInt(matches[1], 10, 32)
if err != nil {
return c, 0, fmt.Errorf("unable to parse FreeBSD CPU Nums from %q: %v", line, err)
}
cpuNum = int(t)
t2, err := strconv.ParseInt(matches[2], 10, 32)
if err != nil {
return c, 0, fmt.Errorf("unable to parse FreeBSD CPU cores from %q: %v", line, err)
}
c.Cores = int32(t2)
}
}
return c, cpuNum, nil
}

9
vendor/github.com/shirou/gopsutil/cpu/cpu_freebsd_386.go generated vendored Normal file
View File

@ -0,0 +1,9 @@
package cpu
type cpuTimes struct {
User uint32
Nice uint32
Sys uint32
Intr uint32
Idle uint32
}

9
vendor/github.com/shirou/gopsutil/cpu/cpu_freebsd_amd64.go generated vendored Normal file
View File

@ -0,0 +1,9 @@
package cpu
type cpuTimes struct {
User uint64
Nice uint64
Sys uint64
Intr uint64
Idle uint64
}

284
vendor/github.com/shirou/gopsutil/cpu/cpu_linux.go generated vendored Normal file
View File

@ -0,0 +1,284 @@
// +build linux
package cpu
import (
"context"
"errors"
"fmt"
"os/exec"
"strconv"
"strings"
"github.com/shirou/gopsutil/internal/common"
)
var CPUTick = float64(100)
func init() {
getconf, err := exec.LookPath("/usr/bin/getconf")
if err != nil {
return
}
out, err := invoke.CommandWithContext(context.Background(), getconf, "CLK_TCK")
// ignore errors
if err == nil {
i, err := strconv.ParseFloat(strings.TrimSpace(string(out)), 64)
if err == nil {
CPUTick = i
}
}
}
func Times(percpu bool) ([]TimesStat, error) {
return TimesWithContext(context.Background(), percpu)
}
func TimesWithContext(ctx context.Context, percpu bool) ([]TimesStat, error) {
filename := common.HostProc("stat")
var lines = []string{}
if percpu {
statlines, err := common.ReadLines(filename)
if err != nil || len(statlines) < 2 {
return []TimesStat{}, nil
}
for _, line := range statlines[1:] {
if !strings.HasPrefix(line, "cpu") {
break
}
lines = append(lines, line)
}
} else {
lines, _ = common.ReadLinesOffsetN(filename, 0, 1)
}
ret := make([]TimesStat, 0, len(lines))
for _, line := range lines {
ct, err := parseStatLine(line)
if err != nil {
continue
}
ret = append(ret, *ct)
}
return ret, nil
}
func sysCPUPath(cpu int32, relPath string) string {
return common.HostSys(fmt.Sprintf("devices/system/cpu/cpu%d", cpu), relPath)
}
func finishCPUInfo(c *InfoStat) error {
var lines []string
var err error
var value float64
if len(c.CoreID) == 0 {
lines, err = common.ReadLines(sysCPUPath(c.CPU, "topology/core_id"))
if err == nil {
c.CoreID = lines[0]
}
}
// override the value of c.Mhz with cpufreq/cpuinfo_max_freq regardless
// of the value from /proc/cpuinfo because we want to report the maximum
// clock-speed of the CPU for c.Mhz, matching the behaviour of Windows
lines, err = common.ReadLines(sysCPUPath(c.CPU, "cpufreq/cpuinfo_max_freq"))
// if we encounter errors below such as there are no cpuinfo_max_freq file,
// we just ignore. so let Mhz is 0.
if err != nil {
return nil
}
value, err = strconv.ParseFloat(lines[0], 64)
if err != nil {
return nil
}
c.Mhz = value / 1000.0 // value is in kHz
if c.Mhz > 9999 {
c.Mhz = c.Mhz / 1000.0 // value in Hz
}
return nil
}
// CPUInfo on linux will return 1 item per physical thread.
//
// CPUs have three levels of counting: sockets, cores, threads.
// Cores with HyperThreading count as having 2 threads per core.
// Sockets often come with many physical CPU cores.
// For example a single socket board with two cores each with HT will
// return 4 CPUInfoStat structs on Linux and the "Cores" field set to 1.
func Info() ([]InfoStat, error) {
return InfoWithContext(context.Background())
}
func InfoWithContext(ctx context.Context) ([]InfoStat, error) {
filename := common.HostProc("cpuinfo")
lines, _ := common.ReadLines(filename)
var ret []InfoStat
var processorName string
c := InfoStat{CPU: -1, Cores: 1}
for _, line := range lines {
fields := strings.Split(line, ":")
if len(fields) < 2 {
continue
}
key := strings.TrimSpace(fields[0])
value := strings.TrimSpace(fields[1])
switch key {
case "Processor":
processorName = value
case "processor":
if c.CPU >= 0 {
err := finishCPUInfo(&c)
if err != nil {
return ret, err
}
ret = append(ret, c)
}
c = InfoStat{Cores: 1, ModelName: processorName}
t, err := strconv.ParseInt(value, 10, 64)
if err != nil {
return ret, err
}
c.CPU = int32(t)
case "vendorId", "vendor_id":
c.VendorID = value
case "cpu family":
c.Family = value
case "model":
c.Model = value
case "model name", "cpu":
c.ModelName = value
if strings.Contains(value, "POWER8") ||
strings.Contains(value, "POWER7") {
c.Model = strings.Split(value, " ")[0]
c.Family = "POWER"
c.VendorID = "IBM"
}
case "stepping", "revision":
val := value
if key == "revision" {
val = strings.Split(value, ".")[0]
}
t, err := strconv.ParseInt(val, 10, 64)
if err != nil {
return ret, err
}
c.Stepping = int32(t)
case "cpu MHz", "clock":
// treat this as the fallback value, thus we ignore error
if t, err := strconv.ParseFloat(strings.Replace(value, "MHz", "", 1), 64); err == nil {
c.Mhz = t
}
case "cache size":
t, err := strconv.ParseInt(strings.Replace(value, " KB", "", 1), 10, 64)
if err != nil {
return ret, err
}
c.CacheSize = int32(t)
case "physical id":
c.PhysicalID = value
case "core id":
c.CoreID = value
case "flags", "Features":
c.Flags = strings.FieldsFunc(value, func(r rune) bool {
return r == ',' || r == ' '
})
case "microcode":
c.Microcode = value
}
}
if c.CPU >= 0 {
err := finishCPUInfo(&c)
if err != nil {
return ret, err
}
ret = append(ret, c)
}
return ret, nil
}
func parseStatLine(line string) (*TimesStat, error) {
fields := strings.Fields(line)
if len(fields) == 0 {
return nil, errors.New("stat does not contain cpu info")
}
if strings.HasPrefix(fields[0], "cpu") == false {
return nil, errors.New("not contain cpu")
}
cpu := fields[0]
if cpu == "cpu" {
cpu = "cpu-total"
}
user, err := strconv.ParseFloat(fields[1], 64)
if err != nil {
return nil, err
}
nice, err := strconv.ParseFloat(fields[2], 64)
if err != nil {
return nil, err
}
system, err := strconv.ParseFloat(fields[3], 64)
if err != nil {
return nil, err
}
idle, err := strconv.ParseFloat(fields[4], 64)
if err != nil {
return nil, err
}
iowait, err := strconv.ParseFloat(fields[5], 64)
if err != nil {
return nil, err
}
irq, err := strconv.ParseFloat(fields[6], 64)
if err != nil {
return nil, err
}
softirq, err := strconv.ParseFloat(fields[7], 64)
if err != nil {
return nil, err
}
ct := &TimesStat{
CPU: cpu,
User: user / CPUTick,
Nice: nice / CPUTick,
System: system / CPUTick,
Idle: idle / CPUTick,
Iowait: iowait / CPUTick,
Irq: irq / CPUTick,
Softirq: softirq / CPUTick,
}
if len(fields) > 8 { // Linux >= 2.6.11
steal, err := strconv.ParseFloat(fields[8], 64)
if err != nil {
return nil, err
}
ct.Steal = steal / CPUTick
}
if len(fields) > 9 { // Linux >= 2.6.24
guest, err := strconv.ParseFloat(fields[9], 64)
if err != nil {
return nil, err
}
ct.Guest = guest / CPUTick
}
if len(fields) > 10 { // Linux >= 3.2.0
guestNice, err := strconv.ParseFloat(fields[10], 64)
if err != nil {
return nil, err
}
ct.GuestNice = guestNice / CPUTick
}
return ct, nil
}

119
vendor/github.com/shirou/gopsutil/cpu/cpu_openbsd.go generated vendored Normal file
View File

@ -0,0 +1,119 @@
// +build openbsd
package cpu
import (
"bytes"
"context"
"encoding/binary"
"fmt"
"os/exec"
"strconv"
"strings"
"github.com/shirou/gopsutil/internal/common"
"golang.org/x/sys/unix"
)
// sys/sched.h
const (
CPUser = 0
CPNice = 1
CPSys = 2
CPIntr = 3
CPIdle = 4
CPUStates = 5
)
// sys/sysctl.h
const (
CTLKern = 1 // "high kernel": proc, limits
KernCptime = 40 // KERN_CPTIME
KernCptime2 = 71 // KERN_CPTIME2
)
var ClocksPerSec = float64(128)
func init() {
getconf, err := exec.LookPath("/usr/bin/getconf")
if err != nil {
return
}
out, err := invoke.Command(getconf, "CLK_TCK")
// ignore errors
if err == nil {
i, err := strconv.ParseFloat(strings.TrimSpace(string(out)), 64)
if err == nil {
ClocksPerSec = float64(i)
}
}
}
func Times(percpu bool) ([]TimesStat, error) {
return TimesWithContext(context.Background(), percpu)
}
func TimesWithContext(ctx context.Context, percpu bool) ([]TimesStat, error) {
var ret []TimesStat
var ncpu int
if percpu {
ncpu, _ = Counts(true)
} else {
ncpu = 1
}
for i := 0; i < ncpu; i++ {
var cpuTimes [CPUStates]int64
var mib []int32
if percpu {
mib = []int32{CTLKern, KernCptime}
} else {
mib = []int32{CTLKern, KernCptime2, int32(i)}
}
buf, _, err := common.CallSyscall(mib)
if err != nil {
return ret, err
}
br := bytes.NewReader(buf)
err = binary.Read(br, binary.LittleEndian, &cpuTimes)
if err != nil {
return ret, err
}
c := TimesStat{
User: float64(cpuTimes[CPUser]) / ClocksPerSec,
Nice: float64(cpuTimes[CPNice]) / ClocksPerSec,
System: float64(cpuTimes[CPSys]) / ClocksPerSec,
Idle: float64(cpuTimes[CPIdle]) / ClocksPerSec,
Irq: float64(cpuTimes[CPIntr]) / ClocksPerSec,
}
if !percpu {
c.CPU = "cpu-total"
} else {
c.CPU = fmt.Sprintf("cpu%d", i)
}
ret = append(ret, c)
}
return ret, nil
}
// Returns only one (minimal) CPUInfoStat on OpenBSD
func Info() ([]InfoStat, error) {
return InfoWithContext(context.Background())
}
func InfoWithContext(ctx context.Context) ([]InfoStat, error) {
var ret []InfoStat
c := InfoStat{}
v, err := unix.Sysctl("hw.model")
if err != nil {
return nil, err
}
c.ModelName = v
return append(ret, c), nil
}

Some files were not shown because too many files have changed in this diff Show More