func getFriendlyName(portname string) string {
// this method panics a lot and i'm not sure why, just catch
// the panic and return empty list
defer func() {
if e := recover(); e != nil {
// e is the interface{} typed-value we passed to panic()
log.Println("Got panic: ", e) // Prints "Whoops: boom!"
}
}()
var friendlyName string
// init COM, oh yeah
ole.CoInitialize(0)
defer ole.CoUninitialize()
unknown, _ := oleutil.CreateObject("WbemScripting.SWbemLocator")
defer unknown.Release()
wmi, _ := unknown.QueryInterface(ole.IID_IDispatch)
defer wmi.Release()
// service is a SWbemServices
serviceRaw, _ := oleutil.CallMethod(wmi, "ConnectServer")
service := serviceRaw.ToIDispatch()
defer service.Release()
// result is a SWBemObjectSet
//pname := syscall.StringToUTF16("SELECT * FROM Win32_PnPEntity where Name like '%" + "COM35" + "%'")
pname := "SELECT * FROM Win32_PnPEntity where Name like '%" + portname + "%'"
resultRaw, _ := oleutil.CallMethod(service, "ExecQuery", pname)
result := resultRaw.ToIDispatch()
defer result.Release()
countVar, _ := oleutil.GetProperty(result, "Count")
count := int(countVar.Val)
for i := 0; i < count; i++ {
// item is a SWbemObject, but really a Win32_Process
itemRaw, _ := oleutil.CallMethod(result, "ItemIndex", i)
item := itemRaw.ToIDispatch()
defer item.Release()
asString, _ := oleutil.GetProperty(item, "Name")
println(asString.ToString())
friendlyName = asString.ToString()
}
return friendlyName
}
func QueryWmiFields(query string, fields []string) ([]map[string]string, error) {
if len(fields) == 1 && fields[0] == "*" {
logging.Errorf("`select * ` not supported, need to address fields explicitly.")
return nil, fmt.Errorf("`select * ` not supported, need to address fields explicitly.")
}
resultRaw, err := oleutil.CallMethod(wmi_service, "ExecQuery", query)
if err != nil {
logging.Error("ExecQuery Failed: ", err)
return nil, fmt.Errorf("ExecQuery Failed: %v", err)
}
result := resultRaw.ToIDispatch()
defer result.Release()
countVar, err := oleutil.GetProperty(result, "Count")
if err != nil {
logging.Errorf("Get result count Failed: %v", err)
return nil, fmt.Errorf("Get result count Failed: %v", err)
}
count := int(countVar.Val)
resultMap := []map[string]string{}
for i := 0; i < count; i++ {
itemMap := make(map[string]string)
itemRaw, err := oleutil.CallMethod(result, "ItemIndex", i)
if err != nil {
return nil, fmt.Errorf("ItemIndex Failed: %v", err)
}
item := itemRaw.ToIDispatch()
defer item.Release()
for _, field := range fields {
asString, err := oleutil.GetProperty(item, field)
if err == nil {
itemMap[field] = fmt.Sprintf("%v", asString.Value())
} else {
fmt.Println(err)
}
}
resultMap = append(resultMap, itemMap)
logging.Tracef("wmi query result: %+v", itemMap)
}
logging.Tracef("wmi query result count: %d", len(resultMap))
return resultMap, nil
}
func (c *win_wmi_collector) query(query string, fields []string) ([]map[string]string, error) {
if c.service != nil {
resultRaw, err := oleutil.CallMethod(c.service, "ExecQuery", query)
if err != nil {
logging.Error("ExecQuery Failed: ", err)
return nil, fmt.Errorf("ExecQuery Failed: %v", err)
}
result := resultRaw.ToIDispatch()
defer result.Release()
countVar, err := oleutil.GetProperty(result, "Count")
if err != nil {
logging.Error("Get result count Failed: ", err)
return nil, fmt.Errorf("Get result count Failed: %v", err)
}
count := int(countVar.Val)
resultMap := []map[string]string{}
for i := 0; i < count; i++ {
itemMap := make(map[string]string)
itemRaw, err := oleutil.CallMethod(result, "ItemIndex", i)
if err != nil {
return nil, fmt.Errorf("ItemIndex Failed: %v", err)
}
item := itemRaw.ToIDispatch()
defer item.Release()
for _, field := range fields {
asString, err := oleutil.GetProperty(item, field)
if err == nil {
itemMap[field] = fmt.Sprintf("%v", asString.Value())
} else {
logging.Errorf("cannot find field in SWbemObject: %v", err)
}
}
resultMap = append(resultMap, itemMap)
logging.Tracef("wmi query result: %+v", itemMap)
}
logging.Tracef("wmi query result count: %d", len(resultMap))
return resultMap, nil
} else {
logging.Error("win_wmi_collector c.service is nil")
return nil, fmt.Errorf("win_wmi_collector c.service is nil")
}
}
func (c *AdodbConn) Prepare(query string) (driver.Stmt, error) {
unknown, err := oleutil.CreateObject("ADODB.Command")
if err != nil {
return nil, err
}
s, err := unknown.QueryInterface(ole.IID_IDispatch)
if err != nil {
return nil, err
}
_, err = oleutil.PutProperty(s, "ActiveConnection", c.db)
if err != nil {
return nil, err
}
_, err = oleutil.PutProperty(s, "CommandText", query)
if err != nil {
return nil, err
}
_, err = oleutil.PutProperty(s, "CommandType", 1)
if err != nil {
return nil, err
}
_, err = oleutil.PutProperty(s, "Prepared", true)
if err != nil {
return nil, err
}
val, err := oleutil.GetProperty(s, "Parameters")
if err != nil {
return nil, err
}
return &AdodbStmt{c, s, val.ToIDispatch(), nil}, nil
}
func (pt *PowerPoint) open(inFile string) (err error) {
ole.CoInitialize(0)
var unknown *ole.IUnknown
unknown, err = oleutil.CreateObject("PowerPoint.Application")
if err != nil {
return
}
pt.app, err = unknown.QueryInterface(ole.IID_IDispatch)
if err != nil {
return
}
_, err = oleutil.PutProperty(pt.app, "DisplayAlerts", 1)
if err != nil {
return
}
pt.presentations, err = oleutil.GetProperty(pt.app, "Presentations")
if err != nil {
return
}
pt.ppt, err = oleutil.CallMethod(pt.presentations.ToIDispatch(), "Open", inFile, -1, 0, 0)
if err != nil {
return
}
return
}
func (rc *AdodbRows) Columns() []string {
if rc.nc != len(rc.cols) {
unknown, err := oleutil.GetProperty(rc.rc, "Fields")
if err != nil {
return []string{}
}
fields := unknown.ToIDispatch()
defer fields.Release()
val, err := oleutil.GetProperty(fields, "Count")
if err != nil {
return []string{}
}
rc.nc = int(val.Val)
rc.cols = make([]string, rc.nc)
for i := 0; i < rc.nc; i++ {
var varval ole.VARIANT
varval.VT = ole.VT_I4
varval.Val = int64(i)
val, err := oleutil.CallMethod(fields, "Item", &varval)
if err != nil {
return []string{}
}
item := val.ToIDispatch()
if err != nil {
return []string{}
}
name, err := oleutil.GetProperty(item, "Name")
if err != nil {
return []string{}
}
rc.cols[i] = name.ToString()
item.Release()
}
}
return rc.cols
}
func (s *AdodbStmt) NumInput() int {
if s.b != nil {
return len(s.b)
}
_, err := oleutil.CallMethod(s.ps, "Refresh")
if err != nil {
return -1
}
val, err := oleutil.GetProperty(s.ps, "Count")
if err != nil {
return -1
}
c := int(val.Val)
return c
}
func (wd *Word) open(inFile string) (err error) {
ole.CoInitialize(0)
var unknown *ole.IUnknown
unknown, err = oleutil.CreateObject("Word.Application")
if err != nil {
return
}
wd.app, err = unknown.QueryInterface(ole.IID_IDispatch)
if err != nil {
return
}
_, err = oleutil.PutProperty(wd.app, "Visible", false)
if err != nil {
return
}
_, err = oleutil.PutProperty(wd.app, "DisplayAlerts", 0)
if err != nil {
return
}
wd.documents, err = oleutil.GetProperty(wd.app, "Documents")
if err != nil {
return
}
wd.doc, err = oleutil.CallMethod(wd.documents.ToIDispatch(), "Open", inFile)
if err != nil {
return
}
return
}
func (rc *AdodbRows) Next(dest []driver.Value) error {
unknown, err := oleutil.GetProperty(rc.rc, "EOF")
if err != nil {
return io.EOF
}
if unknown.Val != 0 {
return io.EOF
}
unknown, err = oleutil.GetProperty(rc.rc, "Fields")
if err != nil {
return err
}
fields := unknown.ToIDispatch()
defer fields.Release()
for i := range dest {
var varval ole.VARIANT
varval.VT = ole.VT_I4
varval.Val = int64(i)
val, err := oleutil.CallMethod(fields, "Item", &varval)
if err != nil {
return err
}
field := val.ToIDispatch()
defer field.Release()
typ, err := oleutil.GetProperty(field, "Type")
if err != nil {
return err
}
val, err = oleutil.GetProperty(field, "Value")
if err != nil {
return err
}
sc, err := oleutil.GetProperty(field, "NumericScale")
field.Release()
if val.VT == 1 /* VT_NULL */ {
dest[i] = nil
continue
}
switch typ.Val {
case 0: // ADEMPTY
dest[i] = nil
case 2: // ADSMALLINT
dest[i] = int64(int16(val.Val))
case 3: // ADINTEGER
dest[i] = int64(int32(val.Val))
case 4: // ADSINGLE
dest[i] = float64(math.Float32frombits(uint32(val.Val)))
case 5: // ADDOUBLE
dest[i] = math.Float64frombits(uint64(val.Val))
case 6: // ADCURRENCY
dest[i] = float64(val.Val) / 10000
case 7: // ADDATE
// see http://blogs.msdn.com/b/ericlippert/archive/2003/09/16/eric-s-complete-guide-to-vt-date.aspx
d, t := math.Modf(math.Float64frombits(uint64(val.Val)))
t = math.Abs(t)
dest[i] = time.Date(1899, 12, 30+int(d), 0, 0, int(t*86400), 0, time.Local)
case 8: // ADBSTR
dest[i] = val.ToString()
case 9: // ADIDISPATCH
dest[i] = val.ToIDispatch()
case 10: // ADERROR
// TODO
case 11: // ADBOOLEAN
dest[i] = val.Val != 0
case 12: // ADVARIANT
dest[i] = val
case 13: // ADIUNKNOWN
dest[i] = val.ToIUnknown()
case 14: // ADDECIMAL
dest[i] = float64(val.Val)
case 16: // ADTINYINT
dest[i] = int8(val.Val)
case 17: // ADUNSIGNEDTINYINT
dest[i] = uint8(val.Val)
case 18: // ADUNSIGNEDSMALLINT
dest[i] = uint16(val.Val)
case 19: // ADUNSIGNEDINT
dest[i] = uint32(val.Val)
case 20: // ADBIGINT
dest[i] = big.NewInt(val.Val)
case 21: // ADUNSIGNEDBIGINT
// TODO
case 72: // ADGUID
dest[i] = val.ToString()
case 128: // ADBINARY
sa := (*ole.SAFEARRAY)(unsafe.Pointer(uintptr(val.Val)))
dest[i] = (*[1 << 30]byte)(unsafe.Pointer(uintptr(sa.Data)))[0:sa.Bounds.Elements]
case 129: // ADCHAR
dest[i] = val.ToString() //uint8(val.Val)
case 130: // ADWCHAR
dest[i] = val.ToString() //uint16(val.Val)
case 131: // ADNUMERIC
sub := math.Pow(10, float64(sc.Val))
dest[i] = float64(float64(val.Val) / sub)
case 132: // ADUSERDEFINED
dest[i] = uintptr(val.Val)
case 133: // ADDBDATE
// see http://blogs.msdn.com/b/ericlippert/archive/2003/09/16/eric-s-complete-guide-to-vt-date.aspx
d := math.Float64frombits(uint64(val.Val))
dest[i] = time.Date(1899, 12, 30+int(d), 0, 0, 0, 0, time.Local)
case 134: // ADDBTIME
t := math.Float64frombits(uint64(val.Val))
dest[i] = time.Date(0, 1, 1, 0, 0, int(t*86400), 0, time.Local)
case 135: // ADDBTIMESTAMP
d, t := math.Modf(math.Float64frombits(uint64(val.Val)))
t = math.Abs(t)
dest[i] = time.Date(1899, 12, 30+int(d), 0, 0, int(t*86400), 0, time.Local)
case 136: // ADCHAPTER
dest[i] = val.ToString()
case 200: // ADVARCHAR
dest[i] = val.ToString()
case 201: // ADLONGVARCHAR
dest[i] = val.ToString()
case 202: // ADVARWCHAR
dest[i] = val.ToString()
case 203: // ADLONGVARWCHAR
dest[i] = val.ToString()
case 204: // ADVARBINARY
// TODO
case 205: // ADLONGVARBINARY
sa := (*ole.SAFEARRAY)(unsafe.Pointer(uintptr(val.Val)))
dest[i] = (*[1 << 30]byte)(unsafe.Pointer(uintptr(sa.Data)))[0:sa.Bounds.Elements]
}
}
_, err = oleutil.CallMethod(rc.rc, "MoveNext")
if err != nil {
return err
}
return nil
}
func getListViaWmiPnpEntity() ([]OsSerialPort, os.SyscallError) {
//log.Println("Doing getListViaWmiPnpEntity()")
// this method panics a lot and i'm not sure why, just catch
// the panic and return empty list
defer func() {
if e := recover(); e != nil {
// e is the interface{} typed-value we passed to panic()
log.Println("Got panic: ", e) // Prints "Whoops: boom!"
}
}()
var err os.SyscallError
//var friendlyName string
// init COM, oh yeah
ole.CoInitialize(0)
defer ole.CoUninitialize()
unknown, _ := oleutil.CreateObject("WbemScripting.SWbemLocator")
defer unknown.Release()
wmi, _ := unknown.QueryInterface(ole.IID_IDispatch)
defer wmi.Release()
// service is a SWbemServices
serviceRaw, _ := oleutil.CallMethod(wmi, "ConnectServer")
service := serviceRaw.ToIDispatch()
defer service.Release()
// result is a SWBemObjectSet
//pname := syscall.StringToUTF16("SELECT * FROM Win32_PnPEntity where Name like '%" + "COM35" + "%'")
pname := "SELECT * FROM Win32_PnPEntity WHERE ConfigManagerErrorCode = 0 and Name like '%(COM%'"
//pname := "SELECT * FROM Win32_PnPEntity WHERE ConfigManagerErrorCode = 0"
resultRaw, err2 := oleutil.CallMethod(service, "ExecQuery", pname)
//log.Println("Got result from oleutil.CallMethod")
if err2 != nil {
// we got back an error or empty list
log.Printf("Got an error back from oleutil.CallMethod. err:%v", err2)
return nil, err
}
result := resultRaw.ToIDispatch()
defer result.Release()
countVar, _ := oleutil.GetProperty(result, "Count")
count := int(countVar.Val)
list := make([]OsSerialPort, count)
for i := 0; i < count; i++ {
// items we're looping thru look like below and
// thus we can query for any of these names
/*
__GENUS : 2
__CLASS : Win32_PnPEntity
__SUPERCLASS : CIM_LogicalDevice
__DYNASTY : CIM_ManagedSystemElement
__RELPATH : Win32_PnPEntity.DeviceID="USB\\VID_1D50&PID_606D&MI_02\\6&2F09EA14&0&0002"
__PROPERTY_COUNT : 24
__DERIVATION : {CIM_LogicalDevice, CIM_LogicalElement, CIM_ManagedSystemElement}
__SERVER : JOHN-ATIV
__NAMESPACE : root\cimv2
__PATH : \\JOHN-ATIV\root\cimv2:Win32_PnPEntity.DeviceID="USB\\VID_1D50&PID_606D&MI_02\\6&2F09EA14
&0&0002"
Availability :
Caption : TinyG v2 (Data Channel) (COM12)
ClassGuid : {4d36e978-e325-11ce-bfc1-08002be10318}
CompatibleID : {USB\Class_02&SubClass_02&Prot_01, USB\Class_02&SubClass_02, USB\Class_02}
ConfigManagerErrorCode : 0
ConfigManagerUserConfig : False
CreationClassName : Win32_PnPEntity
Description : TinyG v2 (Data Channel)
DeviceID : USB\VID_1D50&PID_606D&MI_02\6&2F09EA14&0&0002
ErrorCleared :
ErrorDescription :
HardwareID : {USB\VID_1D50&PID_606D&REV_0097&MI_02, USB\VID_1D50&PID_606D&MI_02}
InstallDate :
LastErrorCode :
Manufacturer : Synthetos (www.synthetos.com)
Name : TinyG v2 (Data Channel) (COM12)
PNPDeviceID : USB\VID_1D50&PID_606D&MI_02\6&2F09EA14&0&0002
PowerManagementCapabilities :
PowerManagementSupported :
Service : usbser
Status : OK
StatusInfo :
SystemCreationClassName : Win32_ComputerSystem
SystemName : JOHN-ATIV
PSComputerName : JOHN-ATIV
*/
// item is a SWbemObject, but really a Win32_Process
itemRaw, _ := oleutil.CallMethod(result, "ItemIndex", i)
item := itemRaw.ToIDispatch()
defer item.Release()
asString, _ := oleutil.GetProperty(item, "Name")
//log.Println(asString.ToString())
// get the com port
//if false {
s := strings.Split(asString.ToString(), "(COM")[1]
s = "COM" + s
s = strings.Split(s, ")")[0]
list[i].Name = s
//}
// get the deviceid so we can figure out related ports
// it will look similar to
// USB\VID_1D50&PID_606D&MI_00\6&2F09EA14&0&0000
deviceIdStr, _ := oleutil.GetProperty(item, "DeviceID")
devIdItems := strings.Split(deviceIdStr.ToString(), "&")
log.Printf("DeviceId elements:%v", devIdItems)
if len(devIdItems) > 3 {
list[i].SerialNumber = devIdItems[3]
list[i].IdProduct = strings.Replace(devIdItems[1], "PID_", "", 1)
list[i].IdVendor = strings.Replace(devIdItems[0], "USB\\VID_", "", 1)
} else {
list[i].SerialNumber = deviceIdStr.ToString()
pidMatch := regexp.MustCompile("PID_(\\d+)").FindAllStringSubmatch(deviceIdStr.ToString(), -1)
if len(pidMatch) > 0 {
if len(pidMatch[0]) > 1 {
list[i].IdProduct = pidMatch[0][1]
}
}
vidMatch := regexp.MustCompile("VID_(\\d+)").FindAllStringSubmatch(deviceIdStr.ToString(), -1)
if len(vidMatch) > 0 {
if len(vidMatch[0]) > 1 {
list[i].IdVendor = vidMatch[0][1]
}
}
}
list[i].IdVendor = "0x" + list[i].IdVendor
list[i].IdProduct = "0x" + list[i].IdProduct
manufStr, _ := oleutil.GetProperty(item, "Manufacturer")
list[i].Manufacturer = manufStr.ToString()
descStr, _ := oleutil.GetProperty(item, "Description")
list[i].Product = descStr.ToString()
//classStr, _ := oleutil.GetProperty(item, "CreationClassName")
//list[i].DeviceClass = classStr.ToString()
}
return list, err
}
func getListViaWmiPnpEntity() ([]OsSerialPort, os.SyscallError) {
//log.Println("Doing getListViaWmiPnpEntity()")
// this method panics a lot and i'm not sure why, just catch
// the panic and return empty list
defer func() {
if e := recover(); e != nil {
// e is the interface{} typed-value we passed to panic()
log.Println("Got panic: ", e) // Prints "Whoops: boom!"
}
}()
var err os.SyscallError
//var friendlyName string
// init COM, oh yeah
ole.CoInitialize(0)
defer ole.CoUninitialize()
unknown, _ := oleutil.CreateObject("WbemScripting.SWbemLocator")
defer unknown.Release()
wmi, _ := unknown.QueryInterface(ole.IID_IDispatch)
defer wmi.Release()
// service is a SWbemServices
serviceRaw, _ := oleutil.CallMethod(wmi, "ConnectServer")
service := serviceRaw.ToIDispatch()
defer service.Release()
// result is a SWBemObjectSet
//pname := syscall.StringToUTF16("SELECT * FROM Win32_PnPEntity where Name like '%" + "COM35" + "%'")
pname := "SELECT * FROM Win32_PnPEntity WHERE ConfigManagerErrorCode = 0 and Name like '%(COM%'"
//pname := "SELECT * FROM Win32_PnPEntity WHERE ConfigManagerErrorCode = 0"
resultRaw, err2 := oleutil.CallMethod(service, "ExecQuery", pname)
//log.Println("Got result from oleutil.CallMethod")
if err2 != nil {
// we got back an error or empty list
log.Printf("Got an error back from oleutil.CallMethod. err:%v", err2)
return nil, err
}
result := resultRaw.ToIDispatch()
defer result.Release()
countVar, _ := oleutil.GetProperty(result, "Count")
count := int(countVar.Val)
list := make([]OsSerialPort, count)
for i := 0; i < count; i++ {
// item is a SWbemObject, but really a Win32_Process
itemRaw, _ := oleutil.CallMethod(result, "ItemIndex", i)
item := itemRaw.ToIDispatch()
defer item.Release()
asString, _ := oleutil.GetProperty(item, "Name")
//log.Println(asString.ToString())
// get the com port
//if false {
s := strings.Split(asString.ToString(), "(COM")[1]
s = "COM" + s
s = strings.Split(s, ")")[0]
list[i].Name = s
list[i].FriendlyName = asString.ToString()
//}
}
/*
for index, element := range list {
log.Println("index ", index, " element ", element.Name+
" friendly ", element.FriendlyName)
}
*/
return list, err
}
