View Javadoc
1   /*
2    * Copyright (c) 2002-2025 Gargoyle Software Inc.
3    *
4    * Licensed under the Apache License, Version 2.0 (the "License");
5    * you may not use this file except in compliance with the License.
6    * You may obtain a copy of the License at
7    * https://www.apache.org/licenses/LICENSE-2.0
8    *
9    * Unless required by applicable law or agreed to in writing, software
10   * distributed under the License is distributed on an "AS IS" BASIS,
11   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12   * See the License for the specific language governing permissions and
13   * limitations under the License.
14   */
15  package org.htmlunit.httpclient;
16  
17  import org.apache.commons.lang3.StringUtils;
18  import org.apache.http.cookie.CommonCookieAttributeHandler;
19  import org.apache.http.cookie.Cookie;
20  import org.apache.http.cookie.CookieOrigin;
21  import org.apache.http.cookie.MalformedCookieException;
22  import org.apache.http.cookie.SetCookie;
23  import org.apache.http.impl.cookie.BasicClientCookie;
24  
25  /**
26   * Customized CookieAttributeHandler for handling of the samesite attribute.
27   *
28   * @author Ronald Brill
29   */
30  final class HtmlUnitSameSiteHandler implements CommonCookieAttributeHandler {
31  
32      private static final String SAMESITE_ATTR = "samesite";
33  
34      @Override
35      public void validate(final Cookie cookie, final CookieOrigin origin) throws MalformedCookieException {
36          // nothing
37      }
38  
39      @Override
40      public void parse(final SetCookie cookie, final String value) throws MalformedCookieException {
41          ((BasicClientCookie) cookie).setAttribute(SAMESITE_ATTR, StringUtils.toRootLowerCase(value));
42      }
43  
44      @Override
45      public boolean match(final Cookie cookie, final CookieOrigin origin) {
46          return true;
47      }
48  
49      @Override
50      public String getAttributeName() {
51          return SAMESITE_ATTR;
52      }
53  }