Migrate to go-ole fork with fixed memory leaks

We can't use a replace trick in a library, so we have to replace original
go-ole with the forked one, until the fixes will be merged:
- go-ole/go-ole#199
- go-ole/go-ole#201

Author: yalegko

connection.go

@@ -8,8 +8,8 @@ import (
 	"reflect"
 	"sync"
 
-	"github.com/go-ole/go-ole"
-	"github.com/go-ole/go-ole/oleutil"
+	"github.com/bi-zone/go-ole"
+	"github.com/bi-zone/go-ole/oleutil"
 	"github.com/hashicorp/go-multierror"
 	"github.com/scjalliance/comshim"
 )

decoder.go

@@ -10,8 +10,8 @@ import (
 	"strings"
 	"time"
 
-	"github.com/go-ole/go-ole"
-	"github.com/go-ole/go-ole/oleutil"
+	"github.com/bi-zone/go-ole"
+	"github.com/bi-zone/go-ole/oleutil"
 	"github.com/hashicorp/go-multierror"
 )
 

decoder_test.go

@@ -9,7 +9,7 @@ import (
 	"testing"
 	"time"
 
-	"github.com/go-ole/go-ole"
+	"github.com/bi-zone/go-ole"
 )
 
 var (

go.mod

@@ -1,9 +1,9 @@
 module github.com/bi-zone/wmi
 
 require (
-	github.com/go-ole/go-ole v1.2.2
+	github.com/bi-zone/go-ole v1.2.5
+	github.com/go-ole/go-ole v1.2.4 // indirect
 	github.com/hashicorp/go-multierror v1.0.0
 	github.com/scjalliance/comshim v0.0.0-20190308082608-cf06d2532c4e
+	golang.org/x/sys v0.0.0-20200806060901-a37d78b92225
 )
-
-replace github.com/go-ole/go-ole => github.com/bi-zone/go-ole v1.2.5-0.20200615091707-5a6c01d333ed

go.sum

@@ -1,5 +1,9 @@
-github.com/bi-zone/go-ole v1.2.5-0.20200615091707-5a6c01d333ed h1:FR6GKPbTPKkA0FrTw0e26gU3AM5Ov75FgEUIwaggGxo=
-github.com/bi-zone/go-ole v1.2.5-0.20200615091707-5a6c01d333ed/go.mod h1:pprOEPIfldk/42T2oK7lQ4v4JSDwmV0As9GaiUsvbm0=
+github.com/bi-zone/go-ole v1.2.5 h1:/4G2KrTbq1e3FsMkd40quzwIrLb4QdxZJnUUlG7UjcM=
+github.com/bi-zone/go-ole v1.2.5/go.mod h1:BxzT498d9QAq10L6G/pTMscpDzqnpKN6DUBbmFKwyQY=
+github.com/go-ole/go-ole v1.2.4 h1:nNBDSCOigTSiarFpYE9J/KtEA1IOW4CNeqT9TQDqCxI=
+github.com/go-ole/go-ole v1.2.4/go.mod h1:XCwSNxSkXRo4vlyPy93sltvi/qJq0jqQhjqQNIwKuxM=
+github.com/gonuts/commander v0.1.0/go.mod h1:qkb5mSlcWodYgo7vs8ulLnXhfinhZsZcm6+H/z1JjgY=
+github.com/gonuts/flag v0.1.0/go.mod h1:ZTmTGtrSPejTo/SRNhCqwLTmiAgyBdCkLYhHrAoBdz4=
 github.com/hashicorp/errwrap v1.0.0 h1:hLrqtEDnRye3+sgx6z4qVLNuviH3MR5aQ0ykNJa/UYA=
 github.com/hashicorp/errwrap v1.0.0/go.mod h1:YH+1FKiLXxHSkmPseP+kNlulaMuP3n2brvKWEqk/Jc4=
 github.com/hashicorp/go-multierror v1.0.0 h1:iVjPR7a6H0tWELX5NxNe7bYopibicUzc7uPribsnS6o=
@@ -8,3 +12,5 @@ github.com/scjalliance/comshim v0.0.0-20190308082608-cf06d2532c4e h1:+/AzLkOdIXE
 github.com/scjalliance/comshim v0.0.0-20190308082608-cf06d2532c4e/go.mod h1:9Tc1SKnfACJb9N7cw2eyuI6xzy845G7uZONBsi5uPEA=
 golang.org/x/sys v0.0.0-20190916202348-b4ddaad3f8a3 h1:7TYNF4UdlohbFwpNH04CoPMp1cHUZgO1Ebq5r2hIjfo=
 golang.org/x/sys v0.0.0-20190916202348-b4ddaad3f8a3/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20200806060901-a37d78b92225 h1:a5kp7Ohh+lqGCGHUBQdPwGHTJXKNhVVWp34F+ncDC9M=
+golang.org/x/sys v0.0.0-20200806060901-a37d78b92225/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=

memory_test.go

@@ -10,8 +10,8 @@ import (
 	"testing"
 	"time"
 
-	"github.com/go-ole/go-ole"
-	"github.com/go-ole/go-ole/oleutil"
+	"github.com/bi-zone/go-ole"
+	"github.com/bi-zone/go-ole/oleutil"
 	"github.com/scjalliance/comshim"
 )
 
@@ -151,6 +151,10 @@ func TestMemory_WMIConcurrent(t *testing.T) {
 //
 // Run using: `TEST_MEM=1 go test -run TestMemory_OLEErrors -timeout 60m`
 func TestMemory_OLEErrors(t *testing.T) {
+	if os.Getenv("TEST_MEM") == "" {
+		t.Skip("Skipping TestMemory_OLEErrors; $TEST_MEM is not set")
+	}
+
 	// Subscribe to some rare event. E.g. removal of the local drive.
 	const query = "SELECT * FROM Win32_VolumeChangeEvent WHERE EventType=3"
 

notification_query.go

@@ -9,8 +9,8 @@ import (
 	"sync"
 	"time"
 
-	"github.com/go-ole/go-ole"
-	"github.com/go-ole/go-ole/oleutil"
+	"github.com/bi-zone/go-ole"
+	"github.com/bi-zone/go-ole/oleutil"
 	"github.com/hashicorp/go-multierror"
 	"github.com/scjalliance/comshim"
 )

swbemservices.go

@@ -6,8 +6,8 @@ import (
 	"fmt"
 	"sync"
 
-	"github.com/go-ole/go-ole"
-	"github.com/go-ole/go-ole/oleutil"
+	"github.com/bi-zone/go-ole"
+	"github.com/bi-zone/go-ole/oleutil"
 	"github.com/hashicorp/go-multierror"
 	"github.com/scjalliance/comshim"
 )