swim.http.header.SecWebSocketKey Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of swim-http Show documentation
Uploads all artifacts belonging to configuration ':swim-http:archives'
There is a newer version: 3.10.0
// Copyright 2015-2019 SWIM.AI 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 swim.http.header;

import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.util.Arrays;
import java.util.concurrent.ThreadLocalRandom;
import swim.codec.Base64;
import swim.codec.Binary;
import swim.codec.Diagnostic;
import swim.codec.Input;
import swim.codec.Output;
import swim.codec.Parser;
import swim.codec.Unicode;
import swim.codec.Writer;
import swim.http.HttpException;
import swim.http.HttpHeader;
import swim.http.HttpParser;
import swim.http.HttpWriter;
import swim.util.Murmur3;

public final class SecWebSocketKey extends HttpHeader {
  final byte[] key;

  SecWebSocketKey(byte[] key) {
    this.key = key;
  }

  @Override
  public String lowerCaseName() {
    return "sec-websocket-key";
  }

  @Override
  public String name() {
    return "Sec-WebSocket-Key";
  }

  public byte[] key() {
    return this.key;
  }

  public SecWebSocketAccept accept() {
    Output output = Binary.byteArrayOutput(60);
    Base64.standard().writeByteArray(this.key, output);
    final String seed = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
    for (int i = 0, n = seed.length(); i < n; i += 1) {
      output = output.write((byte) seed.charAt(i));
    }
    try {
      final byte[] digest = MessageDigest.getInstance("SHA-1").digest(output.bind());
      return new SecWebSocketAccept(digest);
    } catch (NoSuchAlgorithmException cause) {
      throw new HttpException(cause);
    }
  }

  @Override
  public Writer writeHttpValue(Output output, HttpWriter http) {
    return Base64.standard().writeByteArray(this.key, output);
  }

  @Override
  public boolean equals(Object other) {
    if (this == other) {
      return true;
    } else if (other instanceof SecWebSocketKey) {
      final SecWebSocketKey that = (SecWebSocketKey) other;
      return Arrays.equals(this.key, that.key);
    }
    return false;
  }

  @Override
  public int hashCode() {
    if (hashSeed == 0) {
      hashSeed = Murmur3.seed(SecWebSocketKey.class);
    }
    return Murmur3.mash(Murmur3.mix(hashSeed, Arrays.hashCode(this.key)));
  }

  @Override
  public void debug(Output output) {
    output = output.write("SecWebSocketKey").write('.').write("from").write('(').write('"');
    Base64.standard().writeByteArray(this.key, output);
    output = output.write('"').write(')');
  }

  private static int hashSeed;

  public static SecWebSocketKey from(byte[] key) {
    return new SecWebSocketKey(key);
  }

  public static SecWebSocketKey from(String keyString) {
    final Input input = Unicode.stringInput(keyString);
    Parser parser = Base64.standard().parseByteArray(input);
    if (input.isCont() && !parser.isError()) {
      parser = Parser.error(Diagnostic.unexpected(input));
    }
    return new SecWebSocketKey(parser.bind());
  }

  public static SecWebSocketKey generate() {
    final byte[] key = new byte[16];
    ThreadLocalRandom.current().nextBytes(key);
    return new SecWebSocketKey(key);
  }

  public static Parser parseHttpValue(Input input, HttpParser http) {
    return SecWebSocketKeyParser.parse(input);
  }
}