ceph/src/jaegertracing/thrift/lib/java/test/org/apache/thrift/protocol/TestTCompactProtocol.java

   1 /*
   2  * Licensed to the Apache Software Foundation (ASF) under one
   3  * or more contributor license agreements. See the NOTICE file
   4  * distributed with this work for additional information
   5  * regarding copyright ownership. The ASF licenses this file
   6  * to you under the Apache License, Version 2.0 (the
   7  * "License"); you may not use this file except in compliance
   8  * with the License. You may obtain a copy of the License at
   9  *
  10  *   http://www.apache.org/licenses/LICENSE-2.0
  11  *
  12  * Unless required by applicable law or agreed to in writing,
  13  * software distributed under the License is distributed on an
  14  * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  15  * KIND, either express or implied. See the License for the
  16  * specific language governing permissions and limitations
  17  * under the License.
  18  */
  19
  20
  21 package org.apache.thrift.protocol;
  22
  23 import org.apache.thrift.TDeserializer;
  24 import org.apache.thrift.TException;
  25
  26 import thrift.test.Bonk;
  27
  28 public class TestTCompactProtocol extends ProtocolTestBase {
  29   @Override
  30   protected TProtocolFactory getFactory() {
  31     return new TCompactProtocol.Factory();
  32   }
  33
  34   @Override
  35   protected boolean canBeUsedNaked() {
  36     return true;
  37   }
  38
  39   public void testOOMDenialOfService() throws Exception {
  40     // Struct header, Integer.MAX_VALUE length, and only one real
  41     // byte of data
  42     byte [] bytes = {24, -1, -1, -1, -17, 49};
  43     TDeserializer deser = new TDeserializer(new TCompactProtocol
  44                                             .Factory(1000));
  45     Bonk bonk = new Bonk();
  46     try {
  47         deser.deserialize(bonk, bytes);
  48     } catch (TException e) {
  49         // Ignore as we are only checking for OOM in the failure case
  50     }
  51   }
  52
  53   public static void main(String args[]) throws Exception {
  54     new TestTCompactProtocol().benchmark();
  55   }
  56 }