@@ -106,45 +106,55 @@ public void parseMatrixVariablesString() {
106106 }
107107
108108 @ Test
109- public void isValidOriginSuccess () {
110-
109+ public void isValidOrigin () {
111110 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 ));
119113
120114 allowed = Collections .singletonList ("*" );
121- assertTrue (checkOrigin ("mydomain1.com" , -1 , "http://mydomain2.com" , allowed ));
115+ assertTrue (checkValidOrigin ("mydomain1.com" , -1 , "http://mydomain2.com" , allowed ));
122116
123117 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 ));
125120 }
126121
127122 @ 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+ }
129136
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 ));
134137
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 );
137147 }
138148
139- private boolean checkOrigin (String serverName , int port , String originHeader , List < String > allowed ) {
149+ private boolean checkSameOrigin (String serverName , int port , String originHeader ) {
140150 MockHttpServletRequest servletRequest = new MockHttpServletRequest ();
141151 ServerHttpRequest request = new ServletServerHttpRequest (servletRequest );
142152 servletRequest .setServerName (serverName );
143153 if (port != -1 ) {
144154 servletRequest .setServerPort (port );
145155 }
146156 request .getHeaders ().set (HttpHeaders .ORIGIN , originHeader );
147- return WebUtils .isValidOrigin (request , allowed );
157+ return WebUtils .isSameOrigin (request );
148158 }
149159
150160}
0 commit comments