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
|
package com.jcoglan.websocket;
import java.io.IOException;
import org.jruby.Ruby;
import org.jruby.RubyClass;
import org.jruby.RubyModule;
import org.jruby.RubyObject;
import org.jruby.RubyString;
import org.jruby.anno.JRubyMethod;
import org.jruby.runtime.ObjectAllocator;
import org.jruby.runtime.ThreadContext;
import org.jruby.runtime.builtin.IRubyObject;
import org.jruby.runtime.load.BasicLibraryService;
public class WebsocketMaskService implements BasicLibraryService {
private Ruby runtime;
public boolean basicLoad(Ruby runtime) throws IOException {
this.runtime = runtime;
RubyModule websocket = runtime.defineModule("WebSocket");
RubyClass webSocketMask = websocket.defineClassUnder("Mask", runtime.getObject(), getAllocator());
webSocketMask.defineAnnotatedMethods(WebsocketMask.class);
return true;
}
ObjectAllocator getAllocator() {
return new ObjectAllocator() {
public IRubyObject allocate(Ruby runtime, RubyClass rubyClass) {
return new WebsocketMask(runtime, rubyClass);
}
};
}
public class WebsocketMask extends RubyObject {
public WebsocketMask(final Ruby runtime, RubyClass rubyClass) {
super(runtime, rubyClass);
}
@JRubyMethod
public IRubyObject mask(ThreadContext context, IRubyObject payload, IRubyObject mask) {
if (mask.isNil()) return payload;
byte[] payload_a = ((RubyString)payload).getBytes();
byte[] mask_a = ((RubyString)mask).getBytes();
int i, n = payload_a.length;
if (n == 0) return payload;
for (i = 0; i < n; i++) {
payload_a[i] ^= mask_a[i % 4];
}
return RubyString.newStringNoCopy(runtime, payload_a);
}
}
}
|
