1
0
mirror of https://github.com/chylex/TweetDuck.git synced 2024-11-21 20:42:46 +01:00
TweetDuck/linux/TweetImpl.CefGlue/Handlers/Resource/DisableCspResourceHandler.cs
2022-02-19 18:19:13 +01:00

121 lines
3.2 KiB
C#

using System;
using System.IO;
using System.Runtime.InteropServices;
using TweetImpl.CefGlue.Utils;
using Xilium.CefGlue;
namespace TweetImpl.CefGlue.Handlers.Resource {
sealed class DisableCspResourceHandler : CefResourceHandler {
private readonly CefFrame frame;
private Client? client;
private int position;
public DisableCspResourceHandler(CefFrame frame) {
this.frame = frame;
}
protected override bool Open(CefRequest request, out bool handleRequest, CefCallback callback) {
if (client != null) {
throw new InvalidOperationException("DisableCspResourceHandler is not reusable!");
}
client = new Client(callback);
CefRuntime.PostTask(CefThreadId.IO, new CefActionTask(() => frame.CreateUrlRequest(request, client)));
handleRequest = false;
return true;
}
protected override void GetResponseHeaders(CefResponse response, out long responseLength, out string? redirectUrl) {
redirectUrl = null;
if (client is not { Response: {} result }) {
responseLength = 0;
return;
}
response.Url = result.Url;
response.Charset = result.Charset;
response.MimeType = result.MimeType;
response.Status = result.Status;
response.StatusText = result.StatusText;
response.Error = result.Error;
var headers = result.GetHeaderMap();
headers.Remove("Content-Security-Policy");
response.SetHeaderMap(headers);
responseLength = client.Stream.Length;
}
protected override bool Skip(long bytesToSkip, out long bytesSkipped, CefResourceSkipCallback callback) {
throw new NotSupportedException();
}
protected override bool Read(IntPtr dataOut, int bytesToRead, out int bytesRead, CefResourceReadCallback callback) {
callback.Dispose();
if (client == null) {
bytesRead = 0;
return false;
}
if (bytesToRead > 0) {
int maxLength = (int) (client.Stream.Length - position);
if (bytesToRead > maxLength) {
bytesToRead = maxLength;
}
Marshal.Copy(client.Stream.GetBuffer(), position, dataOut, bytesToRead);
position += bytesToRead;
}
bytesRead = bytesToRead;
return bytesRead > 0;
}
protected override void Cancel() {}
protected override void Dispose(bool disposing) {
base.Dispose(disposing);
frame.Dispose();
client?.Dispose();
}
private sealed class Client : CefUrlRequestClient, IDisposable {
public MemoryStream Stream { get; }
public CefResponse? Response { get; private set; }
private readonly CefCallback openCallback;
public Client(CefCallback openCallback) {
this.openCallback = openCallback;
this.Stream = new MemoryStream();
}
protected override void OnDownloadData(CefUrlRequest request, Stream data) {
data.CopyTo(Stream);
}
protected override void OnRequestComplete(CefUrlRequest request) {
if (request.RequestStatus == CefUrlRequestStatus.Success) {
Response = request.GetResponse();
openCallback.Continue();
}
else {
openCallback.Cancel();
}
}
protected override void OnDownloadProgress(CefUrlRequest request, long current, long total) {}
protected override void OnUploadProgress(CefUrlRequest request, long current, long total) {}
public void Dispose() {
Stream.Dispose();
Response?.Dispose();
openCallback.Dispose();
}
}
}
}