backupManager = $backupManager; } /** * Handles updating the state of a backup. * * @return \Illuminate\Http\JsonResponse * * @throws \Throwable */ public function index(ReportBackupCompleteRequest $request, string $backup) { /** @var \Pterodactyl\Models\Backup $model */ $model = Backup::query()->where('uuid', $backup)->firstOrFail(); if ($model->is_successful) { throw new BadRequestHttpException('Cannot update the status of a backup that is already marked as completed.'); } $action = $request->boolean('successful') ? 'server:backup.complete' : 'server:backup.fail'; $log = Activity::event($action)->subject($model, $model->server)->property('name', $model->name); $log->transaction(function () use ($model, $request) { $successful = $request->boolean('successful'); $model->fill([ 'is_successful' => $successful, // Change the lock state to unlocked if this was a failed backup so that it can be // deleted easily. Also does not make sense to have a locked backup on the system // that is failed. 'is_locked' => $successful ? $model->is_locked : false, 'checksum' => $successful ? ($request->input('checksum_type') . ':' . $request->input('checksum')) : null, 'bytes' => $successful ? $request->input('size') : 0, 'completed_at' => CarbonImmutable::now(), ])->save(); // Check if we are using the s3 backup adapter. If so, make sure we mark the backup as // being completed in S3 correctly. $adapter = $this->backupManager->adapter(); if ($adapter instanceof AwsS3Adapter) { $this->completeMultipartUpload($model, $adapter, $successful); } }); return new JsonResponse([], JsonResponse::HTTP_NO_CONTENT); } /** * Handles toggling the restoration status of a server. The server status field should be * set back to null, even if the restoration failed. This is not an unsolvable state for * the server, and the user can keep trying to restore, or just use the reinstall button. * * The only thing the successful field does is update the entry value for the audit logs * table tracking for this restoration. * * @return \Illuminate\Http\JsonResponse * * @throws \Throwable */ public function restore(Request $request, string $backup) { /** @var \Pterodactyl\Models\Backup $model */ $model = Backup::query()->where('uuid', $backup)->firstOrFail(); $model->server->update(['status' => null]); Activity::event($request->boolean('successful') ? 'server:backup.restore-complete' : 'server.backup.restore-failed') ->subject($model, $model->server) ->property('name', $model->name) ->log(); return new JsonResponse([], JsonResponse::HTTP_NO_CONTENT); } /** * Marks a multipart upload in a given S3-compatiable instance as failed or successful for * the given backup. * * @throws \Exception * @throws \Pterodactyl\Exceptions\DisplayException */ protected function completeMultipartUpload(Backup $backup, AwsS3Adapter $adapter, bool $successful) { // This should never really happen, but if it does don't let us fall victim to Amazon's // wildly fun error messaging. Just stop the process right here. if (empty($backup->upload_id)) { // A failed backup doesn't need to error here, this can happen if the backup encouters // an error before we even start the upload. AWS gives you tooling to clear these failed // multipart uploads as needed too. if (!$successful) { return; } throw new DisplayException('Cannot complete backup request: no upload_id present on model.'); } $params = [ 'Bucket' => $adapter->getBucket(), 'Key' => sprintf('%s/%s.tar.gz', $backup->server->uuid, $backup->uuid), 'UploadId' => $backup->upload_id, ]; $client = $adapter->getClient(); if (!$successful) { $client->execute($client->getCommand('AbortMultipartUpload', $params)); return; } // Otherwise send a CompleteMultipartUpload request. $params['MultipartUpload'] = [ 'Parts' => $client->execute($client->getCommand('ListParts', $params))['Parts'], ]; $client->execute($client->getCommand('CompleteMultipartUpload', $params)); } }