1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
/*
* Copyright (C) 2016 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.rpc;
import android.content.Context;
import com.google.android.mobly.snippet.manager.SnippetManager;
import com.google.android.mobly.snippet.util.Log;
import com.google.android.mobly.snippet.util.RpcUtil;
import java.io.BufferedReader;
import java.io.PrintWriter;
import java.net.Socket;
import java.util.LinkedHashSet;
import java.util.Map;
import java.util.Set;
import java.util.TreeMap;
import org.json.JSONArray;
import org.json.JSONException;
import org.json.JSONObject;
/** A JSON RPC server that forwards RPC calls to a specified receiver object. */
public class JsonRpcServer extends SimpleServer {
private static final String CMD_CLOSE_SESSION = "closeSl4aSession";
private static final String CMD_HELP = "help";
private final SnippetManager mSnippetManager;
private final RpcUtil mRpcUtil;
/** Construct a {@link JsonRpcServer} connected to the provided {@link SnippetManager}. */
public JsonRpcServer(Context context) {
mSnippetManager = SnippetManager.initSnippetManager(context);
mRpcUtil = new RpcUtil();
}
@Override
protected void handleRPCConnection(
Socket sock, Integer UID, BufferedReader reader, PrintWriter writer) throws Exception {
Log.d("UID " + UID);
String data;
while ((data = reader.readLine()) != null) {
Log.v("Session " + UID + " Received: " + data);
JSONObject request = new JSONObject(data);
int id = request.getInt("id");
String method = request.getString("method");
JSONArray params = request.getJSONArray("params");
// Handle builtin commands
if (method.equals(CMD_HELP)) {
help(writer, id, mSnippetManager, UID);
continue;
} else if (method.equals(CMD_CLOSE_SESSION)) {
Log.d("Got shutdown signal");
synchronized (writer) {
// Shut down all RPC receivers.
mSnippetManager.shutdown();
// Shut down this client connection. As soon as this happens, the client will
// kill us by triggering the 'stop' action from another instrumentation, so no
// other cleanup steps are guaranteed to execute.
send(writer, JsonRpcResult.empty(id), UID);
reader.close();
writer.close();
sock.close();
// Shut down this server.
shutdown();
}
return;
}
JSONObject returnValue = mRpcUtil.invokeRpc(method, params, id, UID);
send(writer, returnValue, UID);
}
}
private void help(PrintWriter writer, int id, SnippetManager receiverManager, Integer UID)
throws JSONException {
// Create a map from class simple name to the methods inside it.
Map<String, Set<MethodDescriptor>> methods = new TreeMap<>();
for (String method : receiverManager.getMethodNames()) {
MethodDescriptor descriptor = receiverManager.getMethodDescriptor(method);
String snippetClassName = descriptor.getSnippetClass().getSimpleName();
Set<MethodDescriptor> snippetClassMethods = methods.get(snippetClassName);
if (snippetClassMethods == null) {
// Preserve insertion order (alphabetical)
snippetClassMethods = new LinkedHashSet<>();
methods.put(snippetClassName, snippetClassMethods);
}
snippetClassMethods.add(descriptor);
}
StringBuilder result = new StringBuilder();
for (Map.Entry<String, Set<MethodDescriptor>> entry : methods.entrySet()) {
result.append("\nRPCs provided by ").append(entry.getKey()).append(":\n");
for (MethodDescriptor descriptor : entry.getValue()) {
result.append(" ").append(descriptor.getHelp()).append("\n");
}
}
send(writer, JsonRpcResult.result(id, result), UID);
}
private void send(PrintWriter writer, JSONObject result, int UID) {
writer.write(result + "\n");
writer.flush();
Log.v("Session " + UID + " Sent: " + result);
}
@Override
protected void handleConnection(Socket socket) throws Exception {}
}
|
