Skip to content

Commit

Permalink
Rearange code
Browse files Browse the repository at this point in the history
  • Loading branch information
DRON-666 committed Apr 28, 2024
1 parent 7ee889b commit 125dba2
Show file tree
Hide file tree
Showing 2 changed files with 16 additions and 20 deletions.
6 changes: 1 addition & 5 deletions internal/fs/fs_local_vss_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -274,11 +274,7 @@ func TestParseProvider(t *testing.T) {
if err != nil {
result = err.Error()
}
matched, err := regexp.MatchString(test.result, result)
if err != nil {
panic(err)
}
if !matched || test.result == "" {
if test.result != result || test.result == "" {
t.Fatalf("wrong result, want:\n %#v\ngot:\n %#v", test.result, result)
}
} else if !ole.IsEqualGUID(id, test.id) {
Expand Down
30 changes: 15 additions & 15 deletions internal/fs/vss_windows.go
Original file line number Diff line number Diff line change
Expand Up @@ -537,13 +537,6 @@ func vssFreeSnapshotProperties(properties *VssSnapshotProperties) error {
return nil
}

func vssFreeProviderProperties(p *VssProviderProperties) {
ole.CoTaskMemFree(uintptr(unsafe.Pointer(p.providerName)))
p.providerName = nil
ole.CoTaskMemFree(uintptr(unsafe.Pointer(p.providerVersion)))
p.providerName = nil
}

// BackupComplete calls the equivalent VSS api.
func (vss *IVssBackupComponents) BackupComplete() (*IVSSAsync, error) {
var oleIUnknown *ole.IUnknown
Expand Down Expand Up @@ -583,6 +576,13 @@ type VssProviderProperties struct {
classID ole.GUID
}

func vssFreeProviderProperties(p *VssProviderProperties) {
ole.CoTaskMemFree(uintptr(unsafe.Pointer(p.providerName)))
p.providerName = nil
ole.CoTaskMemFree(uintptr(unsafe.Pointer(p.providerVersion)))
p.providerVersion = nil
}

// GetSnapshotDeviceObject returns root path to access the snapshot files
// and folders.
func (p *VssSnapshotProperties) GetSnapshotDeviceObject() string {
Expand Down Expand Up @@ -1084,14 +1084,6 @@ func (p *VssSnapshot) Delete() error {
}

func getProviderID(provider string) (*ole.GUID, error) {
comInterface, err := ole.CreateInstance(CLSID_VSS_COORDINATOR, UIID_IVSS_ADMIN)
if err != nil {
return nil, err
}
defer comInterface.Release()

vssAdmin := (*IVSSAdmin)(unsafe.Pointer(comInterface))

providerLower := strings.ToLower(provider)
switch providerLower {
case "":
Expand All @@ -1100,6 +1092,14 @@ func getProviderID(provider string) (*ole.GUID, error) {
return ole.NewGUID("{b5946137-7b9f-4925-af80-51abd60b20d5}"), nil
}

comInterface, err := ole.CreateInstance(CLSID_VSS_COORDINATOR, UIID_IVSS_ADMIN)
if err != nil {
return nil, err
}
defer comInterface.Release()

vssAdmin := (*IVSSAdmin)(unsafe.Pointer(comInterface))

enum, err := vssAdmin.QueryProviders()
if err != nil {
return nil, err
Expand Down

0 comments on commit 125dba2

Please sign in to comment.