Ejemplo n.º 1
0
 /** Triggered based on events from an {@link io.netty.handler.timeout.IdleStateHandler}. */
 @Override
 public void userEventTriggered(ChannelHandlerContext ctx, Object evt) throws Exception {
   if (evt instanceof IdleStateEvent) {
     IdleStateEvent e = (IdleStateEvent) evt;
     // See class comment for timeout semantics. In addition to ensuring we only timeout while
     // there are outstanding requests, we also do a secondary consistency check to ensure
     // there's no race between the idle timeout and incrementing the numOutstandingRequests
     // (see SPARK-7003).
     //
     // To avoid a race between TransportClientFactory.createClient() and this code which could
     // result in an inactive client being returned, this needs to run in a synchronized block.
     synchronized (this) {
       boolean isActuallyOverdue =
           System.nanoTime() - responseHandler.getTimeOfLastRequestNs() > requestTimeoutNs;
       if (e.state() == IdleState.ALL_IDLE && isActuallyOverdue) {
         if (responseHandler.numOutstandingRequests() > 0) {
           String address = NettyUtils.getRemoteAddress(ctx.channel());
           logger.error(
               "Connection to {} has been quiet for {} ms while there are outstanding "
                   + "requests. Assuming connection is dead; please adjust spark.network.timeout if this "
                   + "is wrong.",
               address,
               requestTimeoutNs / 1000 / 1000);
           client.timeOut();
           ctx.close();
         } else if (closeIdleConnections) {
           // While CloseIdleConnections is enable, we also close idle connection
           client.timeOut();
           ctx.close();
         }
       }
     }
   }
   ctx.fireUserEventTriggered(evt);
 }
Ejemplo n.º 2
0
 private static void processGoAwayWriteResult(
     final ChannelHandlerContext ctx,
     final int lastStreamId,
     final long errorCode,
     final ByteBuf debugData,
     ChannelFuture future) {
   try {
     if (future.isSuccess()) {
       if (errorCode != NO_ERROR.code()) {
         if (logger.isDebugEnabled()) {
           logger.debug(
               format(
                   "Sent GOAWAY: lastStreamId '%d', errorCode '%d', "
                       + "debugData '%s'. Forcing shutdown of the connection.",
                   lastStreamId, errorCode, debugData.toString(UTF_8)),
               future.cause());
         }
         ctx.close();
       }
     } else {
       if (logger.isErrorEnabled()) {
         logger.error(
             format(
                 "Sending GOAWAY failed: lastStreamId '%d', errorCode '%d', "
                     + "debugData '%s'. Forcing shutdown of the connection.",
                 lastStreamId, errorCode, debugData.toString(UTF_8)),
             future.cause());
       }
       ctx.close();
     }
   } finally {
     // We're done with the debug data now.
     debugData.release();
   }
 }
Ejemplo n.º 3
0
 @Override
 public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
   Channel incoming = ctx.channel();
   System.out.println("Client:" + incoming.remoteAddress() + "异常");
   cause.printStackTrace();
   ctx.close();
 }
  @Override
  protected void decode(ChannelHandlerContext ctx, ByteBuf in, List<Object> out) throws Exception {
    ChannelPipeline p = ctx.pipeline();
    SocksProtocolVersion version = SocksProtocolVersion.valueOf(in.readByte());
    System.out.println(version);
    in.resetReaderIndex();
    switch (version) {
      case SOCKS4a:
        p.addLast(new Socks4CmdRequestDecoder());
        p.addLast(Socks4MessageEncoder.INSTANCE);

        break;
      case SOCKS5:
        p.addLast(new Socks5InitRequestDecoder());
        p.addLast(Socks5MessageEncoder.INSTANCE);

        break;
      case UNKNOWN:
        in.clear();
        ctx.close();
        return;
    }
    p.addLast(SocksServerHandler.INSTANCE);
    p.remove(this);
  }
Ejemplo n.º 5
0
  @Override
  public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
    System.out.println("Channel Read");
    if (msg instanceof HttpResponse) {
      HttpResponse response = (HttpResponse) msg;

      System.err.println("STATUS: " + response.getStatus());
      System.err.println("VERSION: " + response.getProtocolVersion());
      System.err.println();

      if (!response.headers().isEmpty()) {
        for (String name : response.headers().names()) {
          for (String value : response.headers().getAll(name)) {
            System.err.println("HEADER: " + name + " = " + value);
          }
        }
        System.err.println();
      }
    }
    if (msg instanceof HttpContent) {
      HttpContent content = (HttpContent) msg;

      System.err.print(content.content().toString(CharsetUtil.UTF_8));
      System.err.flush();

      if (content instanceof LastHttpContent) {
        System.err.println("} END OF CONTENT");
        ctx.close();
      }
    }
  }
 @Override
 public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
   logger.debug("================进入客户端InBoundHandler channelRead============");
   // 如果是response
   if (msg instanceof HttpResponse) {
     HttpResponse response = (HttpResponse) msg;
     logger.debug(
         "客户端收到的响应CONTENT_TYPE:" + response.headers().get(HttpHeaders.Names.CONTENT_TYPE));
     if (HttpHeaders.isContentLengthSet(response)) {
       reader = new ByteBufToBytes((int) HttpHeaders.getContentLength(response));
     }
   }
   if (msg instanceof HttpContent) {
     HttpContent httpContent = (HttpContent) msg;
     ByteBuf content = httpContent.content();
     reader.reading(content);
     content.release();
     if (reader.isEnd()) {
       String resultStr = new String(reader.readFull());
       logger.debug("收到的服务端的消息是:" + resultStr);
       ctx.close();
     }
   }
   logger.debug("================出客户端InBoundHandler channelRead============");
 }
  @Override
  public void userEventTriggered(ChannelHandlerContext ctx, Object event) throws Exception {
    if (event.getClass() == QueueEvent.class) {
      // Initiate envelope processing, after the queue state has
      // changed from empty to non-empty.
      writeAndFlushNextEnvelopeIfPossible();
    } else if (event.getClass() == IdleStateEvent.class) {
      // Channel idle => try to close
      boolean closeConnection = false;

      // Only close the connection, if there are no queued envelopes. We have
      // to ensure that there is no race between closing the channel and
      // enqueuing a new envelope.
      synchronized (channel) {
        if (queuedEnvelopes.isEmpty() && !hasRequestedClose) {

          hasRequestedClose = true;

          closeConnection = true;

          // Notify the connection manager that this channel has been
          // closed.
          connectionManager.close(receiver);
        }
      }

      if (closeConnection) {
        ctx.close().addListener(new ChannelCloseListener());
      }
    } else {
      throw new IllegalStateException("Triggered unknown event.");
    }
  }
 @Override
 public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) {
   if (LOG.isTraceEnabled()) {
     LOG.trace("Channel disconnect caused close:{}", cause);
   }
   ctx.close();
 }
 public void messageReceived(ChannelHandlerContext ctx, DatagramPacket msg) throws Exception {
   String response = msg.content().toString(CharsetUtil.UTF_8);
   if (response.startsWith("QOTM: ")) {
     System.out.println("Quote of the Moment: " + response.substring(6));
     ctx.close();
   }
 }
 @Override
 public void exceptionCaught(ChannelHandlerContext remoteChannelCtx, Throwable cause)
     throws Exception {
   logger.error(
       cause.getMessage() + uaChannel.attr(ApnProxyConnectionAttribute.ATTRIBUTE_KEY), cause);
   remoteChannelCtx.close();
 }
Ejemplo n.º 11
0
 /**
  * 取代netty3.0中的channelDisconnected(),客户端主动关闭连接
  *
  * @param ctx
  * @throws Exception
  */
 @Override
 public void channelInactive(ChannelHandlerContext ctx) throws Exception {
   logger.info("[" + ctx.channel().remoteAddress() + "] : is inactive......");
   handlerDispatcher.removeChannelInfo(ctx.channel());
   System.out.println("[" + ctx.channel().remoteAddress() + "] : is inactive......");
   ctx.close();
 }
Ejemplo n.º 12
0
 @Override
 public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) {
   if (HttpServices.shouldLogException(cause)) {
     logger.warn(cause.getMessage(), cause);
   }
   ctx.close();
 }
Ejemplo n.º 13
0
 @Override
 public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
   logger.error("[" + ctx.channel().remoteAddress() + "] 出异常……");
   //		cause.printStackTrace();
   handlerDispatcher.removeChannelInfo(ctx.channel());
   ctx.close();
 }
Ejemplo n.º 14
0
  @Override
  public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
    if (ctx.channel().isActive()) {
      if (cause instanceof ReadTimeoutException) {
        ProxyServer.getInstance().getLogger().log(Level.WARNING, "{0} - read timed out", handler);
      } else if (cause instanceof BadPacketException) {
        ProxyServer.getInstance()
            .getLogger()
            .log(Level.WARNING, "{0} - bad packet ID, are mods in use!?", handler);
      } else if (cause instanceof IOException) {
        ProxyServer.getInstance()
            .getLogger()
            .log(
                Level.WARNING,
                "{0} - IOException: {1}",
                new Object[] {handler, cause.getMessage()});
      } else {
        ProxyServer.getInstance()
            .getLogger()
            .log(Level.SEVERE, handler + " - encountered exception", cause);
      }

      if (handler != null) {
        try {
          handler.exception(cause);
        } catch (Exception ex) {
          ProxyServer.getInstance()
              .getLogger()
              .log(Level.SEVERE, handler + " - exception processing exception", ex);
        }
      }

      ctx.close();
    }
  }
 @Override
 public void channelReadComplete(ChannelHandlerContext ctx) {
   if (responseAssertProcessor != null) {
     this.responseAssertProcessor.process(responseContext, outgoingMessage);
   }
   ctx.close();
 }
Ejemplo n.º 16
0
  @Override
  public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
    if (msg instanceof EchoFile) {
      EchoFile ef = (EchoFile) msg;
      int SumCountPackage = ef.getSumCountPackage();
      int countPackage = ef.getCountPackage();
      byte[] bytes = ef.getBytes();
      String md5 = ef.getFile_md5(); // 文件名

      String path = file_dir + File.separator + md5;
      File file = new File(path);
      RandomAccessFile randomAccessFile = new RandomAccessFile(file, "rw");
      randomAccessFile.seek(countPackage * dataLength - dataLength);
      randomAccessFile.write(bytes);
      LOGGER.debug("总包数:" + ef.getSumCountPackage());
      LOGGER.debug("收到第" + countPackage + "包");
      LOGGER.debug("本包字节数:" + bytes.length);
      countPackage = countPackage + 1;

      if (countPackage <= SumCountPackage) {
        ef.setCountPackage(countPackage);
        ctx.writeAndFlush(ef);
        randomAccessFile.close();
      } else {
        randomAccessFile.close();
        ctx.close();
      }
    }
  }
Ejemplo n.º 17
0
 /** Is called when a write timeout was detected */
 protected void writeTimedOut(ChannelHandlerContext ctx) throws Exception {
   if (!closed) {
     ctx.fireExceptionCaught(WriteTimeoutException.INSTANCE);
     ctx.close();
     closed = true;
   }
 }
 @Override
 public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
   if (cause instanceof TooLongFrameException) {
     ctx.close();
   } else {
     super.exceptionCaught(ctx, cause);
   }
 }
 // 暴力停止客户端,先执行exceptionCaught,接着执行channelInactive。
 @Override
 public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) { // (7)
   Channel incoming = ctx.channel();
   System.out.println("SimpleChatClient:" + incoming.remoteAddress() + "异常");
   // 当出现异常就关闭连接
   cause.printStackTrace();
   ctx.close();
 }
Ejemplo n.º 20
0
 @Override
 public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
   logger.warn(
       "Exception in connection from " + NettyUtils.getRemoteAddress(ctx.channel()), cause);
   requestHandler.exceptionCaught(cause);
   responseHandler.exceptionCaught(cause);
   ctx.close();
 }
 public void channelRead(ChannelHandlerContext ctx, Object msg) {
     if (msg instanceof HttpRequest)
         try {
             handleHttpRequest(ctx, (HttpRequest) msg);
         } catch (Exception e) {
             e.printStackTrace();
         }
     ctx.close();
 }
Ejemplo n.º 22
0
 @Override
 public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
   if (cause instanceof ConnectException) {
     startTime = -1;
     println("Failed to connect: " + cause.getMessage());
   }
   cause.printStackTrace();
   ctx.close();
 }
Ejemplo n.º 23
0
    @Override
    protected final void die(Throwable cause) {
      super.die(cause);
      if (ctx.channel().isOpen()) ctx.close();

      // Ensure to release server references
      userActor = null;
      ctx = null;
    }
Ejemplo n.º 24
0
 @Override
 public void exceptionCaught(ChannelHandlerContext context, Throwable cause) {
   if (cause instanceof IOException) {
     log.debug("IOException inside channel handling pipeline.", cause);
   } else {
     log.error("non-IOException inside channel handling pipeline.", cause);
   }
   context.close();
 }
 @Override
 public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
   System.out.println("client channelRead..");
   // 服务端返回消息后
   Result result = (Result) ctx.channel().attr(RESULT).get();
   result.set(msg);
   ctx.close();
   System.out.println("Now is :" + msg);
 }
Ejemplo n.º 26
0
  private void safeClose(
      final ChannelHandlerContext ctx, ChannelFuture flushFuture, final ChannelPromise promise) {
    if (!ctx.channel().isActive()) {
      ctx.close(promise);
      return;
    }

    final ScheduledFuture<?> timeoutFuture;
    if (closeNotifyTimeoutMillis > 0) {
      // Force-close the connection if close_notify is not fully sent in time.
      timeoutFuture =
          ctx.executor()
              .schedule(
                  new Runnable() {
                    @Override
                    public void run() {
                      logger.warn(
                          "{} Last write attempt timed out; force-closing the connection.",
                          ctx.channel());

                      // We notify the promise in the TryNotifyListener as there is a "race" where
                      // the close(...) call
                      // by the timeoutFuture and the close call in the flushFuture listener will be
                      // called. Because of
                      // this we need to use trySuccess() and tryFailure(...) as otherwise we can
                      // cause an
                      // IllegalStateException.
                      ctx.close(ctx.newPromise()).addListener(new ChannelPromiseNotifier(promise));
                    }
                  },
                  closeNotifyTimeoutMillis,
                  TimeUnit.MILLISECONDS);
    } else {
      timeoutFuture = null;
    }

    // Close the connection if close_notify is sent in time.
    flushFuture.addListener(
        new ChannelFutureListener() {
          @Override
          public void operationComplete(ChannelFuture f) throws Exception {
            if (timeoutFuture != null) {
              timeoutFuture.cancel(false);
            }
            // Trigger the close in all cases to make sure the promise is notified
            // See https://github.com/netty/netty/issues/2358
            //
            // We notify the promise in the ChannelPromiseNotifier as there is a "race" where the
            // close(...) call
            // by the timeoutFuture and the close call in the flushFuture listener will be called.
            // Because of
            // this we need to use trySuccess() and tryFailure(...) as otherwise we can cause an
            // IllegalStateException.
            ctx.close(ctx.newPromise()).addListener(new ChannelPromiseNotifier(promise));
          }
        });
  }
Ejemplo n.º 27
0
 public synchronized void close() {
   try {
     LOG.info("Closing connection");
     ctx.close().sync();
     connected = false;
   } catch (Exception e) {
     LOG.error("Closing failed", e);
   }
 }
Ejemplo n.º 28
0
 @Override
 public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
   if (!handshakeFuture.isDone()) {
     handshakeFuture.setFailure(cause);
   }
   for (WebSocketEventListener l : listensers) {
     l.onError(ctx, cause, response);
   }
   ctx.close();
 }
 void stopServer() {
   if (messageServerThread != null) {
     messageServerThread.stopServer();
     messageServerThread = null;
     listenerQueue.onDestory();
   }
   if (AuthChannelHandlerContext != null) {
     AuthChannelHandlerContext.close();
     AuthChannelHandlerContext = null;
   }
 }
 protected void decode(ChannelHandlerContext context, ByteBuf buffer) throws Exception {
   ChannelPipeline pipeline = context.pipeline();
   if (detectSsl && SslHandler.isEncrypted(buffer)) {
     SSLEngine engine = SSL_SERVER_CONTEXT.getValue().createSSLEngine();
     engine.setUseClientMode(false);
     pipeline.addLast(
         new SslHandler(engine),
         new ChunkedWriteHandler(),
         new PortUnificationServerHandler(delegatingHttpRequestHandler, false, detectGzip));
   } else {
     int magic1 = buffer.getUnsignedByte(buffer.readerIndex());
     int magic2 = buffer.getUnsignedByte(buffer.readerIndex() + 1);
     if (detectGzip && magic1 == 31 && magic2 == 139) {
       pipeline.addLast(
           new JZlibEncoder(ZlibWrapper.GZIP),
           new JdkZlibDecoder(ZlibWrapper.GZIP),
           new PortUnificationServerHandler(delegatingHttpRequestHandler, detectSsl, false));
     } else if (isHttp(magic1, magic2)) {
       NettyUtil.initHttpHandlers(pipeline);
       pipeline.addLast(delegatingHttpRequestHandler);
       if (BuiltInServer.LOG.isDebugEnabled()) {
         pipeline.addLast(
             new ChannelOutboundHandlerAdapter() {
               @Override
               public void write(
                   ChannelHandlerContext context, Object message, ChannelPromise promise)
                   throws Exception {
                 if (message instanceof HttpResponse) {
                   //                BuiltInServer.LOG.debug("OUT HTTP:\n" + message);
                   HttpResponse response = (HttpResponse) message;
                   BuiltInServer.LOG.debug(
                       "OUT HTTP: "
                           + response.getStatus().code()
                           + " "
                           + response.headers().get("Content-type"));
                 }
                 super.write(context, message, promise);
               }
             });
       }
     } else if (magic1 == 'C' && magic2 == 'H') {
       buffer.skipBytes(2);
       pipeline.addLast(new CustomHandlerDelegator());
     } else {
       BuiltInServer.LOG.warn("unknown request, first two bytes " + magic1 + " " + magic2);
       context.close();
     }
   }
   // must be after new channels handlers addition (netty bug?)
   ensureThatExceptionHandlerIsLast(pipeline);
   pipeline.remove(this);
   context.fireChannelRead(buffer);
 }