package com.jcoglan.websocket;
import java.lang.Long; import java.io.IOException;
import org.jruby.Ruby; import org.jruby.RubyArray; import org.jruby.RubyClass; import org.jruby.RubyFixnum; import org.jruby.RubyModule; import org.jruby.RubyObject; 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(), new ObjectAllocator() { public IRubyObject allocate(Ruby runtime, RubyClass rubyClass) { return new WebsocketMask(runtime, rubyClass); } }); webSocketMask.defineAnnotatedMethods(WebsocketMask.class); return true; } 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() || ((RubyArray)mask).getLength() == 0) { return payload; } int n = ((RubyArray)payload).getLength(), i; long p, m; RubyArray unmasked = RubyArray.newArray(runtime, n); long[] maskArray = { (Long)((RubyArray)mask).get(0), (Long)((RubyArray)mask).get(1), (Long)((RubyArray)mask).get(2), (Long)((RubyArray)mask).get(3) }; for (i = 0; i < n; i++) { p = (Long)((RubyArray)payload).get(i); m = maskArray[i % 4]; unmasked.set(i, p ^ m); } return unmasked; } }
}