Compare commits
No commits in common. "b53cbc3ea56404953bf6813adade8c02ad4d1610" and "a88995290caf88081beb3a8db204177dbd2bc7a6" have entirely different histories.
b53cbc3ea5
...
a88995290c
|
|
@ -8,27 +8,9 @@ const std = @import("std");
|
||||||
const builtin = @import("builtin");
|
const builtin = @import("builtin");
|
||||||
const auth = @import("aws_authentication.zig");
|
const auth = @import("aws_authentication.zig");
|
||||||
|
|
||||||
const log = std.log.scoped(.aws_credentials);
|
pub fn getCredentials(allocator: std.mem.Allocator) !auth.Credentials {
|
||||||
|
|
||||||
pub const Profile = struct {
|
|
||||||
/// Credential file. Defaults to AWS_SHARED_CREDENTIALS_FILE or ~/.aws/credentials
|
|
||||||
credential_file: ?[]const u8 = null,
|
|
||||||
/// Config file. Defaults to AWS_CONFIG_FILE or ~/.aws/config
|
|
||||||
config_file: ?[]const u8 = null,
|
|
||||||
/// Config file. Defaults to AWS_PROFILE or default
|
|
||||||
profile_name: ?[]const u8 = null,
|
|
||||||
};
|
|
||||||
|
|
||||||
pub const Options = struct {
|
|
||||||
profile: Profile = .{},
|
|
||||||
};
|
|
||||||
|
|
||||||
pub fn getCredentials(allocator: std.mem.Allocator, options: Options) !auth.Credentials {
|
|
||||||
if (try getEnvironmentCredentials(allocator)) |cred| return cred;
|
if (try getEnvironmentCredentials(allocator)) |cred| return cred;
|
||||||
// TODO: 2-5
|
// TODO: 2-5
|
||||||
// Note that boto and Java disagree on where this fits in the order
|
|
||||||
if (try getWebIdentityToken(allocator)) |cred| return cred;
|
|
||||||
if (try getProfileCredentials(allocator, options.profile)) |cred| return cred;
|
|
||||||
return error.NotImplemented;
|
return error.NotImplemented;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
@ -42,7 +24,7 @@ fn getEnvironmentCredentials(allocator: std.mem.Allocator) !?auth.Credentials {
|
||||||
(try getEnvironmentVariable(allocator, "AWS_ACCESS_KEY_ID")) orelse return null,
|
(try getEnvironmentVariable(allocator, "AWS_ACCESS_KEY_ID")) orelse return null,
|
||||||
mutable_key,
|
mutable_key,
|
||||||
(try getEnvironmentVariable(allocator, "AWS_SESSION_TOKEN")) orelse
|
(try getEnvironmentVariable(allocator, "AWS_SESSION_TOKEN")) orelse
|
||||||
try getEnvironmentVariable(allocator, "AWS_SECURITY_TOKEN"), // Security token is backward compat only
|
try getEnvironmentVariable(allocator, "AWS_SECURITY_TOKEN"),
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
@ -52,346 +34,3 @@ fn getEnvironmentVariable(allocator: std.mem.Allocator, key: []const u8) !?[]con
|
||||||
else => return e,
|
else => return e,
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
fn getWebIdentityToken(allocator: std.mem.Allocator) !?auth.Credentials {
|
|
||||||
_ = allocator;
|
|
||||||
// This API does not require signing. We can just use zfetch to
|
|
||||||
// shoot a raw request over.
|
|
||||||
// https://docs.aws.amazon.com/STS/latest/APIReference/API_AssumeRoleWithWebIdentity.html
|
|
||||||
// https://github.com/boto/boto3/blob/85b975af30c408f93b654a21930218edd58336ad/docs/source/guide/credentials.rst#assume-role-with-web-identity-provider
|
|
||||||
// https://github.com/aws/aws-sdk-java-v2/blob/master/core/auth/src/main/java/software/amazon/awssdk/auth/credentials/WebIdentityTokenFileCredentialsProvider.java
|
|
||||||
// TODO: implement
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
|
|
||||||
fn getProfileCredentials(allocator: std.mem.Allocator, options: Profile) !?auth.Credentials {
|
|
||||||
var default_path: ?[]const u8 = null;
|
|
||||||
defer if (default_path) |p| allocator.free(p);
|
|
||||||
|
|
||||||
const creds_file_path = try filePath(
|
|
||||||
allocator,
|
|
||||||
options.credential_file,
|
|
||||||
"AWS_SHARED_CREDENTIALS_FILE",
|
|
||||||
default_path,
|
|
||||||
"credentials",
|
|
||||||
);
|
|
||||||
defer allocator.free(creds_file_path.evaluated_path);
|
|
||||||
default_path = default_path orelse creds_file_path.home;
|
|
||||||
const config_file_path = try filePath(
|
|
||||||
allocator,
|
|
||||||
options.credential_file,
|
|
||||||
"AWS_SHARED_CREDENTIALS_FILE",
|
|
||||||
default_path,
|
|
||||||
"config",
|
|
||||||
);
|
|
||||||
defer allocator.free(config_file_path.evaluated_path);
|
|
||||||
default_path = default_path orelse config_file_path.home;
|
|
||||||
|
|
||||||
// Get active profile
|
|
||||||
const profile = (try getEnvironmentVariable(allocator, "AWS_PROFILE")) orelse
|
|
||||||
try allocator.dupe(u8, "default");
|
|
||||||
defer allocator.free(profile);
|
|
||||||
log.debug("Looking for file credentials using profile '{s}'", .{profile});
|
|
||||||
log.debug("Checking credentials file: {s}", .{creds_file_path.evaluated_path});
|
|
||||||
const credentials_file = std.fs.openFileAbsolute(creds_file_path.evaluated_path, .{}) catch null;
|
|
||||||
defer if (credentials_file) |f| f.close();
|
|
||||||
// It's much more likely that we'll find credentials in the credentials file
|
|
||||||
// so we'll try that first
|
|
||||||
const creds_file_creds = try credsForFile(allocator, credentials_file, profile);
|
|
||||||
var conf_file_creds = PartialCredentials{};
|
|
||||||
if (creds_file_creds.access_key == null or creds_file_creds.secret_key == null) {
|
|
||||||
log.debug("Checking config file: {s}", .{config_file_path.evaluated_path});
|
|
||||||
const config_file = std.fs.openFileAbsolute(creds_file_path.evaluated_path, .{}) catch null;
|
|
||||||
defer if (config_file) |f| f.close();
|
|
||||||
conf_file_creds = try credsForFile(allocator, config_file, profile);
|
|
||||||
}
|
|
||||||
const access_key = keyFrom(allocator, creds_file_creds.access_key, conf_file_creds.access_key);
|
|
||||||
const secret_key = keyFrom(allocator, creds_file_creds.secret_key, conf_file_creds.secret_key);
|
|
||||||
defer if (secret_key) |k| allocator.free(k);
|
|
||||||
|
|
||||||
if (access_key == null or secret_key == null) {
|
|
||||||
const partial = access_key != null or secret_key != null;
|
|
||||||
if (partial) {
|
|
||||||
log.warn("Could not find credentials in file (partial creds detected)", .{});
|
|
||||||
} else {
|
|
||||||
log.info("Could not find credentials in file", .{});
|
|
||||||
}
|
|
||||||
if (access_key) |k| allocator.free(k);
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
log.debug("Got full credentials from filesystem. Access key: {s}", .{access_key.?});
|
|
||||||
return auth.Credentials.init(
|
|
||||||
allocator,
|
|
||||||
access_key.?,
|
|
||||||
try allocator.dupe(u8, secret_key.?),
|
|
||||||
null,
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
fn keyFrom(allocator: std.mem.Allocator, priority_1: ?[]const u8, priority_2: ?[]const u8) ?[]const u8 {
|
|
||||||
if (priority_1) |p1| {
|
|
||||||
if (priority_2) |p2| allocator.free(p2);
|
|
||||||
return p1;
|
|
||||||
}
|
|
||||||
return priority_2;
|
|
||||||
}
|
|
||||||
|
|
||||||
// We could conceivably find different portions of the creds in different
|
|
||||||
// files, so let's be super-loose here
|
|
||||||
const PartialCredentials = struct {
|
|
||||||
access_key: ?[]const u8 = null,
|
|
||||||
secret_key: ?[]const u8 = null,
|
|
||||||
};
|
|
||||||
fn credsForFile(allocator: std.mem.Allocator, file: ?std.fs.File, profile: []const u8) !PartialCredentials {
|
|
||||||
if (file == null) return PartialCredentials{};
|
|
||||||
const text = try file.?.readToEndAlloc(allocator, std.math.maxInt(usize));
|
|
||||||
defer allocator.free(text);
|
|
||||||
const partial_creds = try credsForText(text, profile);
|
|
||||||
var ak: ?[]const u8 = null;
|
|
||||||
if (partial_creds.access_key) |k|
|
|
||||||
ak = try allocator.dupe(u8, k);
|
|
||||||
var sk: ?[]const u8 = null;
|
|
||||||
if (partial_creds.secret_key) |k|
|
|
||||||
sk = try allocator.dupe(u8, k);
|
|
||||||
|
|
||||||
return PartialCredentials{
|
|
||||||
.access_key = ak,
|
|
||||||
.secret_key = sk,
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
const LineIterator = struct {
|
|
||||||
text: []const u8,
|
|
||||||
inx: usize = 0,
|
|
||||||
|
|
||||||
const Self = @This();
|
|
||||||
|
|
||||||
pub fn init(text: []const u8) Self {
|
|
||||||
return .{
|
|
||||||
.text = text,
|
|
||||||
.inx = 0,
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn next(self: *Self) ?[]const u8 {
|
|
||||||
if (self.inx >= self.text.len) return null;
|
|
||||||
var current = self.inx;
|
|
||||||
var start = self.inx;
|
|
||||||
for (self.text[self.inx..]) |c, i| {
|
|
||||||
if (c == '\n') {
|
|
||||||
// log.debug("got \\n: {d}", .{i});
|
|
||||||
current += i + 1;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
// log.debug("{d}:{d}", .{ current, self.inx });
|
|
||||||
if (current != self.inx) {
|
|
||||||
self.inx = current;
|
|
||||||
} else { // no \n found
|
|
||||||
self.inx = self.text.len + 1; // add one to capture the last char in return
|
|
||||||
}
|
|
||||||
return self.text[start .. self.inx - 1];
|
|
||||||
}
|
|
||||||
};
|
|
||||||
fn credsForText(text: []const u8, profile: []const u8) !PartialCredentials {
|
|
||||||
var lines = LineIterator.init(text);
|
|
||||||
var is_in_profile = false;
|
|
||||||
var was_in_profile = false;
|
|
||||||
var done = false;
|
|
||||||
var creds: [2]?[]const u8 = [_]?[]const u8{ null, null };
|
|
||||||
|
|
||||||
while (lines.next()) |line| {
|
|
||||||
// log.debug("line: {s}", .{line});
|
|
||||||
var section_start: ?usize = 0;
|
|
||||||
for (line) |c, i| {
|
|
||||||
switch (c) {
|
|
||||||
'#' => break,
|
|
||||||
'[' => section_start = i + 1,
|
|
||||||
']' => {
|
|
||||||
if (section_start) |s| {
|
|
||||||
const current_section = line[s..i];
|
|
||||||
log.debug("got section: {s}", .{current_section});
|
|
||||||
is_in_profile = std.ascii.eqlIgnoreCase(current_section, profile);
|
|
||||||
if (was_in_profile and !is_in_profile) {
|
|
||||||
done = true;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
|
|
||||||
was_in_profile = is_in_profile;
|
|
||||||
break; // got what we need from this line
|
|
||||||
}
|
|
||||||
},
|
|
||||||
'=' => {
|
|
||||||
if (!is_in_profile) continue;
|
|
||||||
const key = std.mem.trim(u8, line[0..i], " \t"); // other whitespace we care about?
|
|
||||||
log.debug("got key: {s}", .{key});
|
|
||||||
for (&[_][]const u8{
|
|
||||||
"aws_access_key_id",
|
|
||||||
"aws_secret_access_key",
|
|
||||||
}) |needle, inx| {
|
|
||||||
if (std.ascii.eqlIgnoreCase(key, needle)) {
|
|
||||||
// TODO: Trim this out
|
|
||||||
creds[inx] = trim(line[i + 1 ..]);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
},
|
|
||||||
else => {},
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if (done) {
|
|
||||||
log.debug("no longer in target section: bailing", .{});
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
log.debug("done parsing text", .{});
|
|
||||||
return PartialCredentials{
|
|
||||||
.access_key = creds[0],
|
|
||||||
.secret_key = creds[1],
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
fn trim(text: []const u8) []const u8 {
|
|
||||||
// " myval # yo";
|
|
||||||
var start: ?usize = null;
|
|
||||||
var end: ?usize = null;
|
|
||||||
|
|
||||||
for (text) |c, i| switch (c) {
|
|
||||||
' ', '\t' => {},
|
|
||||||
'#' => return trimmed(text, start, end),
|
|
||||||
else => {
|
|
||||||
if (start == null) start = i;
|
|
||||||
end = i + 1;
|
|
||||||
},
|
|
||||||
};
|
|
||||||
return trimmed(text, start, end);
|
|
||||||
}
|
|
||||||
|
|
||||||
fn trimmed(text: []const u8, start: ?usize, end: ?usize) []const u8 {
|
|
||||||
if (start == null) return "";
|
|
||||||
if (end == null) return text[start.?..];
|
|
||||||
return text[start.?..end.?];
|
|
||||||
}
|
|
||||||
|
|
||||||
fn filePath(
|
|
||||||
allocator: std.mem.Allocator,
|
|
||||||
specified_path: ?[]const u8,
|
|
||||||
env_var_name: []const u8,
|
|
||||||
config_dir: ?[]const u8,
|
|
||||||
config_file_name: []const u8,
|
|
||||||
) !EvaluatedPath {
|
|
||||||
if (specified_path) |p| return EvaluatedPath{ .evaluated_path = try allocator.dupe(u8, p) };
|
|
||||||
// Not specified. Check environment variable, otherwise, hard coded default
|
|
||||||
if (try getEnvironmentVariable(allocator, env_var_name)) |v| return EvaluatedPath{ .evaluated_path = v };
|
|
||||||
|
|
||||||
// Not in environment variable either. Go fish
|
|
||||||
return try getDefaultPath(allocator, config_dir, ".aws", config_file_name);
|
|
||||||
}
|
|
||||||
|
|
||||||
const EvaluatedPath = struct {
|
|
||||||
home: ?[]const u8 = null,
|
|
||||||
evaluated_path: []const u8,
|
|
||||||
};
|
|
||||||
fn getDefaultPath(allocator: std.mem.Allocator, home_dir: ?[]const u8, dir: []const u8, file: []const u8) !EvaluatedPath {
|
|
||||||
var home = home_dir orelse try getHomeDir(allocator);
|
|
||||||
log.debug("Home directory: {s}", .{home});
|
|
||||||
const rc = try std.fs.path.join(allocator, &[_][]const u8{ home, dir, file });
|
|
||||||
log.debug("Path evaluated as: {s}", .{rc});
|
|
||||||
return EvaluatedPath{ .home = home, .evaluated_path = rc };
|
|
||||||
}
|
|
||||||
|
|
||||||
fn getHomeDir(allocator: std.mem.Allocator) ![]const u8 {
|
|
||||||
switch (builtin.os.tag) {
|
|
||||||
.windows => {
|
|
||||||
var dir_path_ptr: [*:0]u16 = undefined;
|
|
||||||
// https://docs.microsoft.com/en-us/windows/win32/shell/knownfolderid
|
|
||||||
const FOLDERID_Profile = std.os.windows.GUID.parse("{5E6C858F-0E22-4760-9AFE-EA3317B67173}");
|
|
||||||
switch (std.os.windows.shell32.SHGetKnownFolderPath(
|
|
||||||
&FOLDERID_Profile,
|
|
||||||
std.os.windows.KF_FLAG_CREATE,
|
|
||||||
null,
|
|
||||||
&dir_path_ptr,
|
|
||||||
)) {
|
|
||||||
std.os.windows.S_OK => {
|
|
||||||
defer std.os.windows.ole32.CoTaskMemFree(@ptrCast(*anyopaque, dir_path_ptr));
|
|
||||||
const global_dir = std.unicode.utf16leToUtf8Alloc(allocator, std.mem.sliceTo(dir_path_ptr, 0)) catch |err| switch (err) {
|
|
||||||
error.UnexpectedSecondSurrogateHalf => return error.HomeDirUnavailable,
|
|
||||||
error.ExpectedSecondSurrogateHalf => return error.HomeDirUnavailable,
|
|
||||||
error.DanglingSurrogateHalf => return error.HomeDirUnavailable,
|
|
||||||
error.OutOfMemory => return error.OutOfMemory,
|
|
||||||
};
|
|
||||||
return global_dir;
|
|
||||||
// defer allocator.free(global_dir);
|
|
||||||
},
|
|
||||||
std.os.windows.E_OUTOFMEMORY => return error.OutOfMemory,
|
|
||||||
else => return error.HomeDirUnavailable,
|
|
||||||
}
|
|
||||||
},
|
|
||||||
.macos, .linux, .freebsd, .netbsd, .dragonfly, .openbsd, .solaris => {
|
|
||||||
const home_dir = std.os.getenv("HOME") orelse {
|
|
||||||
// TODO look in /etc/passwd
|
|
||||||
return error.HomeDirUnavailable;
|
|
||||||
};
|
|
||||||
return allocator.dupe(u8, home_dir);
|
|
||||||
},
|
|
||||||
// Code from https://github.com/ziglang/zig/blob/9f9f215305389c08a21730859982b68bf2681932/lib/std/fs/get_app_data_dir.zig
|
|
||||||
// be_user_settings magic number is probably different for home directory
|
|
||||||
// .haiku => {
|
|
||||||
// var dir_path_ptr: [*:0]u8 = undefined;
|
|
||||||
// // TODO look into directory_which
|
|
||||||
// const be_user_settings = 0xbbe;
|
|
||||||
// const rc = os.system.find_directory(be_user_settings, -1, true, dir_path_ptr, 1);
|
|
||||||
// const settings_dir = try allocator.dupeZ(u8, mem.sliceTo(dir_path_ptr, 0));
|
|
||||||
// defer allocator.free(settings_dir);
|
|
||||||
// switch (rc) {
|
|
||||||
// 0 => return fs.path.join(allocator, &[_][]const u8{ settings_dir, appname }),
|
|
||||||
// else => return error.AppDataDirUnavailable,
|
|
||||||
// }
|
|
||||||
// },
|
|
||||||
else => @compileError("Unsupported OS"),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
test "filePath" {
|
|
||||||
const allocator = std.testing.allocator;
|
|
||||||
std.testing.log_level = .debug;
|
|
||||||
log.debug("\n", .{});
|
|
||||||
const path = try filePath(allocator, null, "NOTHING", null, "hello");
|
|
||||||
defer allocator.free(path.evaluated_path);
|
|
||||||
defer allocator.free(path.home.?);
|
|
||||||
try std.testing.expect(path.evaluated_path.len > 10);
|
|
||||||
try std.testing.expectEqualStrings("hello", path.evaluated_path[path.evaluated_path.len - 5 ..]);
|
|
||||||
try std.testing.expect(path.home != null);
|
|
||||||
}
|
|
||||||
|
|
||||||
test "ini to creds" {
|
|
||||||
std.testing.log_level = .debug;
|
|
||||||
log.debug("\n", .{});
|
|
||||||
const partial_creds = try credsForText(
|
|
||||||
\\
|
|
||||||
\\# Amazon Web Services Credentials File used by AWS CLI, SDKs, and tools
|
|
||||||
\\# This file was created by the AWS Toolkit for Visual Studio Code extension.
|
|
||||||
\\#
|
|
||||||
\\# Your AWS credentials are represented by access keys associated with IAM users.
|
|
||||||
\\# For information about how to create and manage AWS access keys for a user, see:
|
|
||||||
\\# https://docs.aws.amazon.com/IAM/latest/UserGuide/id_credentials_access-keys.html
|
|
||||||
\\#
|
|
||||||
\\# This credential file can store multiple access keys by placing each one in a
|
|
||||||
\\# named "profile". For information about how to change the access keys in a
|
|
||||||
\\# profile or to add a new profile with a different access key, see:
|
|
||||||
\\# https://docs.aws.amazon.com/cli/latest/userguide/cli-config-files.html
|
|
||||||
\\#
|
|
||||||
\\[other_section]
|
|
||||||
\\access_key_id = NOTYOURACCESSKEY
|
|
||||||
\\
|
|
||||||
\\ [default]
|
|
||||||
\\ # The access key and secret key pair identify your account and grant access to AWS.
|
|
||||||
\\aws_access_key_id = AKIDEXAMPLE # access key
|
|
||||||
\\
|
|
||||||
\\[another_section]
|
|
||||||
\\access_key_id = NOTYOURACCESSKEYEITHER
|
|
||||||
, "default");
|
|
||||||
|
|
||||||
try std.testing.expect(partial_creds.access_key != null);
|
|
||||||
try std.testing.expectEqualStrings("AKIDEXAMPLE", partial_creds.access_key.?);
|
|
||||||
try std.testing.expect(partial_creds.secret_key == null);
|
|
||||||
}
|
|
||||||
|
|
|
||||||
|
|
@ -97,8 +97,7 @@ pub const AwsHttp = struct {
|
||||||
const endpoint = try regionSubDomain(self.allocator, service, options.region, options.dualstack);
|
const endpoint = try regionSubDomain(self.allocator, service, options.region, options.dualstack);
|
||||||
defer endpoint.deinit();
|
defer endpoint.deinit();
|
||||||
log.debug("Calling endpoint {s}", .{endpoint.uri});
|
log.debug("Calling endpoint {s}", .{endpoint.uri});
|
||||||
// TODO: Should we allow customization here?
|
const creds = try credentials.getCredentials(self.allocator);
|
||||||
const creds = try credentials.getCredentials(self.allocator, .{});
|
|
||||||
defer creds.deinit();
|
defer creds.deinit();
|
||||||
const signing_config: signing.Config = .{
|
const signing_config: signing.Config = .{
|
||||||
.region = options.region,
|
.region = options.region,
|
||||||
|
|
|
||||||
|
|
@ -284,9 +284,7 @@ fn getSigningKey(allocator: std.mem.Allocator, signing_date: []const u8, config:
|
||||||
, .{ signing_date, config.region, config.service });
|
, .{ signing_date, config.region, config.service });
|
||||||
var secret = try std.fmt.allocPrint(allocator, "AWS4{s}", .{config.credentials.secret_key});
|
var secret = try std.fmt.allocPrint(allocator, "AWS4{s}", .{config.credentials.secret_key});
|
||||||
defer {
|
defer {
|
||||||
// secureZero avoids compiler optimizations that may say
|
for (secret) |_, i| secret[i] = 0; // zero our copy of secret
|
||||||
// "WTF are you doing this thing? Looks like nothing to me. It's silly and we will remove it"
|
|
||||||
std.crypto.utils.secureZero(u8, secret); // zero our copy of secret
|
|
||||||
allocator.free(secret);
|
allocator.free(secret);
|
||||||
}
|
}
|
||||||
// log.debug("secret: {s}", .{secret});
|
// log.debug("secret: {s}", .{secret});
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue
Block a user