File upload security in Cybersecurity - Mini Project: Build & Apply

File Upload Security

📖 Scenario: You are working on a website that allows users to upload files. To keep the website safe, you need to understand how to check and control the files users upload.

🎯 Goal: Build a simple checklist and rules to ensure uploaded files are safe and do not harm the website or its users.

📋 What You'll Learn

Create a list of allowed file types

Set a maximum file size limit

Check the file extension against allowed types

Add a final rule to reject files that do not meet the criteria

💡 Why This Matters

🌍 Real World

Websites and apps often let users upload pictures or documents. Checking file types and sizes helps keep these platforms safe from harmful files.

💼 Career

Understanding file upload security is important for web developers, security analysts, and IT professionals to protect systems from attacks and data loss.

1

Create a list of allowed file types

Create a list called allowed_types that contains these exact strings: 'jpg', 'png', 'gif', and 'pdf'.

Cybersecurity

# Create the list allowed_types with 'jpg', 'png', 'gif', and 'pdf'
# Your code here

Hint

Think of allowed_types as a list of file endings that are safe to accept.

2

Set a maximum file size limit

Create a variable called max_file_size and set it to 5_000_000 to represent 5 megabytes in bytes.

Cybersecurity

allowed_types = ['jpg', 'png', 'gif', 'pdf']
# Set max_file_size to 5,000,000 bytes
# Your code here

Hint

5 megabytes is 5 million bytes. Use underscores to make the number easier to read.

3

Check the file extension against allowed types

Write a function called is_allowed_file that takes a parameter filename. Inside the function, get the file extension by splitting filename at the last dot and converting it to lowercase. Return True if the extension is in allowed_types, otherwise False.

Cybersecurity

allowed_types = ['jpg', 'png', 'gif', 'pdf']
max_file_size = 5_000_000

# Define is_allowed_file function
# Your code here

Hint

Use rsplit('.', 1) to split from the right and get the extension.

4

Add the final rule to reject unsafe files

Write a function called is_file_safe that takes filename and filesize. It should return True only if is_allowed_file(filename) is True and filesize is less than or equal to max_file_size. Otherwise, return False.

Cybersecurity

allowed_types = ['jpg', 'png', 'gif', 'pdf']
max_file_size = 5_000_000

def is_allowed_file(filename):
    ext = filename.rsplit('.', 1)[-1].lower()
    return ext in allowed_types

# Define is_file_safe function
# Your code here

Hint

Combine the checks for file type and size with and.

Practice

1. What is the main purpose of file upload security in web applications?

easy

A. To increase the file size limit

B. To speed up the file upload process

C. To allow all file types without restrictions

D. To prevent harmful files from being uploaded and executed

Solution

1. Step 1: Understand the risks of file uploads

   Uploading files can introduce harmful content like viruses or scripts that can damage the system.

2. Step 2: Identify the goal of file upload security

   The goal is to stop harmful files from entering and running on the server or user devices.

3. Final Answer:

   To prevent harmful files from being uploaded and executed -> Option D

4. Quick Check:

   File upload security = prevent harmful files [OK]

Hint: File upload security stops dangerous files from entering [OK]

Common Mistakes:

• Thinking file upload security speeds up uploads
• Believing all file types should be allowed
• Confusing file size limits with security

2. Which of the following is a correct practice for validating uploaded files on the server?

easy

A. Check the file's MIME type and scan for malware

B. Only check the file size, ignoring content type

C. Accept all files and scan them later

D. Allow files based only on their file extension

Solution

1. Step 1: Understand file validation methods

   File extension alone can be faked; MIME type and malware scanning provide stronger checks.

2. Step 2: Identify the best validation practice

   Checking MIME type ensures the file is of expected type; scanning detects harmful content.

3. Final Answer:

   Check the file's MIME type and scan for malware -> Option A

4. Quick Check:

   Validate MIME type + scan malware = secure upload [OK]

Hint: Validate MIME type and scan files for safety [OK]

Common Mistakes:

• Relying only on file extensions
• Ignoring malware scanning
• Accepting all files without checks

3. Consider this code snippet for handling file uploads:

if uploaded_file.content_type == 'image/png' and uploaded_file.size <= 1048576:
    save_file(uploaded_file)
else:
    reject_upload()

What will happen if a user uploads a 2MB PNG file?

medium

A. The file will be rejected due to size limit

B. The file will be saved successfully

C. The file will be rejected due to wrong type

D. The code will cause a runtime error

Solution

1. Step 1: Check the file type condition

   The file is PNG, so content_type == 'image/png' is true.

2. Step 2: Check the file size condition

   The file size is 2MB (2,097,152 bytes), which is greater than 1MB (1,048,576 bytes), so size condition fails.

3. Final Answer:

   The file will be rejected due to size limit -> Option A

4. Quick Check:

   File size > limit = reject upload [OK]

Hint: Check both type and size conditions carefully [OK]

Common Mistakes:

• Ignoring the size check and assuming success
• Confusing file size units
• Assuming code errors without cause

4. A developer wrote this code to validate uploaded files:

if uploaded_file.extension == '.jpg' or '.png':
    process_file(uploaded_file)
else:
    reject_file()

What is the main problem with this code?

medium

A. It only accepts .jpg files

B. It rejects all files incorrectly

C. The condition always evaluates to true, accepting all files

D. It causes a syntax error

Solution

1. Step 1: Analyze the condition logic

   The expression 'uploaded_file.extension == '.jpg' or '.png'' always evaluates '.png' as true because non-empty strings are truthy.

2. Step 2: Understand the effect on file acceptance

   Since the condition is always true, all files pass and get processed regardless of extension.

3. Final Answer:

   The condition always evaluates to true, accepting all files -> Option C

4. Quick Check:

   Incorrect or/or logic = always true condition [OK]

Hint: Use explicit comparisons for each extension [OK]

Common Mistakes:

• Assuming it only accepts .jpg or .png
• Thinking it causes syntax error
• Not understanding boolean logic in conditions

5. You want to securely allow users to upload profile pictures but avoid risks. Which combination of these steps is best practice? A) Check file extension only B) Validate MIME type and scan for malware C) Limit file size to 2MB D) Rename files to safe names before saving Choose the best combination.

hard

A. B and D only

B. B, C, and D

C. A and C only

D. A, B, C, and D

Solution

1. Step 1: Evaluate each step's security impact

   Checking extension alone is weak; validating MIME and scanning malware are strong protections. Limiting size prevents large uploads. Renaming files avoids overwriting and path issues.

2. Step 2: Identify the best combination

   Combining MIME validation, malware scan, size limit, and renaming covers multiple security aspects effectively.

3. Final Answer:

   B, C, and D -> Option B

4. Quick Check:

   Multiple layered checks = best security [OK]

Hint: Combine validation, size limit, and renaming for safety [OK]

Common Mistakes:

• Relying only on file extension
• Ignoring file size limits
• Not renaming files before saving