Ejemplo n.º 1
0
 @Override
 public boolean parseFirstLine(SelectionKey key) throws IOException {
   try {
     String firstLine = buffers[0].toString().replaceAll("\\s+", " ");
     String[] parts = firstLine.split(" ");
     if (parts.length != 3) {
       reportError(key, ErrorType.BAD_REQUEST, "Error in first line");
       return false;
     }
     String method = parts[0];
     String resource = parts[1];
     String version = parts[2];
     if (resource.startsWith("http://")) {
       resource = deleteHost(resource);
     }
     httpRequest =
         new HTTPRequest(
             ConverterHTTP.convertMethod(parts[0].toString()),
             parts[1].toString(),
             ConverterHTTP.convertVersion(ConverterHTTP.convertValue(parts[2].toString())));
     buffers[0] = new StringBuffer();
     state = States.READING_HEADER;
     BufferManager.addStringToBuffer(
         method + " " + resource + " " + version + '\r' + '\n', writeBufferWrapper);
   } catch (RuntimeException e) {
     LoggingManager.logError("Buffer reached max capacity");
     reportError(key, ErrorType.INTERNAL_SERVER_ERROR, "Internal proxy error");
     return false;
   }
   return true;
 }
Ejemplo n.º 2
0
 @Override
 public void finishedReading(SelectionKey key) {
   ((AttachmentHTTP) serverChannel.keyFor(key.selector()).attachment()).fullWrittenBuffer = true;
   PoolClientConnectionManager.getInstance().removeClient(key);
   Statistics.getInstance().addMethod(httpRequest.getMethod());
   if (httpRequest.getHost() != null) {
     LoggingManager.logReport("REQ - " + httpRequest.getMethod() + " " + httpRequest.getHost());
   } else {
     LoggingManager.logError("REQ - " + httpRequest.getMethod() + " " + "Unkown host");
   }
 }
Ejemplo n.º 3
0
  @Override
  public boolean parseHeader(SelectionKey key) throws IOException {
    try {
      String line = buffers[0].toString();
      if (line.equals("\n") || line.equals("\r\n")) {
        if (leetOn) {
          if (presentTextPlain || httpRequest.getResource().endsWith(".txt")) {
            presentTextPlain = true;
            BufferManager.addStringToBuffer("Accept-Encoding: identity\r\n", writeBufferWrapper);
          } else {
            if (acceptEncodingValue != null)
              BufferManager.addStringToBuffer(
                  "Accept-Encoding: " + acceptEncodingValue + "\r\n", writeBufferWrapper);
          }
        }
        if (httpRequest.getHost() == null) {
          reportError(key, ErrorType.NO_HOST_SPECIFIED, "No Host specified");
          return false;
        } else {
          buffers[0] = new StringBuffer();
          BufferManager.addStringToBuffer("\r\n", writeBufferWrapper);
          if (httpRequest.getMethod() == Method.POST) {
            buffers[1] = new StringBuffer();
            if (httpRequest.isChunkedBody()) {
              state = States.READING_BODY_CHUNKED_FIRST_LINE;
            } else if (httpRequest.isMultipart() && httpRequest.getBodyLength() == null) {
              state = States.READING_BODY_MULTIPART;
            } else {
              state = States.READING_BODY_NORMAL;
              leftLen = httpRequest.getBodyLength();
            }
          } else {
            state = States.IDLE;
            finishedReading(key);
          }
          return true;
        }
      } else {

        StringBuilder name = new StringBuilder();
        StringBuilder value = new StringBuilder();
        boolean foundColon = false;
        for (int i = 0; i < line.length(); i++) {
          char ch = line.charAt(i);
          if (ch != '\r' && ch != '\n') {
            if (ch == ':') {
              foundColon = true;
            } else {
              if (!foundColon) {
                name.append(ch);
              } else {
                value.append(ch);
              }
            }
          }
        }
        String val = value.toString().trim();
        if (leetOn) {
          if (name.toString().equalsIgnoreCase(Defs.ACCEPT_ENCODING)) {
            acceptEncodingValue = val;
          } else {
            if (name.toString().equalsIgnoreCase(Defs.ACCEPT)) {
              String[] split = val.split(",");
              for (String string : split) {
                String[] vecString = string.split(";");
                if (vecString[0].equalsIgnoreCase("text/plain")) {
                  presentTextPlain = true;
                }
              }
            }
            BufferManager.addStringToBuffer(name + ":" + val + "\r\n", writeBufferWrapper);
          }
        } else {
          BufferManager.addStringToBuffer(name + ":" + val + "\r\n", writeBufferWrapper);
        }

        httpRequest.addHeader(name.toString(), val);
        if (httpRequest.getHost() != null) {
          if (!ConnectionManager.isTryingToConnect(key)) {
            ConnectionManager.startConnectionWithServer(key);
          }
        }
        state = States.READING_HEADER;
        buffers[0] = new StringBuffer();
        return true;
      }
    } catch (RuntimeException e) {
      LoggingManager.logError("Buffer reached max capacity");
      reportError(key, ErrorType.BUFFER_OVERFLOW, "Internal Buffer Overflow");
      return false;
    }
  }