mirror of
https://github.com/safing/portmaster
synced 2025-09-02 02:29:12 +00:00
99 lines
2.9 KiB
Go
99 lines
2.9 KiB
Go
package profile
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/hashicorp/go-version"
|
|
|
|
"github.com/safing/portbase/config"
|
|
"github.com/safing/portbase/database"
|
|
"github.com/safing/portbase/database/migration"
|
|
"github.com/safing/portbase/database/query"
|
|
"github.com/safing/portbase/log"
|
|
"github.com/safing/portmaster/status"
|
|
)
|
|
|
|
func registerMigrations() error {
|
|
return migrations.Add(
|
|
migration.Migration{
|
|
Description: "Migrate to configurable network rating system",
|
|
Version: "v0.7.19",
|
|
MigrateFunc: migrateNetworkRatingSystem,
|
|
},
|
|
migration.Migration{
|
|
Description: "Migrate from LinkedPath to Fingerprints and PresentationPath",
|
|
Version: "v0.9.9",
|
|
MigrateFunc: migrateLinkedPath,
|
|
},
|
|
)
|
|
}
|
|
|
|
func migrateNetworkRatingSystem(ctx context.Context, _, to *version.Version, db *database.Interface) error {
|
|
// determine the default value for the network rating system by searching for
|
|
// a global security level setting that is not set to the default.
|
|
networkRatingEnabled := false
|
|
for _, cfgkey := range securityLevelSettings {
|
|
def, err := config.GetOption(cfgkey)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
intValue := config.Concurrent.GetAsInt(cfgkey, 0)()
|
|
defaultValue, ok := def.DefaultValue.(uint8)
|
|
if ok && defaultValue != uint8(intValue) {
|
|
log.Tracer(ctx).Infof("found global security level setting with changed value. 0x%2x (default) != 0x%2x (current)", def.DefaultValue, intValue)
|
|
networkRatingEnabled = true
|
|
break
|
|
}
|
|
}
|
|
|
|
if networkRatingEnabled {
|
|
err := status.SetNetworkRating(networkRatingEnabled)
|
|
if err != nil {
|
|
log.Warningf("profile: migration to %s failed to set network rating level to %v", to, networkRatingEnabled)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func migrateLinkedPath(ctx context.Context, _, to *version.Version, db *database.Interface) error {
|
|
// Get iterator over all profiles.
|
|
it, err := db.Query(query.New(profilesDBPath))
|
|
if err != nil {
|
|
log.Tracer(ctx).Errorf("profile: failed to migrate from linked path: failed to start query: %s", err)
|
|
return nil
|
|
}
|
|
|
|
// Migrate all profiles.
|
|
for r := range it.Next {
|
|
// Parse profile.
|
|
profile, err := EnsureProfile(r)
|
|
if err != nil {
|
|
log.Tracer(ctx).Debugf("profiles: failed to parse profile %s for migration: %s", r.Key(), err)
|
|
continue
|
|
}
|
|
|
|
// Skip if there is no LinkedPath to migrate from.
|
|
if profile.LinkedPath == "" {
|
|
continue
|
|
}
|
|
|
|
// Update metadata and save if changed.
|
|
if profile.updateMetadata("") {
|
|
err = db.Put(profile)
|
|
if err != nil {
|
|
log.Tracer(ctx).Debugf("profiles: failed to save profile %s after migration: %s", r.Key(), err)
|
|
} else {
|
|
log.Tracer(ctx).Tracef("profiles: migrated profile %s to %s", r.Key(), to)
|
|
}
|
|
}
|
|
}
|
|
|
|
// Check if there was an error while iterating.
|
|
if err := it.Err(); err != nil {
|
|
log.Tracer(ctx).Errorf("profile: failed to migrate from linked path: failed to iterate over profiles for migration: %s", err)
|
|
}
|
|
|
|
return nil
|
|
}
|