如何验证documentdb存储过程REST调用

时间:2016-08-26 08:50:16

标签: c# asp.net azure authentication azure-cosmosdb

编辑:问题是参数.ToString()。也像冷宁刘指出你需要添加' / sprocs / sprocname'

我尝试在ASP.NET Core 1.0 / C#中使用REST调用在Azure DocumentDB中执行存储过程。

我使用此MSDN页面上的方法生成哈希: https://msdn.microsoft.com/library/azure/dn783368.aspx

我获得了401 Unauthorized repsonse。是" sprocs"正确的文件类型,还是我必须输入其他类型?

public class Database
{

    public async Task<HttpResponseMessage> StoredProcedure(string database, string collection, string storedProcedure, string[] parameters)
    {
        //https://{databaseaccount}.documents.azure.com/dbs/{db-id}/colls/{coll-id}/sprocs/{sproc-name}

        string resourceLink = $"dbs/{database}/colls/{collection}";
        var client = Client("POST", resourceLink, "sprocs");

        StringContent content = new StringContent(parameters.ToString(), null, "application/query+json");
        var uri = new Uri(_endpointUri, $"dbs/{database}/colls/{collection}/sprocs/{storedProcedure}");
        HttpResponseMessage response = await client.PostAsync(uri, content);

        return response;
    }

    private HttpClient Client(string verb, string resourceLink, string resourceType)
    {
        var client = new HttpClient();
        var utc_date = DateTime.UtcNow.ToString("r");
        client.DefaultRequestHeaders.Add("x-ms-date", utc_date);
        client.DefaultRequestHeaders.Add("x-ms-version", "2015-12-16");
        client.DefaultRequestHeaders.Add("x-ms-max-item-count", "10000");

        var authHeader = GenerateMasterKeyAuthorizationSignature(utc_date, verb, resourceLink, resourceType, _authorizationKey, "master", "1.0");

        client.DefaultRequestHeaders.Add("authorization", authHeader);

        return client;
    }

    private static string GenerateMasterKeyAuthorizationSignature(string utc_date, string verb, string resourceId, string resourceType, string key, string keyType, string tokenVersion)
    {

        var hmacSha256 = new System.Security.Cryptography.HMACSHA256 { Key = Convert.FromBase64String(key) };

        string payLoad = string.Format(System.Globalization.CultureInfo.InvariantCulture, "{0}\n{1}\n{2}\n{3}\n{4}\n",
                verb.ToLowerInvariant(),
                resourceType.ToLowerInvariant(),
                resourceId,
                utc_date.ToLowerInvariant(),
                ""
        );

        byte[] hashPayLoad = hmacSha256.ComputeHash(System.Text.Encoding.UTF8.GetBytes(payLoad));
        string signature = Convert.ToBase64String(hashPayLoad);

        return System.Net.WebUtility.UrlEncode(String.Format(System.Globalization.CultureInfo.InvariantCulture, "type={0}&ver={1}&sig={2}",
            keyType,
            tokenVersion,
            signature));
    }
}

1 个答案:

答案 0 :(得分:2)

似乎在生成身份验证令牌时,在resourceId部分中,您省略了“/ sprocs / {storedProcedure}”。你把它包含在uri中,这是正确的。

我附加了一个示例powershell脚本,希望它还可以帮助您查看如何生成身份验证令牌。

Add-Type -AssemblyName System.Web 

$accountName  = "<db account name>"
$connectionKey = "<secret key>"
$collectionName = "<coll name>"
$databaseName = "<db name>"

Write-host ("Account " + $accountName)
Write-host ("Database  " + $databaseName)
Write-host ("Collection " + $collectionName)


    function GetKey([System.String]$Verb = '',[System.String]$ResourceId = '',
            [System.String]$ResourceType = '',[System.String]$Date = '',[System.String]$masterKey = '') {
        $keyBytes = [System.Convert]::FromBase64String($masterKey) 
        $text = @($Verb.ToLowerInvariant() + "`n" + $ResourceType.ToLowerInvariant() + "`n" + $ResourceId + "`n" + $Date.ToLowerInvariant() + "`n" + "" + "`n")
        $body =[Text.Encoding]::UTF8.GetBytes($text)
        $hmacsha = new-object -TypeName System.Security.Cryptography.HMACSHA256 -ArgumentList (,$keyBytes) 
        $hash = $hmacsha.ComputeHash($body)
        $signature = [System.Convert]::ToBase64String($hash)

        Write-Host($text)

        [System.Web.HttpUtility]::UrlEncode($('type=master&ver=1.0&sig=' + $signature))

    }

    function GetUTDate() {
        $date = get-date
        $date = $date.ToUniversalTime();
        return $date.ToString("r", [System.Globalization.CultureInfo]::InvariantCulture);
    }

    function GetDatabases() {
        $uri = $rootUri + "/dbs"

        $hdr = BuildHeaders -resType dbs

        $response = Invoke-RestMethod -Uri $uri -Method Get -Headers $hdr
        $response.Databases

        Write-Host ("Found " + $Response.Databases.Count + " Database(s)")

    }

    function GetCollections([string]$dbname){
        $uri = $rootUri + "/" + $dbname + "/colls"
        $headers = BuildHeaders -resType colls -resourceId $dbname
        $response = Invoke-RestMethod -Uri $uri -Method Get -Headers $headers
        $response.DocumentCollections
        Write-Host ("Found " + $Response.DocumentCollections.Count + " DocumentCollection(s)")
   }

    function BuildHeaders([string]$action = "get",[string]$resType, [string]$resourceId){
        $authz = GetKey -Verb $action -ResourceType $resType -ResourceId $resourceId -Date $apiDate -masterKey $connectionKey
        $headers = New-Object "System.Collections.Generic.Dictionary[[String],[String]]"
        $headers.Add("Authorization", $authz)
        $headers.Add("x-ms-version", '2015-12-16')
        $headers.Add("x-ms-date", $apiDate) 
        $headers
    }

    function PostDocument([string]$document, [string]$dbname, [string]$collection){
        $collName = "dbs/"+$dbname+"/colls/" + $collection
        $headers = BuildHeaders -action Post -resType docs -resourceId $collName
        $headers.Add("x-ms-documentdb-is-upsert", "true")
        $uri = $rootUri + "/" + $collName + "/docs"

        Write-host ("Calling " + $uri)

        $response = Invoke-RestMethod $uri -Method Post -Body $json -ContentType 'application/json' -Headers $headers
        $response
    }

    function PostSprocQuery([string]$dbname, [string]$collection){
        $sprocName = "dbs/"+$dbname+"/colls/" + $collection + "/sprocs/samplesproc"
        $headers = BuildHeaders -action Post -resType sprocs -resourceId $sprocName
        $uri = $rootUri + "/" + $sprocName

        Write-host ("Calling " + $uri)
        write-host $authz
        write-host $apiDate

        $response = Invoke-RestMethod $uri -Method Post -Body $json -ContentType 'application/json' -Headers $headers
        $response
    }

    $rootUri = "https://" + $accountName + ".documents.azure.com"
    write-host ("Root URI is " + $rootUri)

    #validate arguments

    $apiDate = GetUTDate

    $db = GetDatabases | where { $_.id -eq $databaseName }

    if ($db -eq $null) {
        write-error "Could not find database in account"
        return
    } 

    $dbname = "dbs/" + $databaseName
    $collection = GetCollections -dbname $dbname | where { $_.id -eq $collectionName }

    if($collection -eq $null){
        write-error "Could not find collection in database"
        return
    }

    $json = @"
{
    "id": "3"
}
"@ 
    PostDocument -document $json -dbname $databaseName -collection $collectionName

    $json = @"
[
    "samplesproc"
]
"@ 
    PostSprocQuery -document $json -dbname $databaseName -collection $collectionName