CSRF protection concept in Flask - Mini Project: Build & Apply

CSRF Protection Concept in Flask

📖 Scenario: You are building a simple Flask web app that has a form where users can submit their favorite color. To keep the app safe from CSRF attacks, you will add CSRF protection step-by-step.

🎯 Goal: Build a Flask app with a form that uses CSRF protection to prevent unauthorized form submissions.

📋 What You'll Learn

Create a Flask app with a route for the form

Add a secret key configuration for CSRF

Use Flask-WTF to add CSRF protection to the form

Render the form with CSRF token included

💡 Why This Matters

🌍 Real World

CSRF protection is essential in web apps to stop attackers from tricking users into submitting unwanted requests.

💼 Career

Understanding CSRF and how to protect against it is a key skill for web developers working with Flask or any web framework.

1

Create a basic Flask app with a form route

Create a Flask app by importing Flask and instantiate it as app. Then create a route /color that returns a simple HTML form with a POST method and an input named color.

Flask

from flask import Flask, request

app = Flask(__name__)

# Create a route '/color' that returns a form with method POST and input named 'color'
# Your code here

Hint

Start by importing Flask and creating the app instance. Then define a route '/color' that accepts GET and POST methods. Return a simple HTML form with an input named 'color'.

2

Add a secret key configuration for CSRF

Add a secret key to the Flask app by setting app.config['SECRET_KEY'] to the string 'mysecretkey'. This key is needed for CSRF protection.

Flask

from flask import Flask, request

app = Flask(__name__)
app.config['SECRET_KEY'] = 'mysecretkey'

# Your code here

Hint

Set the secret key on the app config to enable secure sessions and CSRF protection.

3

Use Flask-WTF to add CSRF protection to the form

Import FlaskForm and StringField from flask_wtf and wtforms. Create a form class called ColorForm that inherits from FlaskForm and has a color field of type StringField. Then import CSRFProtect from flask_wtf and initialize it with the Flask app.

Flask

from flask import Flask, request
from flask_wtf import FlaskForm, CSRFProtect
from wtforms import StringField

app = Flask(__name__)
app.config['SECRET_KEY'] = 'mysecretkey'

csrf = CSRFProtect(app)

class ColorForm(FlaskForm):
    color = StringField('color')

# Your code here

Hint

Import FlaskForm and CSRFProtect. Create a ColorForm class with a color field. Initialize CSRFProtect with the app.

4

Render the form with CSRF token included

Modify the color route to create an instance of ColorForm. Render the form in the response using form.csrf_token inside the form HTML. Use form.color.label and form.color() to render the label and input. Return the form HTML as a string.

Flask

from flask import Flask, request
from flask_wtf import FlaskForm, CSRFProtect
from wtforms import StringField

app = Flask(__name__)
app.config['SECRET_KEY'] = 'mysecretkey'

csrf = CSRFProtect(app)

class ColorForm(FlaskForm):
    color = StringField('color')

@app.route('/color', methods=['GET', 'POST'])
def color():
    form = ColorForm()
    # Render the form with CSRF token and color field label and input
    # Your code here

Hint

Create a ColorForm instance. Use f-string to include form.csrf_token, form.color.label, and form.color() inside the form HTML.

Practice

1. What is the main purpose of CSRF protection in a Flask web application?

easy

A. To speed up the loading time of web pages

B. To encrypt user passwords in the database

C. To prevent attackers from making users perform unwanted actions without their consent

D. To improve the visual design of forms

Solution

1. Step 1: Understand what CSRF stands for

   CSRF means Cross-Site Request Forgery, which tricks users into submitting unwanted requests.

2. Step 2: Identify the purpose of CSRF protection

   CSRF protection stops attackers from forcing users to perform actions they did not intend.

3. Final Answer:

   To prevent attackers from making users perform unwanted actions without their consent -> Option C

4. Quick Check:

   CSRF protection = prevent unwanted user actions [OK]

Hint: CSRF stops fake actions from attackers using your login [OK]

Common Mistakes:

• Confusing CSRF with password encryption
• Thinking CSRF improves page speed
• Believing CSRF changes form appearance

2. Which of the following is the correct way to enable CSRF protection in a Flask app using Flask-WTF?

easy

A. from flask_wtf import CSRFProtect
app = Flask(__name__)
csrf = CSRFProtect(app)
csrf.init_app(app)

B. from flask_wtf import CSRFProtect
csrf = CSRFProtect()
app = Flask(__name__)

C. app = Flask(__name__)
app.csrf = True

D. from flask_wtf import CSRFProtect
csrf = CSRFProtect(app)

Solution

1. Step 1: Recall Flask-WTF CSRF setup

   You import CSRFProtect and pass the Flask app instance to it to enable protection.

2. Step 2: Check the options for correct syntax

   from flask_wtf import CSRFProtect csrf = CSRFProtect(app) correctly imports CSRFProtect and initializes it with the app in one step.

3. Final Answer:

   from flask_wtf import CSRFProtect csrf = CSRFProtect(app) -> Option D

4. Quick Check:

   CSRFProtect(app) initializes protection [OK]

Hint: Initialize CSRFProtect with app instance directly [OK]

Common Mistakes:

• Calling CSRFProtect without passing app
• Setting app.csrf = True (not valid)
• Calling init_app after passing app (redundant)

3. Given this Flask route with CSRF protection enabled, what happens if the form is submitted without a CSRF token?

@app.route('/submit', methods=['POST'])
def submit():
    return 'Success'

medium

A. The server raises a CSRF error and rejects the request

B. The server ignores the CSRF token and processes normally

C. The request is accepted and returns 'Success'

D. The server redirects to the home page without error

Solution

1. Step 1: Understand CSRF token role in POST requests

   CSRF tokens verify that the form submission is from a trusted source.

2. Step 2: Check behavior when token is missing

   Without a valid CSRF token, Flask-WTF raises an error and rejects the request.

3. Final Answer:

   The server raises a CSRF error and rejects the request -> Option A

4. Quick Check:

   Missing CSRF token = error rejection [OK]

Hint: POST without token triggers CSRF error [OK]

Common Mistakes:

• Assuming request is accepted without token
• Thinking server silently ignores missing token
• Believing server redirects instead of error

4. You added CSRF protection to your Flask app but get a 'CSRF token missing' error on form submission. Which fix will solve this issue?

medium

A. Add {{ form.csrf_token }} inside the HTML form

B. Remove the CSRFProtect initialization from the app

C. Change the form method to GET instead of POST

D. Disable CSRF protection in Flask-WTF config

Solution

1. Step 1: Identify cause of missing CSRF token

   The form must include the CSRF token field to send it with the request.

2. Step 2: Choose the fix that adds the token to the form

   Adding {{ form.csrf_token }} in the form template includes the token correctly.

3. Final Answer:

   Add {{ form.csrf_token }} inside the HTML form -> Option A

4. Quick Check:

   Include CSRF token in form to fix error [OK]

Hint: Always include {{ form.csrf_token }} in your forms [OK]

Common Mistakes:

• Removing CSRF protection instead of fixing token
• Changing POST to GET (not secure)
• Disabling CSRF protection globally

5. You want to protect a Flask route that updates user data via POST. Which combination correctly applies CSRF protection and form token usage?

hard

A. Use CSRFProtect(app) but omit the token in the form for simplicity

B. Enable CSRFProtect(app) and include {{ form.csrf_token }} in the HTML form

C. Skip CSRFProtect and rely on session cookies only

D. Add {{ form.csrf_token }} in form but do not enable CSRFProtect in app

Solution

1. Step 1: Understand CSRF protection requirements

   Both enabling CSRFProtect in the app and including the token in the form are needed to protect POST routes.

2. Step 2: Evaluate each option for completeness

   Only Enable CSRFProtect(app) and include {{ form.csrf_token }} in the HTML form correctly enables protection and includes the token in the form.

3. Final Answer:

   Enable CSRFProtect(app) and include {{ form.csrf_token }} in the HTML form -> Option B

4. Quick Check:

   Protect POST routes with CSRFProtect and token [OK]

Hint: Both app and form need CSRF setup for protection [OK]

Common Mistakes:

• Enabling protection but missing token in form
• Including token but not enabling protection
• Relying only on session cookies