mirror of
https://github.com/google/nomulus.git
synced 2025-05-04 05:57:51 +02:00
422ec9b97a
This is in preparation for automatic EPP consolidation.
The assertion will now look like:
assertThatCommand("file.xml"[, substitution])
.atTime(time)
.hasResponse("file2.xml"[, substitution]);
Also, added convenience functions for often reused commands (login and logout)
-------------
Created by MOE: https://github.com/google/moe
MOE_MIGRATED_REVID=190617505
74 lines
2.5 KiB
Java
74 lines
2.5 KiB
Java
// Copyright 2017 The Nomulus Authors. All Rights Reserved.
|
|
//
|
|
// 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 google.registry.flows;
|
|
|
|
import com.google.common.collect.ImmutableMap;
|
|
import google.registry.testing.AppEngineRule;
|
|
import org.junit.Rule;
|
|
import org.junit.Test;
|
|
import org.junit.runner.RunWith;
|
|
import org.junit.runners.JUnit4;
|
|
|
|
/**
|
|
* Tests for XXE attacks.
|
|
*
|
|
*/
|
|
@RunWith(JUnit4.class)
|
|
public class EppXxeAttackTest extends EppTestCase {
|
|
|
|
@Rule
|
|
public final AppEngineRule appEngine = AppEngineRule.builder()
|
|
.withDatastore()
|
|
.build();
|
|
|
|
@Test
|
|
public void testRemoteXmlExternalEntity() throws Exception {
|
|
assertThatLoginSucceeds("NewRegistrar", "foo-BAR2");
|
|
assertThatCommand("contact_create_remote_xxe.xml")
|
|
.hasResponse(
|
|
"response_error_no_cltrid.xml",
|
|
ImmutableMap.of(
|
|
"CODE", "2001",
|
|
"MSG", "Syntax error at line 11, column 34: "
|
|
+ "The entity "remote" was referenced, but not declared."));
|
|
assertThatLogoutSucceeds();
|
|
}
|
|
|
|
@Test
|
|
public void testLocalXmlExtrernalEntity() throws Exception {
|
|
assertThatLoginSucceeds("NewRegistrar", "foo-BAR2");
|
|
assertThatCommand("contact_create_local_xxe.xml")
|
|
.hasResponse(
|
|
"response_error_no_cltrid.xml",
|
|
ImmutableMap.of(
|
|
"CODE", "2001",
|
|
"MSG", "Syntax error at line 11, column 31: "
|
|
+ "The entity "ent" was referenced, but not declared."));
|
|
assertThatLogoutSucceeds();
|
|
}
|
|
|
|
@Test
|
|
public void testBillionLaughsAttack() throws Exception {
|
|
assertThatLoginSucceeds("NewRegistrar", "foo-BAR2");
|
|
assertThatCommand("contact_create_billion_laughs.xml")
|
|
.hasResponse(
|
|
"response_error_no_cltrid.xml",
|
|
ImmutableMap.of(
|
|
"CODE", "2001",
|
|
"MSG", "Syntax error at line 20, column 32: "
|
|
+ "The entity "lol9" was referenced, but not declared."));
|
|
assertThatLogoutSucceeds();
|
|
}
|
|
}
|