fakecloud-ec2 0.41.1

Amazon EC2 implementation for FakeCloud
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124

//! EC2 byoip operations (extracted from the rest long-tail module).

#![allow(clippy::too_many_lines)]

use super::*;

fn byoip_cidr_xml(c: &ByoipCidr) -> String {
    format!(
        "{}{}{}",
        ec2_elem("cidr", &c.cidr),
        ec2_elem("description", &c.description),
        ec2_elem("state", &c.state),
    )
}

/// Update the persisted BYOIP CIDR's state (inserting it when unknown), then
/// render it. Shared by advertise/withdraw so each op round-trips.
fn set_byoip_state(svc: &Ec2Service, req: &AwsRequest, cidr: &str, state: &str) -> ByoipCidr {
    let mut accounts = svc.state.write();
    let account = accounts.get_or_create(&req.account_id);
    let entry = account
        .byoip_cidrs
        .entry(cidr.to_string())
        .or_insert_with(|| ByoipCidr {
            cidr: cidr.to_string(),
            description: String::new(),
            state: state.to_string(),
        });
    entry.state = state.to_string();
    entry.clone()
}

pub(crate) fn advertise_byoip_cidr(
    svc: &Ec2Service,
    req: &AwsRequest,
) -> Result<AwsResponse, AwsServiceError> {
    let cidr = require(&req.query_params, "Cidr")?;
    let c = set_byoip_state(svc, req, &cidr, "advertised");
    Ok(Ec2Service::respond(
        "AdvertiseByoipCidr",
        &req.request_id,
        &format!("<byoipCidr>{}</byoipCidr>", byoip_cidr_xml(&c)),
    ))
}

pub(crate) fn deprovision_byoip_cidr(
    svc: &Ec2Service,
    req: &AwsRequest,
) -> Result<AwsResponse, AwsServiceError> {
    let cidr = require(&req.query_params, "Cidr")?;
    // Deprovision removes the CIDR from the pool; report the transitional state.
    let mut c = set_byoip_state(svc, req, &cidr, "pending-deprovision");
    {
        let mut accounts = svc.state.write();
        accounts
            .get_or_create(&req.account_id)
            .byoip_cidrs
            .remove(&cidr);
    }
    c.state = "pending-deprovision".to_string();
    Ok(Ec2Service::respond(
        "DeprovisionByoipCidr",
        &req.request_id,
        &format!("<byoipCidr>{}</byoipCidr>", byoip_cidr_xml(&c)),
    ))
}

pub(crate) fn describe_byoip_cidrs(
    svc: &Ec2Service,
    req: &AwsRequest,
) -> Result<AwsResponse, AwsServiceError> {
    require(&req.query_params, "MaxResults")?;
    validate_max_results(&req.query_params, 1, 100)?;
    let accounts = svc.state.read();
    let empty = Ec2State::new(&req.account_id, &req.region);
    let state = accounts.get(&req.account_id).unwrap_or(&empty);
    let items: Vec<String> = state.byoip_cidrs.values().map(byoip_cidr_xml).collect();
    Ok(Ec2Service::respond(
        "DescribeByoipCidrs",
        &req.request_id,
        &ec2_list("byoipCidrSet", &items),
    ))
}

pub(crate) fn provision_byoip_cidr(
    svc: &Ec2Service,
    req: &AwsRequest,
) -> Result<AwsResponse, AwsServiceError> {
    let cidr = require(&req.query_params, "Cidr")?;
    let c = ByoipCidr {
        cidr: cidr.clone(),
        description: req
            .query_params
            .get("Description")
            .cloned()
            .unwrap_or_default(),
        state: "provisioned".to_string(),
    };
    {
        let mut accounts = svc.state.write();
        accounts
            .get_or_create(&req.account_id)
            .byoip_cidrs
            .insert(cidr, c.clone());
    }
    Ok(Ec2Service::respond(
        "ProvisionByoipCidr",
        &req.request_id,
        &format!("<byoipCidr>{}</byoipCidr>", byoip_cidr_xml(&c)),
    ))
}

pub(crate) fn withdraw_byoip_cidr(
    svc: &Ec2Service,
    req: &AwsRequest,
) -> Result<AwsResponse, AwsServiceError> {
    let cidr = require(&req.query_params, "Cidr")?;
    let c = set_byoip_state(svc, req, &cidr, "provisioned");
    Ok(Ec2Service::respond(
        "WithdrawByoipCidr",
        &req.request_id,
        &format!("<byoipCidr>{}</byoipCidr>", byoip_cidr_xml(&c)),
    ))
}