Skip to content

Commit

Permalink
feat(akamai): unzip user content
Browse files Browse the repository at this point in the history
  • Loading branch information
guilhem committed Sep 12, 2024
1 parent 3b3a27b commit 19fab24
Show file tree
Hide file tree
Showing 3 changed files with 52 additions and 1 deletion.
1 change: 1 addition & 0 deletions docs/release-notes.md
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,7 @@ nav_order: 9

- Support partitioning disk with mounted partitions
- Support Proxmox VE
- Support gzipped Akamai user_data

### Changes

Expand Down
9 changes: 8 additions & 1 deletion internal/providers/akamai/akamai.go
Original file line number Diff line number Diff line change
Expand Up @@ -91,7 +91,14 @@ func fetchConfig(f *resource.Fetcher) (types.Config, report.Report, error) {
return types.Config{}, report.Report{}, fmt.Errorf("decode base64: %w", err)
}

return util.ParseConfig(f.Logger, data[:n])
// The Linode Metadata Service can compress userdata.
// We have to gunzip if needed.
unzipData, err := util.GunzipIfNeeded(data[:n])
if err != nil {
return types.Config{}, report.Report{}, fmt.Errorf("unzip: %w", err)
}

return util.ParseConfig(f.Logger, unzipData)
}

// defaultTokenTTL is the time-to-live (TTL; in seconds) for an authorization
Expand Down
43 changes: 43 additions & 0 deletions internal/providers/util/unzip.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,43 @@
// Copyright 2024 CoreOS, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package util

import (
"bytes"
"compress/gzip"
"io"
)

var (
gzipMagic = []byte{0x1F, 0x8B, 0x08}
)

func hasGzipMagicNumber(source []byte) bool {
return bytes.HasPrefix(source, gzipMagic)
}

func GunzipIfNeeded(raw []byte) ([]byte, error) {
if !hasGzipMagicNumber(raw) {
return raw, nil
}

reader, err := gzip.NewReader(bytes.NewReader(raw))
if err != nil {
return nil, err
}
defer reader.Close()

return io.ReadAll(reader)
}

0 comments on commit 19fab24

Please sign in to comment.