Skip to content

Commit 7cba680

Browse files
abhinav-qlogicJesseLovelace
abhinav-qlogic
authored and
JesseLovelace
committed
---
yaml --- r: 34391 b: refs/heads/autosynth-tasks c: 03c1786 h: refs/heads/master i: 34389: 67cd6a2 34387: 1078b42 34383: 9e8cca5
1 parent cafe5f7 commit 7cba680

11 files changed

Lines changed: 290 additions & 1 deletion

File tree

‎[refs]‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -139,7 +139,7 @@ refs/heads/autosynth-redis: 2b698ebe7faaa0b93246576a88941682e0888e57
139139
refs/heads/autosynth-scheduler: a3de6480746d1cd586ca8b9d75c55a636f371539
140140
refs/heads/autosynth-spanner: d963fe4368e79cf6abae5d511785e8ced8ac57f4
141141
refs/heads/autosynth-speech: c563dcd420cce0a37c39b1b9c24be1b9ba604dc7
142-
refs/heads/autosynth-tasks: 47884a3c08d216d4ae7a7a891291af717fb47ef5
142+
refs/heads/autosynth-tasks: 03c1786b55cd79171e9ac73909d81ee49b7d8452
143143
refs/heads/autosynth-texttospeech: 2c442fe0b7f089fbab266edfe4dd83c532e82dd0
144144
refs/heads/autosynth-trace: c94eef6e4d9c6fd24888216e28ca7271959c1cf0
145145
refs/heads/autosynth-websecurityscanner: fa561b356aabcd92d415ae8dc88fd8d87dbc5b23

‎branches/autosynth-tasks/google-cloud-clients/google-cloud-contrib/google-cloud-nio/src/main/java/com/google/cloud/storage/contrib/nio/testing/FakeStorageRpc.java‎

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -312,6 +312,11 @@ public String open(StorageObject object, Map<Option, ?> options) throws StorageE
312312
return fullname(object);
313313
}
314314

315+
@Override
316+
public String open(String signedURL) {
317+
return null;
318+
}
319+
315320
@Override
316321
public void write(
317322
String uploadId, byte[] toWrite, int toWriteOffset, long destOffset, int length, boolean last)

‎branches/autosynth-tasks/google-cloud-clients/google-cloud-storage/.attach_pid8524‎

Whitespace-only changes.

‎branches/autosynth-tasks/google-cloud-clients/google-cloud-storage/src/main/java/com/google/cloud/storage/BlobWriteChannel.java‎

Lines changed: 49 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@
2424
import com.google.cloud.RetryHelper;
2525
import com.google.cloud.WriteChannel;
2626
import com.google.cloud.storage.spi.v1.StorageRpc;
27+
import java.net.URL;
2728
import java.util.Map;
2829
import java.util.concurrent.Callable;
2930

@@ -34,10 +35,18 @@ class BlobWriteChannel extends BaseWriteChannel<StorageOptions, BlobInfo> {
3435
this(options, blob, open(options, blob, optionsMap));
3536
}
3637

38+
BlobWriteChannel(StorageOptions options, URL signedURL) {
39+
this(options, open(signedURL, options));
40+
}
41+
3742
BlobWriteChannel(StorageOptions options, BlobInfo blobInfo, String uploadId) {
3843
super(options, blobInfo, uploadId);
3944
}
4045

46+
BlobWriteChannel(StorageOptions options, String uploadId) {
47+
super(options, null, uploadId);
48+
}
49+
4150
@Override
4251
protected void flushBuffer(final int length, final boolean last) {
4352
try {
@@ -83,6 +92,46 @@ public String call() {
8392
}
8493
}
8594

95+
private static String open(final URL signedURL, final StorageOptions options) {
96+
try {
97+
return runWithRetries(
98+
new Callable<String>() {
99+
@Override
100+
public String call() {
101+
if (!isValidSignedURL(signedURL.getQuery())) {
102+
throw new StorageException(2, "invalid signedURL");
103+
}
104+
return options.getStorageRpcV1().open(signedURL.toString());
105+
}
106+
},
107+
options.getRetrySettings(),
108+
StorageImpl.EXCEPTION_HANDLER,
109+
options.getClock());
110+
} catch (RetryHelper.RetryHelperException e) {
111+
throw StorageException.translateAndThrow(e);
112+
}
113+
}
114+
115+
private static boolean isValidSignedURL(String signedURLQuery) {
116+
boolean isValid = true;
117+
if (signedURLQuery.startsWith("X-Goog-Algorithm=")) {
118+
if (!signedURLQuery.contains("&X-Goog-Credential=")
119+
|| !signedURLQuery.contains("&X-Goog-Date=")
120+
|| !signedURLQuery.contains("&X-Goog-Expires=")
121+
|| !signedURLQuery.contains("&X-Goog-SignedHeaders=")
122+
|| !signedURLQuery.contains("&X-Goog-Signature=")) {
123+
isValid = false;
124+
}
125+
} else if (signedURLQuery.startsWith("GoogleAccessId=")) {
126+
if (!signedURLQuery.contains("&Expires=") || !signedURLQuery.contains("&Signature=")) {
127+
isValid = false;
128+
}
129+
} else {
130+
isValid = false;
131+
}
132+
return isValid;
133+
}
134+
86135
static class StateImpl extends BaseWriteChannel.BaseState<StorageOptions, BlobInfo> {
87136

88137
private static final long serialVersionUID = -9028324143780151286L;

‎branches/autosynth-tasks/google-cloud-clients/google-cloud-storage/src/main/java/com/google/cloud/storage/Storage.java‎

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2091,6 +2091,27 @@ Blob create(
20912091
*/
20922092
WriteChannel writer(BlobInfo blobInfo, BlobWriteOption... options);
20932093

2094+
/**
2095+
* Accepts signed URL and return a channel for writing content.
2096+
*
2097+
* <p>Example of writing content through a writer using signed URL.
2098+
*
2099+
* <pre>{@code
2100+
* String bucketName = "my_unique_bucket";
2101+
* String blobName = "my_blob_name";
2102+
* BlobId blobId = BlobId.of(bucketName, blobName);
2103+
* byte[] content = "Hello, World!".getBytes(UTF_8);
2104+
* BlobInfo blobInfo = BlobInfo.newBuilder(blobId).setContentType("text/plain").build();
2105+
* URL signedURL = storage.signUrl(blobInfo, 1, TimeUnit.HOURS, Storage.SignUrlOption.httpMethod(HttpMethod.POST));
2106+
* try (WriteChannel writer = storage.writer(signedURL)) {
2107+
* writer.write(ByteBuffer.wrap(content, 0, content.length));
2108+
* }
2109+
* }</pre>
2110+
*
2111+
* @throws StorageException upon failure
2112+
*/
2113+
WriteChannel writer(URL signedURL);
2114+
20942115
/**
20952116
* Generates a signed URL for a blob. If you have a blob that you want to allow access to for a
20962117
* fixed amount of time, you can use this method to generate a URL that is only valid within a

‎branches/autosynth-tasks/google-cloud-clients/google-cloud-storage/src/main/java/com/google/cloud/storage/StorageImpl.java‎

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -599,6 +599,11 @@ public BlobWriteChannel writer(BlobInfo blobInfo, BlobWriteOption... options) {
599599
return writer(targetOptions.x(), targetOptions.y());
600600
}
601601

602+
@Override
603+
public BlobWriteChannel writer(URL signedURL) {
604+
return new BlobWriteChannel(getOptions(), signedURL);
605+
}
606+
602607
private BlobWriteChannel writer(BlobInfo blobInfo, BlobTargetOption... options) {
603608
final Map<StorageRpc.Option, ?> optionsMap = optionMap(blobInfo, options);
604609
return new BlobWriteChannel(getOptions(), blobInfo, optionsMap);

‎branches/autosynth-tasks/google-cloud-clients/google-cloud-storage/src/main/java/com/google/cloud/storage/spi/v1/HttpStorageRpc.java‎

Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -822,6 +822,39 @@ public String open(StorageObject object, Map<Option, ?> options) {
822822
}
823823
}
824824

825+
@Override
826+
public String open(String signedURL) {
827+
Span span = startSpan(HttpStorageRpcSpans.SPAN_NAME_OPEN);
828+
Scope scope = tracer.withSpan(span);
829+
try {
830+
GenericUrl url = new GenericUrl(signedURL);
831+
url.set("uploadType", "resumable");
832+
String bytesArrayParameters = "";
833+
byte[] bytesArray = new byte[bytesArrayParameters.length()];
834+
HttpRequestFactory requestFactory = storage.getRequestFactory();
835+
HttpRequest httpRequest =
836+
requestFactory.buildPostRequest(
837+
url, new ByteArrayContent("", bytesArray, 0, bytesArray.length));
838+
HttpHeaders requestHeaders = httpRequest.getHeaders();
839+
requestHeaders.set("X-Upload-Content-Type", "");
840+
requestHeaders.set("x-goog-resumable", "start");
841+
HttpResponse response = httpRequest.execute();
842+
if (response.getStatusCode() != 201) {
843+
GoogleJsonError error = new GoogleJsonError();
844+
error.setCode(response.getStatusCode());
845+
error.setMessage(response.getStatusMessage());
846+
throw translate(error);
847+
}
848+
return response.getHeaders().getLocation();
849+
} catch (IOException ex) {
850+
span.setStatus(Status.UNKNOWN.withDescription(ex.getMessage()));
851+
throw translate(ex);
852+
} finally {
853+
scope.close();
854+
span.end();
855+
}
856+
}
857+
825858
@Override
826859
public RewriteResponse openRewrite(RewriteRequest rewriteRequest) {
827860
Span span = startSpan(HttpStorageRpcSpans.SPAN_NAME_OPEN_REWRITE);

‎branches/autosynth-tasks/google-cloud-clients/google-cloud-storage/src/main/java/com/google/cloud/storage/spi/v1/StorageRpc.java‎

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -289,6 +289,13 @@ StorageObject compose(
289289
*/
290290
String open(StorageObject object, Map<Option, ?> options);
291291

292+
/**
293+
* Opens a resumable upload channel for a given signedURL.
294+
*
295+
* @throws StorageException upon failure
296+
*/
297+
String open(String signedURL);
298+
292299
/**
293300
* Writes the provided bytes to a storage object at the provided location.
294301
*

‎branches/autosynth-tasks/google-cloud-clients/google-cloud-storage/src/test/java/com/google/cloud/storage/BlobWriteChannelTest.java‎

Lines changed: 131 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,9 @@
3636
import com.google.cloud.storage.spi.v1.StorageRpc;
3737
import com.google.common.collect.ImmutableMap;
3838
import java.io.IOException;
39+
import java.net.MalformedURLException;
3940
import java.net.SocketException;
41+
import java.net.URL;
4042
import java.nio.ByteBuffer;
4143
import java.util.Arrays;
4244
import java.util.Map;
@@ -60,6 +62,8 @@ public class BlobWriteChannelTest {
6062
private static final int DEFAULT_CHUNK_SIZE = 8 * MIN_CHUNK_SIZE;
6163
private static final int CUSTOM_CHUNK_SIZE = 4 * MIN_CHUNK_SIZE;
6264
private static final Random RANDOM = new Random();
65+
private static final String SIGNED_URL =
66+
"http://www.test.com/test-bucket/[email protected]&Expires=1553839761&Signature=MJUBXAZ7";
6367

6468
@Rule public ExpectedException thrown = ExpectedException.none();
6569

@@ -265,6 +269,133 @@ public void testStateEquals() {
265269
assertEquals(state.toString(), state2.toString());
266270
}
267271

272+
@Test
273+
public void testWriteWithSignedURLAndWithoutFlush() throws IOException {
274+
expect(storageRpcMock.open(SIGNED_URL)).andReturn(UPLOAD_ID);
275+
replay(storageRpcMock);
276+
writer = new BlobWriteChannel(options, new URL(SIGNED_URL));
277+
assertEquals(MIN_CHUNK_SIZE, writer.write(ByteBuffer.allocate(MIN_CHUNK_SIZE)));
278+
}
279+
280+
@Test
281+
public void testWriteWithSignedURLAndWithFlush() throws IOException {
282+
expect(storageRpcMock.open(SIGNED_URL)).andReturn(UPLOAD_ID);
283+
Capture<byte[]> capturedBuffer = Capture.newInstance();
284+
storageRpcMock.write(
285+
eq(UPLOAD_ID), capture(capturedBuffer), eq(0), eq(0L), eq(CUSTOM_CHUNK_SIZE), eq(false));
286+
replay(storageRpcMock);
287+
writer = new BlobWriteChannel(options, new URL(SIGNED_URL));
288+
writer.setChunkSize(CUSTOM_CHUNK_SIZE);
289+
ByteBuffer buffer = randomBuffer(CUSTOM_CHUNK_SIZE);
290+
assertEquals(CUSTOM_CHUNK_SIZE, writer.write(buffer));
291+
assertArrayEquals(buffer.array(), capturedBuffer.getValue());
292+
}
293+
294+
@Test
295+
public void testWriteWithSignedURLAndFlush() throws IOException {
296+
expect(storageRpcMock.open(SIGNED_URL)).andReturn(UPLOAD_ID);
297+
Capture<byte[]> capturedBuffer = Capture.newInstance();
298+
storageRpcMock.write(
299+
eq(UPLOAD_ID), capture(capturedBuffer), eq(0), eq(0L), eq(DEFAULT_CHUNK_SIZE), eq(false));
300+
replay(storageRpcMock);
301+
writer = new BlobWriteChannel(options, new URL(SIGNED_URL));
302+
ByteBuffer[] buffers = new ByteBuffer[DEFAULT_CHUNK_SIZE / MIN_CHUNK_SIZE];
303+
for (int i = 0; i < buffers.length; i++) {
304+
buffers[i] = randomBuffer(MIN_CHUNK_SIZE);
305+
assertEquals(MIN_CHUNK_SIZE, writer.write(buffers[i]));
306+
}
307+
for (int i = 0; i < buffers.length; i++) {
308+
assertArrayEquals(
309+
buffers[i].array(),
310+
Arrays.copyOfRange(
311+
capturedBuffer.getValue(), MIN_CHUNK_SIZE * i, MIN_CHUNK_SIZE * (i + 1)));
312+
}
313+
}
314+
315+
@Test
316+
public void testCloseWithSignedURLWithoutFlush() throws IOException {
317+
expect(storageRpcMock.open(SIGNED_URL)).andReturn(UPLOAD_ID);
318+
Capture<byte[]> capturedBuffer = Capture.newInstance();
319+
storageRpcMock.write(eq(UPLOAD_ID), capture(capturedBuffer), eq(0), eq(0L), eq(0), eq(true));
320+
replay(storageRpcMock);
321+
writer = new BlobWriteChannel(options, new URL(SIGNED_URL));
322+
assertTrue(writer.isOpen());
323+
writer.close();
324+
assertArrayEquals(new byte[0], capturedBuffer.getValue());
325+
assertTrue(!writer.isOpen());
326+
}
327+
328+
@Test
329+
public void testCloseWithSignedURLWithFlush() throws IOException {
330+
expect(storageRpcMock.open(SIGNED_URL)).andReturn(UPLOAD_ID);
331+
Capture<byte[]> capturedBuffer = Capture.newInstance();
332+
ByteBuffer buffer = randomBuffer(MIN_CHUNK_SIZE);
333+
storageRpcMock.write(
334+
eq(UPLOAD_ID), capture(capturedBuffer), eq(0), eq(0L), eq(MIN_CHUNK_SIZE), eq(true));
335+
replay(storageRpcMock);
336+
writer = new BlobWriteChannel(options, new URL(SIGNED_URL));
337+
assertTrue(writer.isOpen());
338+
writer.write(buffer);
339+
writer.close();
340+
assertEquals(DEFAULT_CHUNK_SIZE, capturedBuffer.getValue().length);
341+
assertArrayEquals(buffer.array(), Arrays.copyOf(capturedBuffer.getValue(), MIN_CHUNK_SIZE));
342+
assertTrue(!writer.isOpen());
343+
}
344+
345+
@Test
346+
public void testWriteWithSignedURLClosed() throws IOException {
347+
expect(storageRpcMock.open(SIGNED_URL)).andReturn(UPLOAD_ID);
348+
Capture<byte[]> capturedBuffer = Capture.newInstance();
349+
storageRpcMock.write(eq(UPLOAD_ID), capture(capturedBuffer), eq(0), eq(0L), eq(0), eq(true));
350+
replay(storageRpcMock);
351+
writer = new BlobWriteChannel(options, new URL(SIGNED_URL));
352+
writer.close();
353+
try {
354+
writer.write(ByteBuffer.allocate(MIN_CHUNK_SIZE));
355+
fail("Expected BlobWriteChannel write to throw IOException");
356+
} catch (IOException ex) {
357+
// expected
358+
}
359+
}
360+
361+
@Test
362+
public void testSaveAndRestoreWithSignedURL() throws IOException {
363+
expect(storageRpcMock.open(SIGNED_URL)).andReturn(UPLOAD_ID);
364+
Capture<byte[]> capturedBuffer = Capture.newInstance(CaptureType.ALL);
365+
Capture<Long> capturedPosition = Capture.newInstance(CaptureType.ALL);
366+
storageRpcMock.write(
367+
eq(UPLOAD_ID),
368+
capture(capturedBuffer),
369+
eq(0),
370+
captureLong(capturedPosition),
371+
eq(DEFAULT_CHUNK_SIZE),
372+
eq(false));
373+
expectLastCall().times(2);
374+
replay(storageRpcMock);
375+
ByteBuffer buffer1 = randomBuffer(DEFAULT_CHUNK_SIZE);
376+
ByteBuffer buffer2 = randomBuffer(DEFAULT_CHUNK_SIZE);
377+
writer = new BlobWriteChannel(options, new URL(SIGNED_URL));
378+
assertEquals(DEFAULT_CHUNK_SIZE, writer.write(buffer1));
379+
assertArrayEquals(buffer1.array(), capturedBuffer.getValues().get(0));
380+
assertEquals(new Long(0L), capturedPosition.getValues().get(0));
381+
RestorableState<WriteChannel> writerState = writer.capture();
382+
WriteChannel restoredWriter = writerState.restore();
383+
assertEquals(DEFAULT_CHUNK_SIZE, restoredWriter.write(buffer2));
384+
assertArrayEquals(buffer2.array(), capturedBuffer.getValues().get(1));
385+
assertEquals(new Long(DEFAULT_CHUNK_SIZE), capturedPosition.getValues().get(1));
386+
}
387+
388+
@Test
389+
public void testRuntimeExceptionWithSignedURL() throws MalformedURLException {
390+
String exceptionMessage = "invalid signedURL";
391+
expect(new BlobWriteChannel(options, new URL(SIGNED_URL)))
392+
.andThrow(new RuntimeException(exceptionMessage));
393+
replay(storageRpcMock);
394+
thrown.expect(StorageException.class);
395+
thrown.expectMessage(exceptionMessage);
396+
writer = new BlobWriteChannel(options, new URL(SIGNED_URL));
397+
}
398+
268399
private static ByteBuffer randomBuffer(int size) {
269400
byte[] byteArray = new byte[size];
270401
RANDOM.nextBytes(byteArray);

‎branches/autosynth-tasks/google-cloud-clients/google-cloud-storage/src/test/java/com/google/cloud/storage/StorageImplTest.java‎

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,7 @@
5959
import java.io.ByteArrayInputStream;
6060
import java.io.IOException;
6161
import java.io.UnsupportedEncodingException;
62+
import java.net.MalformedURLException;
6263
import java.net.URL;
6364
import java.net.URLDecoder;
6465
import java.nio.ByteBuffer;
@@ -319,6 +320,9 @@ public class StorageImplTest {
319320
+ "EkPPhszldvQTY486uPxyD/D7HdfnGW/Nbw5JUhfvecAdudDEhNAQ3PNabyDMI+TpiHy4NTWOrgdcWrzj6VXcdc"
320321
+ "+uuABnPwRCdcyJ1xl2kOrPksRnp1auNGMLOe4IpEBjGY7baX9UG8+A45MbG0aHmkR59Op/aR9XowIDAQAB";
321322

323+
private static final String SIGNED_URL =
324+
"http://www.test.com/test-bucket/[email protected]&Expires=1553839761&Signature=MJUBXAZ7";
325+
322326
private static final ApiClock TIME_SOURCE =
323327
new ApiClock() {
324328
@Override
@@ -2835,4 +2839,14 @@ public void testRuntimeException() {
28352839
thrown.expectMessage(exceptionMessage);
28362840
storage.get(blob);
28372841
}
2842+
2843+
@Test
2844+
public void testWriterWithSignedURL() throws MalformedURLException {
2845+
EasyMock.expect(storageRpcMock.open(SIGNED_URL)).andReturn("upload-id");
2846+
EasyMock.replay(storageRpcMock);
2847+
initializeService();
2848+
WriteChannel writer = new BlobWriteChannel(options, new URL(SIGNED_URL));
2849+
assertNotNull(writer);
2850+
assertTrue(writer.isOpen());
2851+
}
28382852
}

0 commit comments

Comments
 (0)