| | @@ -70,12 +70,113 @@ |
| 70 | 70 | // 実行する
|
| 71 | 71 | return (HttpResponse)method.invoke(cls.newInstance(),req);
|
| 72 | 72 | }
|
| 73 | | - private static void executeWebSokcet(HttpRequest req, String PackageName, OutputStream ops, InputStream ips) throws Exception {
|
| 73 | + private static void executeWebSokcet(HttpRequest req, String PackageName, OutputStream ops, InputStream ips) throws Exception {
|
| 74 | + String host = req.getHeader().get("Host");
|
| 75 | + StringBuffer sb = new StringBuffer();
|
| 76 | +
|
| 77 | + String path = req.getRowPath();
|
| 78 | + String key1 = req.getHeader().get("Sec-WebSocket-Key1");
|
| 79 | + String key2 = req.getHeader().get("Sec-WebSocket-Key2");
|
| 80 | + String origin = req.getHeader().get("Origin");
|
| 81 | + String res_ws = "ws://" + host + path;
|
| 82 | + byte[] body_code = getChalengeKey(req.headerBytes);
|
| 83 | + long key1_num = chooseNum(key1);
|
| 84 | + long key2_num = chooseNum(key2);
|
| 85 | + int key1_s = chooseSpace(key1);
|
| 86 | + int key2_s = chooseSpace(key2);
|
| 87 | + long key1_w = key1_num / key1_s;
|
| 88 | + long key2_w = key2_num / key2_s;
|
| 89 | + byte[] key_head = bigendiean(key1_w, key2_w);
|
| 90 | + byte[] byte_keys = new byte[16];
|
| 91 | + System.arraycopy(key_head, 0, byte_keys, 0, 8);
|
| 92 | + System.arraycopy(body_code, 0, byte_keys, 8, 8);
|
| 93 | +
|
| 94 | + MessageDigest md = MessageDigest.getInstance("MD5");
|
| 95 | + md.update(byte_keys);
|
| 96 | + byte[] hash = md.digest();
|
| 97 | +
|
| 98 | + sb.append("HTTP/1.1 101 WebSocket Protocol Handshake\r\n");
|
| 99 | + sb.append("Upgrade: WebSocket\r\n");
|
| 100 | + sb.append("Connection: Upgrade\r\n");
|
| 101 | + sb.append("Sec-WebSocket-Location: " + res_ws + "\r\n");
|
| 102 | + sb.append("Sec-WebSocket-Origin: http://" + host + "\r\n");
|
| 103 | + sb.append("\r\n");
|
| 104 | + ops.write(sb.toString().getBytes());
|
| 105 | + ops.write(hash);
|
| 106 | +
|
| 107 | +
|
| 74 | 108 | Class<Action> cls = null;
|
| 75 | 109 |
|
| 76 | 110 | cls = (Class<Action>)Class.forName(PackageName);
|
| 77 | | - Method method = cls.getMethod("execute", HttpRequest.class);
|
| 111 | + Method method = cls.getMethod("execute", HttpRequest.class, OutputStream.class, InputStream.class);
|
| 78 | 112 | // 実行する
|
| 79 | | - method.invoke(cls.newInstance(),req);
|
| 113 | + method.invoke(cls.newInstance(), req, ops, ips);
|
| 80 | 114 | }
|
| 115 | + public static byte[] getChalengeKey(byte[] header) {
|
| 116 | + byte[] ret = null;
|
| 117 | + try {
|
| 118 | + for(int i = 0;i < header.length;i++) {
|
| 119 | + if(header[i] == '\r' && header[i + 1] == '\n') {
|
| 120 | + if(header[i + 2] == '\r' && header[i + 3] == '\n') {
|
| 121 | + int lastAt = header.length - (i + 4);
|
| 122 | + ret = new byte[lastAt];
|
| 123 | + System.arraycopy(header, i + 4, ret, 0, lastAt);
|
| 124 | + break;
|
| 125 | + }
|
| 126 | + }
|
| 127 | + }
|
| 128 | + } catch(Exception e) {
|
| 129 | + e.printStackTrace();
|
| 130 | + }
|
| 131 | + return ret;
|
| 132 | + }
|
| 133 | + public static long chooseNum(String str) {
|
| 134 | + String code = "";
|
| 135 | + for(int i = 0;i < str.length();i++) {
|
| 136 | + try {
|
| 137 | + int num = Integer.parseInt("" + str.charAt(i));
|
| 138 | + code += num;
|
| 139 | + } catch(Exception e) {}
|
| 140 | + }
|
| 141 | + long ret = 0;
|
| 142 | + try {
|
| 143 | + ret = Long.parseLong(code);
|
| 144 | + } catch(Exception e) {}
|
| 145 | + return ret;
|
| 146 | + }
|
| 147 | + public static int chooseSpace(String str) {
|
| 148 | + int ret = 0;
|
| 149 | + for(int i =0;i < str.length();i++) {
|
| 150 | + if(str.charAt(i) == ' ') {
|
| 151 | + ret++;
|
| 152 | + }
|
| 153 | + }
|
| 154 | + return ret;
|
| 155 | + }
|
| 156 | + public static byte[] bigendiean(long key1_w, long key2_w) {
|
| 157 | + long lon1 = key1_w;
|
| 158 | + long lon2 = key2_w;
|
| 159 | + byte[] bytes = new byte[8];
|
| 160 | + int[] ints = new int[8];
|
| 161 | + ints[3] = (int)(lon1 % 256);
|
| 162 | + lon1 /= 256;
|
| 163 | + ints[2] = (int)(lon1 % 256);
|
| 164 | + lon1 /= 256;
|
| 165 | + ints[1] = (int)(lon1 % 256);
|
| 166 | + lon1 /= 256;
|
| 167 | + ints[0] = (int)lon1;
|
| 168 | +
|
| 169 | + ints[7] = (int)(lon2 % 256);
|
| 170 | + lon2 /= 256;
|
| 171 | + ints[6] = (int)(lon2 % 256);
|
| 172 | + lon2 /= 256;
|
| 173 | + ints[5] = (int)(lon2 % 256);
|
| 174 | + lon2 /= 256;
|
| 175 | + ints[4] = (int)lon2;
|
| 176 | +
|
| 177 | + for(int i = 0;i < 8;i++) {
|
| 178 | + bytes[i] = (byte)ints[i];
|
| 179 | + }
|
| 180 | + return bytes;
|
| 181 | + }
|
| 81 | 182 | }
|
branch/Basic 0.2x/startup_action/startup_action/StartTest.java
branch/Basic 0.2x/startup_action/startup_action