1 files changed, 151 insertions, 0 deletions
diff --git a/src/main/java/com/google/android/mobly/snippet/bundled/NetworkingSnippet.java b/src/main/java/com/google/android/mobly/snippet/bundled/NetworkingSnippet.java
new file mode 100644
index 0000000..636c0fd
--- /dev/null
+++ b/src/main/java/com/google/android/mobly/snippet/bundled/NetworkingSnippet.java
@@ -0,0 +1,151 @@
+/*
+ * Copyright (C) 2017 Google Inc.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License"); you may not
+ * use this file except in compliance with the License. You may obtain a copy of
+ * the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+ * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+ * License for the specific language governing permissions and limitations under
+ * the License.
+ */
+
+package com.google.android.mobly.snippet.bundled;
+
+import android.app.DownloadManager;
+import android.content.BroadcastReceiver;
+import android.content.Context;
+import android.content.Intent;
+import android.content.IntentFilter;
+import android.net.Uri;
+import android.os.Environment;
+import androidx.test.platform.app.InstrumentationRegistry;
+import com.google.android.mobly.snippet.Snippet;
+import com.google.android.mobly.snippet.bundled.utils.Utils;
+import com.google.android.mobly.snippet.rpc.Rpc;
+import com.google.android.mobly.snippet.util.Log;
+import java.io.IOException;
+import java.net.InetAddress;
+import java.net.Socket;
+import java.net.UnknownHostException;
+import java.util.List;
+import java.util.Locale;
+
+/** Snippet class for networking RPCs. */
+public class NetworkingSnippet implements Snippet {
+
+    private final Context mContext;
+    private final DownloadManager mDownloadManager;
+    private volatile boolean mIsDownloadComplete = false;
+    private volatile long mReqid = 0;
+
+    public NetworkingSnippet() {
+        mContext = InstrumentationRegistry.getInstrumentation().getContext();
+        mDownloadManager = (DownloadManager) mContext.getSystemService(Context.DOWNLOAD_SERVICE);
+    }
+
+    private static class NetworkingSnippetException extends Exception {
+
+        private static final long serialVersionUID = 8080L;
+
+        public NetworkingSnippetException(String msg) {
+            super(msg);
+        }
+    }
+
+    @Rpc(description = "Check if a host and port are connectable using a TCP connection attempt.")
+    public boolean networkIsTcpConnectable(String host, int port) {
+        InetAddress addr;
+        try {
+            addr = InetAddress.getByName(host);
+        } catch (UnknownHostException uherr) {
+            Log.d("Host name lookup failure: " + uherr.getMessage());
+            return false;
+        }
+
+        try {
+            Socket sock = new Socket(addr, port);
+            sock.close();
+        } catch (IOException ioerr) {
+            Log.d("Did not make connection to host: " + ioerr.getMessage());
+            return false;
+        }
+        return true;
+    }
+
+    @Rpc(
+            description =
+                    "Download a file using HTTP. Return content Uri (file remains on device). "
+                            + "The Uri should be treated as an opaque handle for further operations.")
+    public String networkHttpDownload(String url)
+            throws IllegalArgumentException, NetworkingSnippetException {
+
+        Uri uri = Uri.parse(url);
+        List<String> pathsegments = uri.getPathSegments();
+        if (pathsegments.size() < 1) {
+            throw new IllegalArgumentException(
+                    String.format(Locale.US, "The Uri %s does not have a path.", uri.toString()));
+        }
+        DownloadManager.Request request = new DownloadManager.Request(uri);
+        request.setDestinationInExternalPublicDir(
+                Environment.DIRECTORY_DOWNLOADS, pathsegments.get(pathsegments.size() - 1));
+        mIsDownloadComplete = false;
+        mReqid = 0;
+        IntentFilter filter = new IntentFilter(DownloadManager.ACTION_DOWNLOAD_COMPLETE);
+        BroadcastReceiver receiver = new DownloadReceiver();
+        mContext.registerReceiver(receiver, filter);
+        try {
+            mReqid = mDownloadManager.enqueue(request);
+            Log.d(
+                    String.format(
+                            Locale.US,
+                            "networkHttpDownload download of %s with id %d",
+                            url,
+                            mReqid));
+            if (!Utils.waitUntil(() -> mIsDownloadComplete, 30)) {
+                Log.d(
+                        String.format(
+                                Locale.US, "networkHttpDownload timed out waiting for completion"));
+                throw new NetworkingSnippetException("networkHttpDownload timed out.");
+            }
+        } finally {
+            mContext.unregisterReceiver(receiver);
+        }
+        Uri resp = mDownloadManager.getUriForDownloadedFile(mReqid);
+        if (resp != null) {
+            Log.d(String.format(Locale.US, "networkHttpDownload completed to %s", resp.toString()));
+            mReqid = 0;
+            return resp.toString();
+        } else {
+            Log.d(
+                    String.format(
+                            Locale.US,
+                            "networkHttpDownload Failed to download %s",
+                            uri.toString()));
+            throw new NetworkingSnippetException("networkHttpDownload didn't get downloaded file.");
+        }
+    }
+
+    private class DownloadReceiver extends BroadcastReceiver {
+
+        @Override
+        public void onReceive(Context context, Intent intent) {
+            String action = intent.getAction();
+            long gotid = (long) intent.getExtras().get("extra_download_id");
+            if (DownloadManager.ACTION_DOWNLOAD_COMPLETE.equals(action) && gotid == mReqid) {
+                mIsDownloadComplete = true;
+            }
+        }
+    }
+
+    @Override
+    public void shutdown() {
+        if (mReqid != 0) {
+            mDownloadManager.remove(mReqid);
+        }
+    }
+}