|
7 | 7 | import java.util.List; |
8 | 8 | import java.util.Map; |
9 | 9 | import java.util.Set; |
| 10 | +import jakarta.ws.rs.core.AbstractMultivaluedMap; |
10 | 11 | import jakarta.ws.rs.core.CacheControl; |
11 | 12 | import jakarta.ws.rs.core.Cookie; |
12 | 13 | import jakarta.ws.rs.core.EntityTag; |
|
15 | 16 | import jakarta.ws.rs.core.HttpHeaders; |
16 | 17 | import jakarta.ws.rs.core.Link; |
17 | 18 | import jakarta.ws.rs.core.MediaType; |
| 19 | +import jakarta.ws.rs.core.MultivaluedHashMap; |
18 | 20 | import jakarta.ws.rs.core.MultivaluedMap; |
19 | 21 | import jakarta.ws.rs.core.NewCookie; |
20 | 22 | import jakarta.ws.rs.core.PathSegment; |
@@ -149,6 +151,42 @@ void testMultivaluedMapputSingle(MultivaluedMap<String, String> mm1, Multivalued |
149 | 151 | sink(mm2.get("key").get(0)); // $hasValueFlow |
150 | 152 | } |
151 | 153 |
|
| 154 | + class MyAbstractMultivaluedMapJak<K, V> extends AbstractMultivaluedMap<K, V> { |
| 155 | + public MyAbstractMultivaluedMapJak(Map<K, List<V>> map) { |
| 156 | + super(map); |
| 157 | + } |
| 158 | + } |
| 159 | + |
| 160 | + void testAbstractMultivaluedMap(Map<String, List<String>> map1, Map<String, List<String>> map2, List<String> list) { |
| 161 | + map1.put(taint(), list); |
| 162 | + AbstractMultivaluedMap<String, String> amm1 = new MyAbstractMultivaluedMapJak<String, String>(map1); |
| 163 | + sink(amm1.keySet().iterator().next()); // $hasValueFlow |
| 164 | + |
| 165 | + list.add(taint()); |
| 166 | + map2.put("key", list); |
| 167 | + AbstractMultivaluedMap<String, String> amm2 = new MyAbstractMultivaluedMapJak<String, String>(map2); |
| 168 | + sink(amm2.get("key").get(0)); // $hasValueFlow |
| 169 | + } |
| 170 | + |
| 171 | + void testMultivaluedHashMap(Map<String, String> map1, Map<String, String> map2, |
| 172 | + MultivaluedMap<String, String> mm1, MultivaluedMap<String, String> mm2) { |
| 173 | + map1.put(taint(), "value"); |
| 174 | + MultivaluedHashMap<String, String> mhm1 = new MultivaluedHashMap<String, String>(map1); |
| 175 | + sink(mhm1.keySet().iterator().next()); // $hasValueFlow |
| 176 | + |
| 177 | + map2.put("key", taint()); |
| 178 | + MultivaluedHashMap<String, String> mhm2 = new MultivaluedHashMap<String, String>(map2); |
| 179 | + sink(mhm2.get("key").get(0)); // $hasValueFlow |
| 180 | + |
| 181 | + mm1.add(taint(), "value"); |
| 182 | + MultivaluedHashMap<String, String> mhm3 = new MultivaluedHashMap<String, String>(mm1); |
| 183 | + sink(mhm3.keySet().iterator().next()); // $hasValueFlow |
| 184 | + |
| 185 | + mm2.add("key", taint()); |
| 186 | + MultivaluedHashMap<String, String> mhm4 = new MultivaluedHashMap<String, String>(mm2); |
| 187 | + sink(mhm4.get("key").get(0)); // $hasValueFlow |
| 188 | + } |
| 189 | + |
152 | 190 | void testPathSegment(PathSegment ps1, PathSegment ps2) { |
153 | 191 | sink(taint(ps1).getMatrixParameters()); // $hasTaintFlow |
154 | 192 | sink(taint(ps2).getPath()); // $hasTaintFlow |
|
0 commit comments