From 874d835e7571fcb0ab6bf5deea62f93e84e5de42 Mon Sep 17 00:00:00 2001
From: adriancole <adrian.f.cole@gmail.com>
Date: Sun, 21 Apr 2013 13:13:09 -0700
Subject: [PATCH] fix issue #1492: invalidate dynect session on ip mismatch

---
 .../dynect/v3/filters/SessionManager.java     |  7 ++--
 .../dynect/v3/filters/SessionManagerTest.java | 32 +++++++++++++++++++
 .../src/test/resources/ip_mismatch.json       | 17 ++++++++++
 3 files changed, 54 insertions(+), 2 deletions(-)
 create mode 100644 providers/dynect/src/test/resources/ip_mismatch.json

diff --git a/providers/dynect/src/main/java/org/jclouds/dynect/v3/filters/SessionManager.java b/providers/dynect/src/main/java/org/jclouds/dynect/v3/filters/SessionManager.java
index 07174e5611..7cfdb0253e 100644
--- a/providers/dynect/src/main/java/org/jclouds/dynect/v3/filters/SessionManager.java
+++ b/providers/dynect/src/main/java/org/jclouds/dynect/v3/filters/SessionManager.java
@@ -101,12 +101,15 @@ public final class SessionManager extends BackoffLimitedRetryHandler implements
       return builder.build();
    }
 
+   private static final String IP_MISMATCH = "IP address does not match current session";
+
    @Override
    public boolean shouldRetryRequest(HttpCommand command, HttpResponse response) {
       boolean retry = false; // default
       try {
-         if (response.getStatusCode() == 401) {
-            closeClientButKeepContentStream(response);
+         byte[] data = closeClientButKeepContentStream(response);
+         String message = data != null ? new String(data) : null;
+         if (response.getStatusCode() == 401 || (message != null && message.indexOf(IP_MISMATCH) != -1)) {
             logger.debug("invalidating session");
             sessionCache.invalidateAll();
             retry = super.shouldRetryRequest(command, response);
diff --git a/providers/dynect/src/test/java/org/jclouds/dynect/v3/filters/SessionManagerTest.java b/providers/dynect/src/test/java/org/jclouds/dynect/v3/filters/SessionManagerTest.java
index 66f25aea55..c80401e75b 100644
--- a/providers/dynect/src/test/java/org/jclouds/dynect/v3/filters/SessionManagerTest.java
+++ b/providers/dynect/src/test/java/org/jclouds/dynect/v3/filters/SessionManagerTest.java
@@ -18,6 +18,8 @@
  */
 package org.jclouds.dynect.v3.filters;
 
+import static com.google.common.io.Resources.getResource;
+import static javax.ws.rs.core.Response.Status.BAD_REQUEST;
 import static javax.ws.rs.core.Response.Status.FORBIDDEN;
 import static javax.ws.rs.core.Response.Status.UNAUTHORIZED;
 import static org.easymock.EasyMock.createMock;
@@ -30,6 +32,8 @@ import static org.testng.Assert.assertFalse;
 import static org.testng.Assert.assertSame;
 import static org.testng.Assert.assertTrue;
 
+import java.io.IOException;
+
 import org.jclouds.domain.Credentials;
 import org.jclouds.dynect.v3.domain.Session;
 import org.jclouds.dynect.v3.domain.SessionCredentials;
@@ -95,6 +99,34 @@ public class SessionManagerTest {
       verify(creds, sessionCache, sessionApi, command);
    }
 
+   @SuppressWarnings("unchecked")
+   @Test
+   public void testIPMismatchShouldInvalidateSessionAndRetry() throws IOException {
+      HttpCommand command = createMock(HttpCommand.class);
+      Supplier<Credentials> creds = createMock(Supplier.class);
+      LoadingCache<Credentials, Session> sessionCache = createMock(LoadingCache.class);
+      SessionApi sessionApi = createMock(SessionApi.class);
+
+      sessionCache.invalidateAll();
+      expectLastCall();
+      expect(command.incrementFailureCount()).andReturn(1);
+      expect(command.isReplayable()).andReturn(true);
+      expect(command.getFailureCount()).andReturn(1).atLeastOnce();
+
+      replay(creds, sessionCache, sessionApi, command);
+
+      HttpResponse response = HttpResponse.builder()
+                                          .statusCode(BAD_REQUEST.getStatusCode())
+                                          .payload(getResource("ip_mismatch.json").openStream())
+                                          .build();
+
+      SessionManager retry = new SessionManager(creds, sessionCache, sessionApi);
+
+      assertTrue(retry.shouldRetryRequest(command, response));
+
+      verify(creds, sessionCache, sessionApi, command);
+   }
+
    @SuppressWarnings("unchecked")
    @Test
    public void testForbiddenShouldNotInvalidateSessionOrRetry() {
diff --git a/providers/dynect/src/test/resources/ip_mismatch.json b/providers/dynect/src/test/resources/ip_mismatch.json
new file mode 100644
index 0000000000..1fdcb6b3f0
--- /dev/null
+++ b/providers/dynect/src/test/resources/ip_mismatch.json
@@ -0,0 +1,17 @@
+{
+    "status": "failure",
+    "data": {},
+    "job_id": 305900967,
+    "msgs": [{
+            "INFO": "login: IP address does not match current session",
+            "SOURCE": "BLL",
+            "ERR_CD": "INVALID_DATA",
+            "LVL": "ERROR"
+        }, {
+            "INFO": "login: There was a problem with your credentials",
+            "SOURCE": "BLL",
+            "ERR_CD": null,
+            "LVL": "INFO"
+        }
+    ]
+}