зеркало из https://github.com/mono/nuget.git
Fix of work item 4050: push does not follow redirect.
This commit is contained in:
Родитель
6c62b543c7
Коммит
2cda345ffd
|
@ -1,7 +1,7 @@
|
|||
//------------------------------------------------------------------------------
|
||||
// <auto-generated>
|
||||
// This code was generated by a tool.
|
||||
// Runtime Version:4.0.30319.34003
|
||||
// Runtime Version:4.0.30319.34011
|
||||
//
|
||||
// Changes to this file may cause incorrect behavior and will be lost if
|
||||
// the code is regenerated.
|
||||
|
@ -285,6 +285,15 @@ namespace NuGet.Resources {
|
|||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Looks up a localized string similar to Too many automatic redirections were attempted..
|
||||
/// </summary>
|
||||
public static string Error_TooManyRedirections {
|
||||
get {
|
||||
return ResourceManager.GetString("Error_TooManyRedirections", resourceCulture);
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Looks up a localized string similar to An error occurred while loading packages from '{0}': {1}.
|
||||
/// </summary>
|
||||
|
|
|
@ -469,4 +469,7 @@
|
|||
<data name="Log_UninstallPackageFromProject" xml:space="preserve">
|
||||
<value>Remove '{0}' from project {1}.</value>
|
||||
</data>
|
||||
<data name="Error_TooManyRedirections" xml:space="preserve">
|
||||
<value>Too many automatic redirections were attempted.</value>
|
||||
</data>
|
||||
</root>
|
|
@ -10,8 +10,9 @@ namespace NuGet
|
|||
{
|
||||
private const string ServiceEndpoint = "/api/v2/package";
|
||||
private const string ApiKeyHeader = "X-NuGet-ApiKey";
|
||||
private const int MaxRediretionCount = 20;
|
||||
|
||||
private readonly Lazy<Uri> _baseUri;
|
||||
private Lazy<Uri> _baseUri;
|
||||
private readonly string _source;
|
||||
private readonly string _userAgent;
|
||||
|
||||
|
@ -78,6 +79,9 @@ namespace NuGet
|
|||
Func<Stream> packageStreamFactory,
|
||||
long packageSize,
|
||||
int timeout)
|
||||
{
|
||||
int redirectionCount = 0;
|
||||
while (true)
|
||||
{
|
||||
HttpClient client = GetClient("", "PUT", "application/octet-stream");
|
||||
|
||||
|
@ -106,7 +110,20 @@ namespace NuGet
|
|||
multiPartRequest.CreateMultipartRequest(request);
|
||||
};
|
||||
|
||||
EnsureSuccessfulResponse(client);
|
||||
// Since AllowWriteStreamBuffering is set to false, redirection will not be handled
|
||||
// automatically by HttpWebRequest. So we need to check redirect status code and
|
||||
// update _baseUri and retry if redirection happens.
|
||||
if (EnsureSuccessfulResponse(client))
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
++redirectionCount;
|
||||
if (redirectionCount > MaxRediretionCount)
|
||||
{
|
||||
throw new WebException(NuGetResources.Error_TooManyRedirections);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
|
@ -214,7 +231,15 @@ namespace NuGet
|
|||
return requestUri;
|
||||
}
|
||||
|
||||
private static void EnsureSuccessfulResponse(HttpClient client, HttpStatusCode? expectedStatusCode = null)
|
||||
/// <summary>
|
||||
/// Ensures that success response is received.
|
||||
/// </summary>
|
||||
/// <param name="client">The client that is making the request.</param>
|
||||
/// <param name="expectedStatusCode">The exected status code.</param>
|
||||
/// <returns>True if success response is received; false if redirection response is received.
|
||||
/// In this case, _baseUri will be updated to be the new redirected Uri and the requrest
|
||||
/// should be retried.</returns>
|
||||
private bool EnsureSuccessfulResponse(HttpClient client, HttpStatusCode? expectedStatusCode = null)
|
||||
{
|
||||
HttpWebResponse response = null;
|
||||
try
|
||||
|
@ -229,6 +254,8 @@ namespace NuGet
|
|||
{
|
||||
throw new InvalidOperationException(String.Format(CultureInfo.CurrentCulture, NuGetResources.PackageServerError, response.StatusDescription, String.Empty));
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
catch (WebException e)
|
||||
{
|
||||
|
@ -237,10 +264,31 @@ namespace NuGet
|
|||
throw;
|
||||
}
|
||||
response = (HttpWebResponse)e.Response;
|
||||
|
||||
// Check if the error is caused by redirection
|
||||
if (response.StatusCode == HttpStatusCode.MultipleChoices ||
|
||||
response.StatusCode == HttpStatusCode.MovedPermanently ||
|
||||
response.StatusCode == HttpStatusCode.Found ||
|
||||
response.StatusCode == HttpStatusCode.SeeOther ||
|
||||
response.StatusCode == HttpStatusCode.TemporaryRedirect)
|
||||
{
|
||||
var location = response.Headers["Location"];
|
||||
Uri newUri;
|
||||
if (!Uri.TryCreate(client.Uri, location, out newUri))
|
||||
{
|
||||
throw;
|
||||
}
|
||||
|
||||
_baseUri = new Lazy<Uri>(() => newUri);
|
||||
return false;
|
||||
}
|
||||
|
||||
if (expectedStatusCode != response.StatusCode)
|
||||
{
|
||||
throw new InvalidOperationException(String.Format(CultureInfo.CurrentCulture, NuGetResources.PackageServerError, response.StatusDescription, e.Message), e);
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
finally
|
||||
{
|
||||
|
|
Загрузка…
Ссылка в новой задаче