@@ -106,45 +106,55 @@ public void parseMatrixVariablesString() {
106
106
}
107
107
108
108
@ Test
109
- public void isValidOriginSuccess () {
110
-
109
+ public void isValidOrigin () {
111
110
List <String > allowed = Collections .emptyList ();
112
- assertTrue (checkOrigin ("mydomain1.com" , -1 , "http://mydomain1.com" , allowed ));
113
- assertTrue (checkOrigin ("mydomain1.com" , -1 , "http://mydomain1.com:80" , allowed ));
114
- assertTrue (checkOrigin ("mydomain1.com" , 443 , "https://mydomain1.com" , allowed ));
115
- assertTrue (checkOrigin ("mydomain1.com" , 443 , "https://mydomain1.com:443" , allowed ));
116
- assertTrue (checkOrigin ("mydomain1.com" , 123 , "http://mydomain1.com:123" , allowed ));
117
- assertTrue (checkOrigin ("mydomain1.com" , -1 , "ws://mydomain1.com" , allowed ));
118
- assertTrue (checkOrigin ("mydomain1.com" , 443 , "wss://mydomain1.com" , allowed ));
111
+ assertTrue (checkValidOrigin ("mydomain1.com" , -1 , "http://mydomain1.com" , allowed ));
112
+ assertFalse (checkValidOrigin ("mydomain1.com" , -1 , "http://mydomain2.com" , allowed ));
119
113
120
114
allowed = Collections .singletonList ("*" );
121
- assertTrue (checkOrigin ("mydomain1.com" , -1 , "http://mydomain2.com" , allowed ));
115
+ assertTrue (checkValidOrigin ("mydomain1.com" , -1 , "http://mydomain2.com" , allowed ));
122
116
123
117
allowed = Collections .singletonList ("http://mydomain1.com" );
124
- assertTrue (checkOrigin ("mydomain2.com" , -1 , "http://mydomain1.com" , allowed ));
118
+ assertTrue (checkValidOrigin ("mydomain2.com" , -1 , "http://mydomain1.com" , allowed ));
119
+ assertFalse (checkValidOrigin ("mydomain2.com" , -1 , "http://mydomain3.com" , allowed ));
125
120
}
126
121
127
122
@ Test
128
- public void isValidOriginFailure () {
123
+ public void isSameOrigin () {
124
+ assertTrue (checkSameOrigin ("mydomain1.com" , -1 , "http://mydomain1.com" ));
125
+ assertTrue (checkSameOrigin ("mydomain1.com" , -1 , "http://mydomain1.com:80" ));
126
+ assertTrue (checkSameOrigin ("mydomain1.com" , 443 , "https://mydomain1.com" ));
127
+ assertTrue (checkSameOrigin ("mydomain1.com" , 443 , "https://mydomain1.com:443" ));
128
+ assertTrue (checkSameOrigin ("mydomain1.com" , 123 , "http://mydomain1.com:123" ));
129
+ assertTrue (checkSameOrigin ("mydomain1.com" , -1 , "ws://mydomain1.com" ));
130
+ assertTrue (checkSameOrigin ("mydomain1.com" , 443 , "wss://mydomain1.com" ));
131
+
132
+ assertFalse (checkSameOrigin ("mydomain1.com" , -1 , "http://mydomain2.com" ));
133
+ assertFalse (checkSameOrigin ("mydomain1.com" , -1 , "https://mydomain1.com" ));
134
+ assertFalse (checkSameOrigin ("mydomain1.com" , -1 , "invalid-origin" ));
135
+ }
129
136
130
- List <String > allowed = Collections .emptyList ();
131
- assertFalse (checkOrigin ("mydomain1.com" , -1 , "http://mydomain2.com" , allowed ));
132
- assertFalse (checkOrigin ("mydomain1.com" , -1 , "https://mydomain1.com" , allowed ));
133
- assertFalse (checkOrigin ("mydomain1.com" , -1 , "invalid-origin" , allowed ));
134
137
135
- allowed = Collections .singletonList ("http://mydomain1.com" );
136
- assertFalse (checkOrigin ("mydomain2.com" , -1 , "http://mydomain3.com" , allowed ));
138
+ private boolean checkValidOrigin (String serverName , int port , String originHeader , List <String > allowed ) {
139
+ MockHttpServletRequest servletRequest = new MockHttpServletRequest ();
140
+ ServerHttpRequest request = new ServletServerHttpRequest (servletRequest );
141
+ servletRequest .setServerName (serverName );
142
+ if (port != -1 ) {
143
+ servletRequest .setServerPort (port );
144
+ }
145
+ request .getHeaders ().set (HttpHeaders .ORIGIN , originHeader );
146
+ return WebUtils .isValidOrigin (request , allowed );
137
147
}
138
148
139
- private boolean checkOrigin (String serverName , int port , String originHeader , List < String > allowed ) {
149
+ private boolean checkSameOrigin (String serverName , int port , String originHeader ) {
140
150
MockHttpServletRequest servletRequest = new MockHttpServletRequest ();
141
151
ServerHttpRequest request = new ServletServerHttpRequest (servletRequest );
142
152
servletRequest .setServerName (serverName );
143
153
if (port != -1 ) {
144
154
servletRequest .setServerPort (port );
145
155
}
146
156
request .getHeaders ().set (HttpHeaders .ORIGIN , originHeader );
147
- return WebUtils .isValidOrigin (request , allowed );
157
+ return WebUtils .isSameOrigin (request );
148
158
}
149
159
150
160
}
0 commit comments