Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

{Attestation} az attestation signer/policy: Bump API version from 2022-10-01 to 2022-08-01 #7059

Merged
merged 21 commits into from
Dec 11, 2023
Merged
Prev Previous commit
Next Next commit
update
  • Loading branch information
ZengTaoxu committed Dec 6, 2023
commit ec8cfb39787eb54cf6a7043df4fe00c65f5e3bb5
64 changes: 0 additions & 64 deletions src/attestation/azext_attestation/manual/custom.py
Original file line number Diff line number Diff line change
Expand Up @@ -309,41 +309,6 @@ def _output(self, *args, **kwargs):
return result


def get_policy(cmd, client, attestation_type, resource_group_name=None, provider_name=None):
""" Retrieves the current policy for a given kind of attestation type. """

provider_client = cf_attestation_provider(cmd.cli_ctx)
provider = provider_client.get(resource_group_name=resource_group_name, provider_name=provider_name)
token = client.get(tenant_base_url=provider.attest_uri, tee=tee_mapping[attestation_type]).token
result = {}

if token:
import jwt
policy = jwt.decode(token, algorithms=['RS256'], options={"verify_signature": False}).get('x-ms-policy', '')
result['Jwt'] = policy
result['JwtLength'] = len(policy)
result['Algorithm'] = None

if policy:
try:
decoded_policy = jwt.decode(policy, algorithms=['RS256'], options={"verify_signature": False})
decoded_policy = decoded_policy.get('AttestationPolicy', '')
try:
new_decoded_policy = base64.b64decode(_b64url_to_b64(decoded_policy)).decode('ascii')
decoded_policy = new_decoded_policy
except: # pylint: disable=bare-except
pass
finally:
result['Text'] = decoded_policy
result['TextLength'] = len(decoded_policy)
result['Algorithm'] = jwt.get_unverified_header(policy).get('alg', None)
except: # pylint: disable=bare-except
result['Text'] = ''
result['TextLength'] = 0

return result


class GetPolicy(_GetPolicy):
@classmethod
def _build_arguments_schema(cls, *args, **kwargs):
Expand Down Expand Up @@ -533,35 +498,6 @@ def _output(self, *args, **kwargs):
return handle_policy_output(token)


def attest_open_enclave(cmd, client, report=None, runtime_data=None, runtime_data_type=None, init_time_data=None,
init_time_data_type=None, resource_group_name=None, provider_name=None):

provider_client = cf_attestation_provider(cmd.cli_ctx)
provider = provider_client.get(resource_group_name=resource_group_name, provider_name=provider_name)

request = AttestOpenEnclaveRequest(
report=report,
runtime_data=RuntimeData(
data=runtime_data,
data_type=runtime_data_type
),
init_time_data=InitTimeData(
data=init_time_data,
data_type=init_time_data_type
)
)

return client.attest_open_enclave(
tenant_base_url=provider.attest_uri,
request=request
)


def attestation_attestation_provider_get_default_by_location(client,
loc):
return client.get_default_by_location(location=loc)


class AttestationGetDefaultByLocation(_AttestationGetDefaultByLocation):
@classmethod
def _build_arguments_schema(cls, *args, **kwargs):
Expand Down
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
|Scenario|Result|ErrorMessage|ErrorStack|ErrorNormalized|StartDt|EndDt|
|step__attestationproviders_put|successed||||2023-12-06 02:40:18.942935|2023-12-06 02:40:19.323602|
|step__attestationproviders_get|successed||||2023-12-06 02:40:19.323602|2023-12-06 02:40:19.417352|
|step__attestationproviders_get2|successed||||2023-12-06 02:40:19.418300|2023-12-06 02:40:19.509976|
|step__attestationproviders_get3|successed||||2023-12-06 02:40:19.509976|2023-12-06 02:40:19.631672|
|step__attestationproviders_patch|successed||||2023-12-06 02:40:19.631672|2023-12-06 02:40:19.739275|
|step__attestationproviders_delete|successed||||2023-12-06 02:40:19.739275|2023-12-06 02:40:19.832407|
|step__attestationproviders_put|successed||||2023-12-06 02:59:17.849492|2023-12-06 02:59:18.052992|
|step__attestationproviders_get|successed||||2023-12-06 02:59:18.052992|2023-12-06 02:59:18.132149|
|step__attestationproviders_get2|successed||||2023-12-06 02:59:18.132149|2023-12-06 02:59:18.226650|
|step__attestationproviders_get3|successed||||2023-12-06 02:59:18.226650|2023-12-06 02:59:18.320428|
|step__attestationproviders_patch|successed||||2023-12-06 02:59:18.320428|2023-12-06 02:59:18.400530|
|step__attestationproviders_delete|successed||||2023-12-06 02:59:18.400530|2023-12-06 02:59:18.462107|
Coverage: 6/6
Loading