tests.py 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685
  1. # -*- coding: utf-8 -*-
  2. from __future__ import unicode_literals
  3. import logging
  4. import re
  5. import warnings
  6. from django.conf import settings
  7. from django.http import HttpRequest, HttpResponse
  8. from django.middleware.csrf import (
  9. CSRF_TOKEN_LENGTH, CsrfViewMiddleware,
  10. _compare_salted_tokens as equivalent_tokens, get_token,
  11. )
  12. from django.template import RequestContext, Template
  13. from django.template.context_processors import csrf
  14. from django.test import SimpleTestCase, override_settings
  15. from django.utils.encoding import force_bytes
  16. from django.utils.six import text_type
  17. from django.views.decorators.csrf import (
  18. csrf_exempt, ensure_csrf_cookie, requires_csrf_token,
  19. )
  20. # Response/views used for CsrfResponseMiddleware and CsrfViewMiddleware tests
  21. def post_form_response():
  22. resp = HttpResponse(content="""
  23. <html><body><h1>\u00a1Unicode!<form method="post"><input type="text" /></form></body></html>
  24. """, mimetype="text/html")
  25. return resp
  26. def post_form_view(request):
  27. """A view that returns a POST form (without a token)"""
  28. return post_form_response()
  29. # Response/views used for template tag tests
  30. def token_view(request):
  31. """A view that uses {% csrf_token %}"""
  32. context = RequestContext(request, processors=[csrf])
  33. template = Template("{% csrf_token %}")
  34. return HttpResponse(template.render(context))
  35. def non_token_view_using_request_processor(request):
  36. """
  37. A view that doesn't use the token, but does use the csrf view processor.
  38. """
  39. context = RequestContext(request, processors=[csrf])
  40. template = Template("")
  41. return HttpResponse(template.render(context))
  42. class TestingHttpRequest(HttpRequest):
  43. """
  44. A version of HttpRequest that allows us to change some things
  45. more easily
  46. """
  47. def is_secure(self):
  48. return getattr(self, '_is_secure_override', False)
  49. class CsrfViewMiddlewareTest(SimpleTestCase):
  50. _csrf_id = _csrf_id_cookie = '1bcdefghij2bcdefghij3bcdefghij4bcdefghij5bcdefghij6bcdefghijABCD'
  51. def _get_GET_no_csrf_cookie_request(self):
  52. return TestingHttpRequest()
  53. def _get_GET_csrf_cookie_request(self):
  54. req = TestingHttpRequest()
  55. req.COOKIES[settings.CSRF_COOKIE_NAME] = self._csrf_id_cookie
  56. return req
  57. def _get_POST_csrf_cookie_request(self):
  58. req = self._get_GET_csrf_cookie_request()
  59. req.method = "POST"
  60. return req
  61. def _get_POST_no_csrf_cookie_request(self):
  62. req = self._get_GET_no_csrf_cookie_request()
  63. req.method = "POST"
  64. return req
  65. def _get_POST_request_with_token(self):
  66. req = self._get_POST_csrf_cookie_request()
  67. req.POST['csrfmiddlewaretoken'] = self._csrf_id
  68. return req
  69. def _get_POST_bare_secret_csrf_cookie_request(self):
  70. req = self._get_POST_no_csrf_cookie_request()
  71. req.COOKIES[settings.CSRF_COOKIE_NAME] = self._csrf_id_cookie[:32]
  72. return req
  73. def _get_POST_bare_secret_csrf_cookie_request_with_token(self):
  74. req = self._get_POST_bare_secret_csrf_cookie_request()
  75. req.POST['csrfmiddlewaretoken'] = self._csrf_id_cookie[:32]
  76. return req
  77. def _check_token_present(self, response, csrf_id=None):
  78. text = text_type(response.content, response.charset)
  79. match = re.search("name='csrfmiddlewaretoken' value='(.*?)'", text)
  80. csrf_token = csrf_id or self._csrf_id
  81. self.assertTrue(
  82. match and equivalent_tokens(csrf_token, match.group(1)),
  83. "Could not find csrfmiddlewaretoken to match %s" % csrf_token
  84. )
  85. def test_process_view_token_too_long(self):
  86. """
  87. If the token is longer than expected, it is ignored and a new token is
  88. created.
  89. """
  90. req = self._get_GET_no_csrf_cookie_request()
  91. req.COOKIES[settings.CSRF_COOKIE_NAME] = 'x' * 100000
  92. CsrfViewMiddleware().process_view(req, token_view, (), {})
  93. resp = token_view(req)
  94. resp2 = CsrfViewMiddleware().process_response(req, resp)
  95. csrf_cookie = resp2.cookies.get(settings.CSRF_COOKIE_NAME, False)
  96. self.assertEqual(len(csrf_cookie.value), CSRF_TOKEN_LENGTH)
  97. def test_process_view_token_invalid_chars(self):
  98. """
  99. If the token contains non-alphanumeric characters, it is ignored and a
  100. new token is created.
  101. """
  102. token = ('!@#' + self._csrf_id)[:CSRF_TOKEN_LENGTH]
  103. req = self._get_GET_no_csrf_cookie_request()
  104. req.COOKIES[settings.CSRF_COOKIE_NAME] = token
  105. CsrfViewMiddleware().process_view(req, token_view, (), {})
  106. resp = token_view(req)
  107. resp2 = CsrfViewMiddleware().process_response(req, resp)
  108. csrf_cookie = resp2.cookies.get(settings.CSRF_COOKIE_NAME, False)
  109. self.assertEqual(len(csrf_cookie.value), CSRF_TOKEN_LENGTH)
  110. self.assertNotEqual(csrf_cookie.value, token)
  111. def test_process_view_token_invalid_bytes(self):
  112. """
  113. If the token contains improperly encoded unicode, it is ignored and a
  114. new token is created.
  115. """
  116. token = (b"<1>\xc2\xa1" + force_bytes(self._csrf_id, 'ascii'))[:CSRF_TOKEN_LENGTH]
  117. req = self._get_GET_no_csrf_cookie_request()
  118. req.COOKIES[settings.CSRF_COOKIE_NAME] = token
  119. # We expect a UnicodeWarning here, because we used broken utf-8 on purpose
  120. with warnings.catch_warnings():
  121. warnings.filterwarnings("ignore", category=UnicodeWarning)
  122. CsrfViewMiddleware().process_view(req, token_view, (), {})
  123. resp = token_view(req)
  124. resp2 = CsrfViewMiddleware().process_response(req, resp)
  125. csrf_cookie = resp2.cookies.get(settings.CSRF_COOKIE_NAME, False)
  126. self.assertEqual(len(csrf_cookie.value), CSRF_TOKEN_LENGTH)
  127. self.assertNotEqual(csrf_cookie.value, token)
  128. def test_process_response_get_token_used(self):
  129. """
  130. When get_token is used, check that the cookie is created and headers
  131. patched.
  132. """
  133. req = self._get_GET_no_csrf_cookie_request()
  134. # Put tests for CSRF_COOKIE_* settings here
  135. with self.settings(CSRF_COOKIE_NAME='myname',
  136. CSRF_COOKIE_DOMAIN='.example.com',
  137. CSRF_COOKIE_PATH='/test/',
  138. CSRF_COOKIE_SECURE=True,
  139. CSRF_COOKIE_HTTPONLY=True):
  140. # token_view calls get_token() indirectly
  141. CsrfViewMiddleware().process_view(req, token_view, (), {})
  142. resp = token_view(req)
  143. resp2 = CsrfViewMiddleware().process_response(req, resp)
  144. csrf_cookie = resp2.cookies.get('myname', False)
  145. self.assertNotEqual(csrf_cookie, False)
  146. self.assertEqual(csrf_cookie['domain'], '.example.com')
  147. self.assertEqual(csrf_cookie['secure'], True)
  148. self.assertEqual(csrf_cookie['httponly'], True)
  149. self.assertEqual(csrf_cookie['path'], '/test/')
  150. self.assertIn('Cookie', resp2.get('Vary', ''))
  151. def test_process_response_get_token_not_used(self):
  152. """
  153. Check that if get_token() is not called, the view middleware does not
  154. add a cookie.
  155. """
  156. # This is important to make pages cacheable. Pages which do call
  157. # get_token(), assuming they use the token, are not cacheable because
  158. # the token is specific to the user
  159. req = self._get_GET_no_csrf_cookie_request()
  160. # non_token_view_using_request_processor does not call get_token(), but
  161. # does use the csrf request processor. By using this, we are testing
  162. # that the view processor is properly lazy and doesn't call get_token()
  163. # until needed.
  164. CsrfViewMiddleware().process_view(req, non_token_view_using_request_processor, (), {})
  165. resp = non_token_view_using_request_processor(req)
  166. resp2 = CsrfViewMiddleware().process_response(req, resp)
  167. csrf_cookie = resp2.cookies.get(settings.CSRF_COOKIE_NAME, False)
  168. self.assertEqual(csrf_cookie, False)
  169. # Check the request processing
  170. def test_process_request_no_csrf_cookie(self):
  171. """
  172. Check that if no CSRF cookies is present, the middleware rejects the
  173. incoming request. This will stop login CSRF.
  174. """
  175. req = self._get_POST_no_csrf_cookie_request()
  176. req2 = CsrfViewMiddleware().process_view(req, post_form_view, (), {})
  177. self.assertEqual(403, req2.status_code)
  178. def test_process_request_csrf_cookie_no_token(self):
  179. """
  180. Check that if a CSRF cookie is present but no token, the middleware
  181. rejects the incoming request.
  182. """
  183. req = self._get_POST_csrf_cookie_request()
  184. req2 = CsrfViewMiddleware().process_view(req, post_form_view, (), {})
  185. self.assertEqual(403, req2.status_code)
  186. def test_process_request_csrf_cookie_and_token(self):
  187. """
  188. Check that if both a cookie and a token is present, the middleware lets it through.
  189. """
  190. req = self._get_POST_request_with_token()
  191. req2 = CsrfViewMiddleware().process_view(req, post_form_view, (), {})
  192. self.assertIsNone(req2)
  193. def test_process_request_csrf_cookie_no_token_exempt_view(self):
  194. """
  195. Check that if a CSRF cookie is present and no token, but the csrf_exempt
  196. decorator has been applied to the view, the middleware lets it through
  197. """
  198. req = self._get_POST_csrf_cookie_request()
  199. req2 = CsrfViewMiddleware().process_view(req, csrf_exempt(post_form_view), (), {})
  200. self.assertIsNone(req2)
  201. def test_csrf_token_in_header(self):
  202. """
  203. Check that we can pass in the token in a header instead of in the form
  204. """
  205. req = self._get_POST_csrf_cookie_request()
  206. req.META['HTTP_X_CSRFTOKEN'] = self._csrf_id
  207. req2 = CsrfViewMiddleware().process_view(req, post_form_view, (), {})
  208. self.assertIsNone(req2)
  209. @override_settings(CSRF_HEADER_NAME='HTTP_X_CSRFTOKEN_CUSTOMIZED')
  210. def test_csrf_token_in_header_with_customized_name(self):
  211. """
  212. settings.CSRF_HEADER_NAME can be used to customize the CSRF header name
  213. """
  214. req = self._get_POST_csrf_cookie_request()
  215. req.META['HTTP_X_CSRFTOKEN_CUSTOMIZED'] = self._csrf_id
  216. req2 = CsrfViewMiddleware().process_view(req, post_form_view, (), {})
  217. self.assertIsNone(req2)
  218. def test_put_and_delete_rejected(self):
  219. """
  220. Tests that HTTP PUT and DELETE methods have protection
  221. """
  222. req = TestingHttpRequest()
  223. req.method = 'PUT'
  224. req2 = CsrfViewMiddleware().process_view(req, post_form_view, (), {})
  225. self.assertEqual(403, req2.status_code)
  226. req = TestingHttpRequest()
  227. req.method = 'DELETE'
  228. req2 = CsrfViewMiddleware().process_view(req, post_form_view, (), {})
  229. self.assertEqual(403, req2.status_code)
  230. def test_put_and_delete_allowed(self):
  231. """
  232. Tests that HTTP PUT and DELETE methods can get through with
  233. X-CSRFToken and a cookie
  234. """
  235. req = self._get_GET_csrf_cookie_request()
  236. req.method = 'PUT'
  237. req.META['HTTP_X_CSRFTOKEN'] = self._csrf_id
  238. req2 = CsrfViewMiddleware().process_view(req, post_form_view, (), {})
  239. self.assertIsNone(req2)
  240. req = self._get_GET_csrf_cookie_request()
  241. req.method = 'DELETE'
  242. req.META['HTTP_X_CSRFTOKEN'] = self._csrf_id
  243. req2 = CsrfViewMiddleware().process_view(req, post_form_view, (), {})
  244. self.assertIsNone(req2)
  245. # Tests for the template tag method
  246. def test_token_node_no_csrf_cookie(self):
  247. """
  248. Check that CsrfTokenNode works when no CSRF cookie is set
  249. """
  250. req = self._get_GET_no_csrf_cookie_request()
  251. resp = token_view(req)
  252. token = get_token(req)
  253. self.assertIsNotNone(token)
  254. self._check_token_present(resp, token)
  255. def test_token_node_empty_csrf_cookie(self):
  256. """
  257. Check that we get a new token if the csrf_cookie is the empty string
  258. """
  259. req = self._get_GET_no_csrf_cookie_request()
  260. req.COOKIES[settings.CSRF_COOKIE_NAME] = b""
  261. CsrfViewMiddleware().process_view(req, token_view, (), {})
  262. resp = token_view(req)
  263. token = get_token(req)
  264. self.assertIsNotNone(token)
  265. self._check_token_present(resp, token)
  266. def test_token_node_with_csrf_cookie(self):
  267. """
  268. Check that CsrfTokenNode works when a CSRF cookie is set
  269. """
  270. req = self._get_GET_csrf_cookie_request()
  271. CsrfViewMiddleware().process_view(req, token_view, (), {})
  272. resp = token_view(req)
  273. self._check_token_present(resp)
  274. def test_get_token_for_exempt_view(self):
  275. """
  276. Check that get_token still works for a view decorated with 'csrf_exempt'.
  277. """
  278. req = self._get_GET_csrf_cookie_request()
  279. CsrfViewMiddleware().process_view(req, csrf_exempt(token_view), (), {})
  280. resp = token_view(req)
  281. self._check_token_present(resp)
  282. def test_get_token_for_requires_csrf_token_view(self):
  283. """
  284. Check that get_token works for a view decorated solely with requires_csrf_token
  285. """
  286. req = self._get_GET_csrf_cookie_request()
  287. resp = requires_csrf_token(token_view)(req)
  288. self._check_token_present(resp)
  289. def test_token_node_with_new_csrf_cookie(self):
  290. """
  291. Check that CsrfTokenNode works when a CSRF cookie is created by
  292. the middleware (when one was not already present)
  293. """
  294. req = self._get_GET_no_csrf_cookie_request()
  295. CsrfViewMiddleware().process_view(req, token_view, (), {})
  296. resp = token_view(req)
  297. resp2 = CsrfViewMiddleware().process_response(req, resp)
  298. csrf_cookie = resp2.cookies[settings.CSRF_COOKIE_NAME]
  299. self._check_token_present(resp, csrf_id=csrf_cookie.value)
  300. def test_cookie_not_reset_on_accepted_request(self):
  301. """
  302. The csrf token used in posts is changed on every request (although
  303. stays equivalent). The csrf cookie should not change on accepted
  304. requests. If it appears in the response, it should keep its value.
  305. """
  306. req = self._get_POST_request_with_token()
  307. CsrfViewMiddleware().process_view(req, token_view, (), {})
  308. resp = token_view(req)
  309. resp = CsrfViewMiddleware().process_response(req, resp)
  310. csrf_cookie = resp.cookies.get(settings.CSRF_COOKIE_NAME, None)
  311. if csrf_cookie:
  312. self.assertEqual(
  313. csrf_cookie.value, self._csrf_id_cookie,
  314. "CSRF cookie was changed on an accepted request"
  315. )
  316. def test_bare_secret_accepted_and_replaced(self):
  317. """
  318. The csrf token is reset from a bare secret.
  319. """
  320. req = self._get_POST_bare_secret_csrf_cookie_request_with_token()
  321. req2 = CsrfViewMiddleware().process_view(req, token_view, (), {})
  322. self.assertIsNone(req2)
  323. resp = token_view(req)
  324. resp = CsrfViewMiddleware().process_response(req, resp)
  325. self.assertIn(settings.CSRF_COOKIE_NAME, resp.cookies, "Cookie was not reset from bare secret")
  326. csrf_cookie = resp.cookies[settings.CSRF_COOKIE_NAME]
  327. self.assertEqual(len(csrf_cookie.value), CSRF_TOKEN_LENGTH)
  328. self._check_token_present(resp, csrf_id=csrf_cookie.value)
  329. @override_settings(DEBUG=True)
  330. def test_https_bad_referer(self):
  331. """
  332. Test that a POST HTTPS request with a bad referer is rejected
  333. """
  334. req = self._get_POST_request_with_token()
  335. req._is_secure_override = True
  336. req.META['HTTP_HOST'] = 'www.example.com'
  337. req.META['HTTP_REFERER'] = 'https://www.evil.org/somepage'
  338. req.META['SERVER_PORT'] = '443'
  339. response = CsrfViewMiddleware().process_view(req, post_form_view, (), {})
  340. self.assertContains(
  341. response,
  342. 'Referer checking failed - https://www.evil.org/somepage does not '
  343. 'match any trusted origins.',
  344. status_code=403,
  345. )
  346. @override_settings(DEBUG=True)
  347. def test_https_malformed_referer(self):
  348. """
  349. A POST HTTPS request with a bad referer is rejected.
  350. """
  351. malformed_referer_msg = 'Referer checking failed - Referer is malformed.'
  352. req = self._get_POST_request_with_token()
  353. req._is_secure_override = True
  354. req.META['HTTP_REFERER'] = 'http://http://www.example.com/'
  355. response = CsrfViewMiddleware().process_view(req, post_form_view, (), {})
  356. self.assertContains(
  357. response,
  358. 'Referer checking failed - Referer is insecure while host is secure.',
  359. status_code=403,
  360. )
  361. # Empty
  362. req.META['HTTP_REFERER'] = ''
  363. response = CsrfViewMiddleware().process_view(req, post_form_view, (), {})
  364. self.assertContains(response, malformed_referer_msg, status_code=403)
  365. # Non-ASCII
  366. req.META['HTTP_REFERER'] = b'\xd8B\xf6I\xdf'
  367. response = CsrfViewMiddleware().process_view(req, post_form_view, (), {})
  368. self.assertContains(response, malformed_referer_msg, status_code=403)
  369. # missing scheme
  370. # >>> urlparse('//example.com/')
  371. # ParseResult(scheme='', netloc='example.com', path='/', params='', query='', fragment='')
  372. req.META['HTTP_REFERER'] = '//example.com/'
  373. response = CsrfViewMiddleware().process_view(req, post_form_view, (), {})
  374. self.assertContains(response, malformed_referer_msg, status_code=403)
  375. # missing netloc
  376. # >>> urlparse('https://')
  377. # ParseResult(scheme='https', netloc='', path='', params='', query='', fragment='')
  378. req.META['HTTP_REFERER'] = 'https://'
  379. response = CsrfViewMiddleware().process_view(req, post_form_view, (), {})
  380. self.assertContains(response, malformed_referer_msg, status_code=403)
  381. @override_settings(ALLOWED_HOSTS=['www.example.com'])
  382. def test_https_good_referer(self):
  383. """
  384. A POST HTTPS request with a good referer is accepted.
  385. """
  386. req = self._get_POST_request_with_token()
  387. req._is_secure_override = True
  388. req.META['HTTP_HOST'] = 'www.example.com'
  389. req.META['HTTP_REFERER'] = 'https://www.example.com/somepage'
  390. req2 = CsrfViewMiddleware().process_view(req, post_form_view, (), {})
  391. self.assertIsNone(req2)
  392. @override_settings(ALLOWED_HOSTS=['www.example.com'])
  393. def test_https_good_referer_2(self):
  394. """
  395. A POST HTTPS request with a good referer is accepted where the referer
  396. contains no trailing slash.
  397. """
  398. # See ticket #15617
  399. req = self._get_POST_request_with_token()
  400. req._is_secure_override = True
  401. req.META['HTTP_HOST'] = 'www.example.com'
  402. req.META['HTTP_REFERER'] = 'https://www.example.com'
  403. req2 = CsrfViewMiddleware().process_view(req, post_form_view, (), {})
  404. self.assertIsNone(req2)
  405. @override_settings(ALLOWED_HOSTS=['www.example.com'], CSRF_COOKIE_DOMAIN='.example.com', USE_X_FORWARDED_PORT=True)
  406. def test_https_good_referer_behind_proxy(self):
  407. """
  408. A POST HTTPS request is accepted when USE_X_FORWARDED_PORT=True.
  409. """
  410. req = self._get_POST_request_with_token()
  411. req._is_secure_override = True
  412. req.META.update({
  413. 'HTTP_HOST': '10.0.0.2',
  414. 'HTTP_REFERER': 'https://www.example.com/somepage',
  415. 'SERVER_PORT': '8080',
  416. 'HTTP_X_FORWARDED_HOST': 'www.example.com',
  417. 'HTTP_X_FORWARDED_PORT': '443',
  418. })
  419. req2 = CsrfViewMiddleware().process_view(req, post_form_view, (), {})
  420. self.assertIsNone(req2)
  421. @override_settings(ALLOWED_HOSTS=['www.example.com'], CSRF_TRUSTED_ORIGINS=['dashboard.example.com'])
  422. def test_https_csrf_trusted_origin_allowed(self):
  423. """
  424. A POST HTTPS request with a referer added to the CSRF_TRUSTED_ORIGINS
  425. setting is accepted.
  426. """
  427. req = self._get_POST_request_with_token()
  428. req._is_secure_override = True
  429. req.META['HTTP_HOST'] = 'www.example.com'
  430. req.META['HTTP_REFERER'] = 'https://dashboard.example.com'
  431. req2 = CsrfViewMiddleware().process_view(req, post_form_view, (), {})
  432. self.assertIsNone(req2)
  433. @override_settings(ALLOWED_HOSTS=['www.example.com'], CSRF_TRUSTED_ORIGINS=['.example.com'])
  434. def test_https_csrf_wildcard_trusted_origin_allowed(self):
  435. """
  436. A POST HTTPS request with a referer that matches a CSRF_TRUSTED_ORIGINS
  437. wildcard is accepted.
  438. """
  439. req = self._get_POST_request_with_token()
  440. req._is_secure_override = True
  441. req.META['HTTP_HOST'] = 'www.example.com'
  442. req.META['HTTP_REFERER'] = 'https://dashboard.example.com'
  443. response = CsrfViewMiddleware().process_view(req, post_form_view, (), {})
  444. self.assertIsNone(response)
  445. @override_settings(ALLOWED_HOSTS=['www.example.com'], CSRF_COOKIE_DOMAIN='.example.com')
  446. def test_https_good_referer_matches_cookie_domain(self):
  447. """
  448. A POST HTTPS request with a good referer should be accepted from a
  449. subdomain that's allowed by CSRF_COOKIE_DOMAIN.
  450. """
  451. req = self._get_POST_request_with_token()
  452. req._is_secure_override = True
  453. req.META['HTTP_REFERER'] = 'https://foo.example.com/'
  454. req.META['SERVER_PORT'] = '443'
  455. response = CsrfViewMiddleware().process_view(req, post_form_view, (), {})
  456. self.assertIsNone(response)
  457. @override_settings(ALLOWED_HOSTS=['www.example.com'], CSRF_COOKIE_DOMAIN='.example.com')
  458. def test_https_good_referer_matches_cookie_domain_with_different_port(self):
  459. """
  460. A POST HTTPS request with a good referer should be accepted from a
  461. subdomain that's allowed by CSRF_COOKIE_DOMAIN and a non-443 port.
  462. """
  463. req = self._get_POST_request_with_token()
  464. req._is_secure_override = True
  465. req.META['HTTP_HOST'] = 'www.example.com'
  466. req.META['HTTP_REFERER'] = 'https://foo.example.com:4443/'
  467. req.META['SERVER_PORT'] = '4443'
  468. response = CsrfViewMiddleware().process_view(req, post_form_view, (), {})
  469. self.assertIsNone(response)
  470. @override_settings(CSRF_COOKIE_DOMAIN='.example.com', DEBUG=True)
  471. def test_https_reject_insecure_referer(self):
  472. """
  473. A POST HTTPS request from an insecure referer should be rejected.
  474. """
  475. req = self._get_POST_request_with_token()
  476. req._is_secure_override = True
  477. req.META['HTTP_REFERER'] = 'http://example.com/'
  478. req.META['SERVER_PORT'] = '443'
  479. response = CsrfViewMiddleware().process_view(req, post_form_view, (), {})
  480. self.assertContains(
  481. response,
  482. 'Referer checking failed - Referer is insecure while host is secure.',
  483. status_code=403,
  484. )
  485. def test_ensures_csrf_cookie_no_middleware(self):
  486. """
  487. The ensure_csrf_cookie() decorator works without middleware.
  488. """
  489. @ensure_csrf_cookie
  490. def view(request):
  491. # Doesn't insert a token or anything
  492. return HttpResponse(content="")
  493. req = self._get_GET_no_csrf_cookie_request()
  494. resp = view(req)
  495. self.assertTrue(resp.cookies.get(settings.CSRF_COOKIE_NAME, False))
  496. self.assertIn('Cookie', resp.get('Vary', ''))
  497. def test_ensures_csrf_cookie_with_middleware(self):
  498. """
  499. The ensure_csrf_cookie() decorator works with the CsrfViewMiddleware
  500. enabled.
  501. """
  502. @ensure_csrf_cookie
  503. def view(request):
  504. # Doesn't insert a token or anything
  505. return HttpResponse(content="")
  506. req = self._get_GET_no_csrf_cookie_request()
  507. CsrfViewMiddleware().process_view(req, view, (), {})
  508. resp = view(req)
  509. resp2 = CsrfViewMiddleware().process_response(req, resp)
  510. self.assertTrue(resp2.cookies.get(settings.CSRF_COOKIE_NAME, False))
  511. self.assertIn('Cookie', resp2.get('Vary', ''))
  512. def test_ensures_csrf_cookie_no_logging(self):
  513. """
  514. ensure_csrf_cookie() doesn't log warnings (#19436).
  515. """
  516. @ensure_csrf_cookie
  517. def view(request):
  518. # Doesn't insert a token or anything
  519. return HttpResponse(content="")
  520. class TestHandler(logging.Handler):
  521. def emit(self, record):
  522. raise Exception("This shouldn't have happened!")
  523. logger = logging.getLogger('django.request')
  524. test_handler = TestHandler()
  525. old_log_level = logger.level
  526. try:
  527. logger.addHandler(test_handler)
  528. logger.setLevel(logging.WARNING)
  529. req = self._get_GET_no_csrf_cookie_request()
  530. view(req)
  531. finally:
  532. logger.removeHandler(test_handler)
  533. logger.setLevel(old_log_level)
  534. def test_csrf_cookie_age(self):
  535. """
  536. CSRF cookie age can be set using settings.CSRF_COOKIE_AGE.
  537. """
  538. req = self._get_GET_no_csrf_cookie_request()
  539. MAX_AGE = 123
  540. with self.settings(CSRF_COOKIE_NAME='csrfcookie',
  541. CSRF_COOKIE_DOMAIN='.example.com',
  542. CSRF_COOKIE_AGE=MAX_AGE,
  543. CSRF_COOKIE_PATH='/test/',
  544. CSRF_COOKIE_SECURE=True,
  545. CSRF_COOKIE_HTTPONLY=True):
  546. # token_view calls get_token() indirectly
  547. CsrfViewMiddleware().process_view(req, token_view, (), {})
  548. resp = token_view(req)
  549. resp2 = CsrfViewMiddleware().process_response(req, resp)
  550. max_age = resp2.cookies.get('csrfcookie').get('max-age')
  551. self.assertEqual(max_age, MAX_AGE)
  552. def test_csrf_cookie_age_none(self):
  553. """
  554. CSRF cookie age does not have max age set and therefore uses
  555. session-based cookies.
  556. """
  557. req = self._get_GET_no_csrf_cookie_request()
  558. MAX_AGE = None
  559. with self.settings(CSRF_COOKIE_NAME='csrfcookie',
  560. CSRF_COOKIE_DOMAIN='.example.com',
  561. CSRF_COOKIE_AGE=MAX_AGE,
  562. CSRF_COOKIE_PATH='/test/',
  563. CSRF_COOKIE_SECURE=True,
  564. CSRF_COOKIE_HTTPONLY=True):
  565. # token_view calls get_token() indirectly
  566. CsrfViewMiddleware().process_view(req, token_view, (), {})
  567. resp = token_view(req)
  568. resp2 = CsrfViewMiddleware().process_response(req, resp)
  569. max_age = resp2.cookies.get('csrfcookie').get('max-age')
  570. self.assertEqual(max_age, '')
  571. def test_post_data_read_failure(self):
  572. """
  573. #20128 -- IOErrors during POST data reading should be caught and
  574. treated as if the POST data wasn't there.
  575. """
  576. class CsrfPostRequest(HttpRequest):
  577. """
  578. HttpRequest that can raise an IOError when accessing POST data
  579. """
  580. def __init__(self, token, raise_error):
  581. super(CsrfPostRequest, self).__init__()
  582. self.method = 'POST'
  583. self.raise_error = False
  584. self.COOKIES[settings.CSRF_COOKIE_NAME] = token
  585. self.POST['csrfmiddlewaretoken'] = token
  586. self.raise_error = raise_error
  587. def _load_post_and_files(self):
  588. raise IOError('error reading input data')
  589. def _get_post(self):
  590. if self.raise_error:
  591. self._load_post_and_files()
  592. return self._post
  593. def _set_post(self, post):
  594. self._post = post
  595. POST = property(_get_post, _set_post)
  596. token = ('ABC' + self._csrf_id)[:CSRF_TOKEN_LENGTH]
  597. req = CsrfPostRequest(token, raise_error=False)
  598. resp = CsrfViewMiddleware().process_view(req, post_form_view, (), {})
  599. self.assertIsNone(resp)
  600. req = CsrfPostRequest(token, raise_error=True)
  601. resp = CsrfViewMiddleware().process_view(req, post_form_view, (), {})
  602. self.assertEqual(resp.status_code, 403)