256 lines
9.6 KiB
PHP
256 lines
9.6 KiB
PHP
<?php
|
||
|
||
namespace app\controllers;
|
||
|
||
use app\models\RenameForm;
|
||
use app\utils\FileTypeDetector;
|
||
use Yii;
|
||
use yii\filters\VerbFilter;
|
||
use yii\helpers\ArrayHelper;
|
||
use yii\web\Controller;
|
||
use yii\web\NotFoundHttpException;
|
||
use yii\web\Response;
|
||
|
||
class HomeController extends Controller
|
||
{
|
||
public function behaviors()
|
||
{
|
||
return array_merge(
|
||
parent::behaviors(),
|
||
[
|
||
'verbs' => [
|
||
'class' => VerbFilter::class,
|
||
'actions' => [
|
||
'index' => ['GET'],
|
||
'download' => ['GET'],
|
||
'rename' => ['POST'],
|
||
'delete' => ['POST'],
|
||
],
|
||
],
|
||
]
|
||
);
|
||
}
|
||
|
||
/**
|
||
* diplay the page of the file manager (accepts a directory parameter like cd command)
|
||
* visit it via https://devs.chenx221.cyou:8081/index.php?r=home
|
||
*
|
||
* @return string|Response
|
||
* @throws NotFoundHttpException
|
||
*/
|
||
public function actionIndex($directory = null)
|
||
{
|
||
//Warning: Security Vulnerability: access via $directory parameter = ../ will display the internal files of the server
|
||
if (Yii::$app->user->isGuest) {
|
||
return $this->redirect(Yii::$app->user->loginUrl);
|
||
}
|
||
$rootDataDirectory = Yii::getAlias(Yii::$app->params['dataDirectory']);
|
||
$userId = Yii::$app->user->id;
|
||
|
||
if ($directory === '.' ||$directory == null) {
|
||
$directory = null;
|
||
$parentDirectory = null;
|
||
} elseif ($directory === '..' || str_contains($directory, '../')) {
|
||
throw new NotFoundHttpException('Invalid directory.');
|
||
} else {
|
||
$parentDirectory = dirname($directory);
|
||
}
|
||
$directoryContents = $this->getDirectoryContents(join(DIRECTORY_SEPARATOR, [$rootDataDirectory, $userId, $directory ?: '.']));
|
||
foreach ($directoryContents as $key => $item) {
|
||
$relativePath = $directory ? $directory . '/' . $item : $item;
|
||
$absolutePath = Yii::getAlias('@app') . '/data/' . Yii::$app->user->id . '/' . $relativePath;
|
||
$type = FileTypeDetector::detect($absolutePath);
|
||
$directoryContents[$key] = ['name' => $item, 'type' => $type];
|
||
}
|
||
return $this->render('index', [
|
||
'directoryContents' => $directoryContents,
|
||
'parentDirectory' => $parentDirectory,
|
||
'directory' => $directory, // 将$directory传递给视图
|
||
]);
|
||
}
|
||
|
||
/**
|
||
* 获取指定路径下的文件和文件夹内容
|
||
* @param string $path 路径
|
||
* @return array 文件和文件夹内容数组
|
||
* @throws NotFoundHttpException 如果路径不存在
|
||
*/
|
||
protected function getDirectoryContents($path)
|
||
{
|
||
// 确定路径是否存在
|
||
if (!is_dir($path)) {
|
||
throw new NotFoundHttpException('Directory not found.');
|
||
}
|
||
|
||
// 获取路径下的所有文件和文件夹
|
||
$directoryContents = scandir($path);
|
||
|
||
// 移除 '.' 和 '..'
|
||
$directoryContents = array_diff($directoryContents, ['.', '..']);
|
||
|
||
// 使用 usort 对目录内容进行排序，使文件夹始终在文件之前
|
||
usort($directoryContents, function ($a, $b) use ($path) {
|
||
$aIsDir = is_dir($path . '/' . $a);
|
||
$bIsDir = is_dir($path . '/' . $b);
|
||
if ($aIsDir === $bIsDir) {
|
||
return strnatcasecmp($a, $b); // 如果两者都是文件夹或都是文件，按名称排序
|
||
}
|
||
return $aIsDir ? -1 : 1; // 文件夹始终在文件之前
|
||
});
|
||
|
||
return $directoryContents;
|
||
}
|
||
|
||
/**
|
||
* 下载指定路径下的文件
|
||
* download link:https://devs.chenx221.cyou:8081/index.php?r=home%2Fdownload&relativePath={the relative path of the file}
|
||
*
|
||
* @param string $relativePath 文件的相对路径
|
||
* @throws NotFoundHttpException 如果文件不存在
|
||
*/
|
||
public function actionDownload($relativePath)
|
||
{
|
||
// 对相对路径进行解码
|
||
$relativePath = rawurldecode($relativePath);
|
||
|
||
// 检查相对路径是否只包含允许的字符
|
||
if (!preg_match('/^[\w\-.\/\s]+$/u', $relativePath) || $relativePath === '.' || $relativePath === '..' || str_contains($relativePath, '../')) {
|
||
throw new NotFoundHttpException('Invalid file path.');
|
||
}
|
||
|
||
// 确定文件的绝对路径
|
||
$absolutePath = Yii::getAlias(Yii::$app->params['dataDirectory']) . '/' . Yii::$app->user->id . '/' . $relativePath;
|
||
|
||
// 使用realpath函数解析路径，并检查解析后的路径是否在预期的目录中
|
||
$realPath = realpath($absolutePath);
|
||
$dataDirectory = str_replace('/', '\\', Yii::getAlias(Yii::$app->params['dataDirectory']));
|
||
if (!$realPath || !str_starts_with($realPath, $dataDirectory)) {
|
||
throw new NotFoundHttpException('File not found.');
|
||
}
|
||
|
||
// 检查文件是否存在
|
||
if (!file_exists($realPath)) {
|
||
throw new NotFoundHttpException('File not found.');
|
||
}
|
||
|
||
// 将文件发送给用户进行下载
|
||
Yii::$app->response->sendFile($realPath)->send();
|
||
}
|
||
|
||
/**
|
||
* 重命名文件或文件夹
|
||
* @param string $relativePath 文件或文件夹的相对路径
|
||
* @param string $newName 新名称
|
||
* @throws NotFoundHttpException 如果文件或文件夹不存在
|
||
*/
|
||
public function actionRename()
|
||
{
|
||
$relativePath = Yii::$app->request->post('relativePath');
|
||
|
||
// 对相对路径进行解码
|
||
$relativePath = rawurldecode($relativePath);
|
||
|
||
// 检查相对路径是否只包含允许的字符
|
||
if (!preg_match('/^[\w\-.\/\s]+$/u', $relativePath) || $relativePath === '.' || $relativePath === '..' || str_contains($relativePath, '../')) {
|
||
throw new NotFoundHttpException('Invalid file path.');
|
||
}
|
||
|
||
// 确定文件的绝对路径
|
||
$absolutePath = Yii::getAlias(Yii::$app->params['dataDirectory']) . '/' . Yii::$app->user->id . '/' . $relativePath;
|
||
|
||
// 检查文件或文件夹是否存在
|
||
if (!file_exists($absolutePath)) {
|
||
throw new NotFoundHttpException('File or directory not found.');
|
||
}
|
||
|
||
$model = new RenameForm();
|
||
|
||
$model->newName = ArrayHelper::getValue(Yii::$app->request->post('RenameForm'), 'newName');
|
||
|
||
if (!$model->validate()) {
|
||
Yii::$app->response->statusCode = 400;
|
||
return $model->getFirstError('newName');
|
||
}
|
||
|
||
// 检查新名称是否和现有的文件或文件夹重名
|
||
if (file_exists(dirname($absolutePath) . '/' . $model->newName)) {
|
||
Yii::$app->session->setFlash('error', 'Failed to rename file or directory.');
|
||
return $this->redirect(['index', 'directory' => dirname($relativePath)]);
|
||
}
|
||
|
||
// 重命名文件或文件夹
|
||
if (!rename($absolutePath, dirname($absolutePath) . '/' . $model->newName)) {
|
||
Yii::$app->session->setFlash('error', 'Failed to rename file or directory.');
|
||
} else {
|
||
Yii::$app->session->setFlash('success', 'File or directory renamed successfully.');
|
||
}
|
||
return $this->redirect(['index', 'directory' => dirname($relativePath)]);
|
||
}
|
||
|
||
/**
|
||
* 删除文件或文件夹
|
||
* @throws NotFoundHttpException 如果文件或文件夹不存在
|
||
*/
|
||
public function actionDelete()
|
||
{
|
||
$relativePath = Yii::$app->request->post('relativePath');
|
||
$relativePath = rawurldecode($relativePath);
|
||
if (!preg_match('/^[\w\-.\/\s]+$/u', $relativePath) || $relativePath === '.' || $relativePath === '..' || str_contains($relativePath, '../')) {
|
||
throw new NotFoundHttpException('Invalid file path.');
|
||
}
|
||
$absolutePath = Yii::getAlias(Yii::$app->params['dataDirectory']) . '/' . Yii::$app->user->id . '/' . $relativePath;
|
||
if (!file_exists($absolutePath)) {
|
||
throw new NotFoundHttpException('File or directory not found.');
|
||
} else {
|
||
$realPath = realpath($absolutePath);
|
||
$expectedPathPrefix = realpath(Yii::getAlias(Yii::$app->params['dataDirectory']) . '/' . Yii::$app->user->id);
|
||
if (!str_starts_with($realPath, $expectedPathPrefix)) {
|
||
throw new NotFoundHttpException('File or directory not found.');
|
||
}
|
||
}
|
||
|
||
if (is_dir($absolutePath)) {
|
||
if (!$this->deleteDirectory($absolutePath)) {
|
||
Yii::$app->session->setFlash('error', 'Failed to delete directory.');
|
||
} else {
|
||
Yii::$app->session->setFlash('success', 'Directory deleted successfully.');
|
||
}
|
||
} else {
|
||
if (!unlink($absolutePath)) {
|
||
Yii::$app->session->setFlash('error', 'Failed to delete file.');
|
||
} else {
|
||
Yii::$app->session->setFlash('success', 'File deleted successfully.');
|
||
}
|
||
}
|
||
return $this->redirect(['index', 'directory' => dirname($relativePath)]);
|
||
}
|
||
|
||
/**
|
||
* 递归删除目录及其内容
|
||
* @param string $dir 目录路径
|
||
*/
|
||
protected function deleteDirectory($dir): bool
|
||
{
|
||
if (!is_dir($dir)) {
|
||
return false;
|
||
}
|
||
|
||
$files = array_diff(scandir($dir), ['.', '..']);
|
||
foreach ($files as $file) {
|
||
if (is_dir("$dir/$file")) {
|
||
if (!$this->deleteDirectory("$dir/$file")) {
|
||
return false;
|
||
}
|
||
} else {
|
||
if (!unlink("$dir/$file")) {
|
||
return false;
|
||
}
|
||
}
|
||
}
|
||
if (!rmdir($dir)) {
|
||
return false;
|
||
}
|
||
return true;
|
||
}
|
||
}
|