Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Replace ammonia lib with manual html escaping (fixes #3774) #3938

Merged
merged 7 commits into from
Sep 6, 2023
Merged
Show file tree
Hide file tree
Changes from 4 commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
20 changes: 0 additions & 20 deletions Cargo.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

13 changes: 11 additions & 2 deletions api_tests/src/post.spec.ts
Original file line number Diff line number Diff line change
Expand Up @@ -513,13 +513,22 @@ test("Sanitize HTML", async () => {
}

let name = randomString(5);
let body = "<script>alert('xss');</script> hello";
let body = "<script>alert('xss');</script> hello &\"'";
let form: CreatePost = {
name,
body,
auth: beta.auth,
community_id: betaCommunity.community.id,
};
let post = await beta.client.createPost(form);
expect(post.post_view.post.body).toBe(" hello");
// first escaping for the api
expect(post.post_view.post.body).toBe(
"&lt;script>alert(&#x27;xss&#x27;);&lt;/script> hello &amp;&quot;&#x27;",
);

let alphaPost = (await resolvePost(alpha, post.post_view.post)).post;
// second escaping over federation, avoid double escape of &
expect(alphaPost?.post.body).toBe(
"&lt;script>alert(&#x27;xss&#x27;);&lt;/script> hello &amp;&quot;&#x27;",
);
});
4 changes: 2 additions & 2 deletions crates/api/src/comment_report/create.rs
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@ use lemmy_api_common::{
utils::{
check_community_ban,
local_user_view_from_jwt,
sanitize_html,
sanitize_html_api,
send_new_report_email_to_admins,
},
};
Expand All @@ -31,7 +31,7 @@ pub async fn create_comment_report(
let local_user_view = local_user_view_from_jwt(&data.auth, &context).await?;
let local_site = LocalSite::read(&mut context.pool()).await?;

let reason = sanitize_html(data.reason.trim());
let reason = sanitize_html_api(data.reason.trim());
check_report_reason(&reason, &local_site)?;

let person_id = local_user_view.person.id;
Expand Down
4 changes: 2 additions & 2 deletions crates/api/src/community/ban.rs
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@ use lemmy_api_common::{
is_mod_or_admin,
local_user_view_from_jwt,
remove_user_data_in_community,
sanitize_html_opt,
sanitize_html_api_opt,
},
};
use lemmy_db_schema::{
Expand Down Expand Up @@ -86,7 +86,7 @@ pub async fn ban_from_community(
mod_person_id: local_user_view.person.id,
other_person_id: data.person_id,
community_id: data.community_id,
reason: sanitize_html_opt(&data.reason),
reason: sanitize_html_api_opt(&data.reason),
banned: Some(data.ban),
expires,
};
Expand Down
4 changes: 2 additions & 2 deletions crates/api/src/community/hide.rs
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@ use lemmy_api_common::{
community::{CommunityResponse, HideCommunity},
context::LemmyContext,
send_activity::{ActivityChannel, SendActivityData},
utils::{is_admin, local_user_view_from_jwt, sanitize_html_opt},
utils::{is_admin, local_user_view_from_jwt, sanitize_html_api_opt},
};
use lemmy_db_schema::{
source::{
Expand Down Expand Up @@ -33,7 +33,7 @@ pub async fn hide_community(
let mod_hide_community_form = ModHideCommunityForm {
community_id: data.community_id,
mod_person_id: local_user_view.person.id,
reason: sanitize_html_opt(&data.reason),
reason: sanitize_html_api_opt(&data.reason),
hidden: Some(data.hidden),
};

Expand Down
4 changes: 2 additions & 2 deletions crates/api/src/local_user/ban_person.rs
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@ use lemmy_api_common::{
context::LemmyContext,
person::{BanPerson, BanPersonResponse},
send_activity::{ActivityChannel, SendActivityData},
utils::{is_admin, local_user_view_from_jwt, remove_user_data, sanitize_html_opt},
utils::{is_admin, local_user_view_from_jwt, remove_user_data, sanitize_html_api_opt},
};
use lemmy_db_schema::{
source::{
Expand Down Expand Up @@ -54,7 +54,7 @@ pub async fn ban_from_site(
let form = ModBanForm {
mod_person_id: local_user_view.person.id,
other_person_id: data.person_id,
reason: sanitize_html_opt(&data.reason),
reason: sanitize_html_api_opt(&data.reason),
banned: Some(data.ban),
expires,
};
Expand Down
8 changes: 4 additions & 4 deletions crates/api/src/local_user/save_settings.rs
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@ use actix_web::web::{Data, Json};
use lemmy_api_common::{
context::LemmyContext,
person::{LoginResponse, SaveUserSettings},
utils::{local_user_view_from_jwt, sanitize_html_opt, send_verification_email},
utils::{local_user_view_from_jwt, sanitize_html_api_opt, send_verification_email},
};
use lemmy_db_schema::{
source::{
Expand Down Expand Up @@ -34,8 +34,8 @@ pub async fn save_user_settings(
let local_user_view = local_user_view_from_jwt(&data.auth, &context).await?;
let site_view = SiteView::read_local(&mut context.pool()).await?;

let bio = sanitize_html_opt(&data.bio);
let display_name = sanitize_html_opt(&data.display_name);
let bio = sanitize_html_api_opt(&data.bio);
let display_name = sanitize_html_api_opt(&data.display_name);

let avatar = diesel_option_overwrite_to_url(&data.avatar)?;
let banner = diesel_option_overwrite_to_url(&data.banner)?;
Expand Down Expand Up @@ -85,7 +85,7 @@ pub async fn save_user_settings(
let person_id = local_user_view.person.id;
let default_listing_type = data.default_listing_type;
let default_sort_type = data.default_sort_type;
let theme = sanitize_html_opt(&data.theme);
let theme = sanitize_html_api_opt(&data.theme);

let person_form = PersonUpdateForm {
display_name,
Expand Down
4 changes: 2 additions & 2 deletions crates/api/src/post_report/create.rs
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@ use lemmy_api_common::{
utils::{
check_community_ban,
local_user_view_from_jwt,
sanitize_html,
sanitize_html_api,
send_new_report_email_to_admins,
},
};
Expand All @@ -31,7 +31,7 @@ pub async fn create_post_report(
let local_user_view = local_user_view_from_jwt(&data.auth, &context).await?;
let local_site = LocalSite::read(&mut context.pool()).await?;

let reason = sanitize_html(data.reason.trim());
let reason = sanitize_html_api(data.reason.trim());
check_report_reason(&reason, &local_site)?;

let person_id = local_user_view.person.id;
Expand Down
4 changes: 2 additions & 2 deletions crates/api/src/private_message_report/create.rs
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@ use actix_web::web::{Data, Json};
use lemmy_api_common::{
context::LemmyContext,
private_message::{CreatePrivateMessageReport, PrivateMessageReportResponse},
utils::{local_user_view_from_jwt, sanitize_html, send_new_report_email_to_admins},
utils::{local_user_view_from_jwt, sanitize_html_api, send_new_report_email_to_admins},
};
use lemmy_db_schema::{
source::{
Expand All @@ -24,7 +24,7 @@ pub async fn create_pm_report(
let local_user_view = local_user_view_from_jwt(&data.auth, &context).await?;
let local_site = LocalSite::read(&mut context.pool()).await?;

let reason = sanitize_html(data.reason.trim());
let reason = sanitize_html_api(data.reason.trim());
check_report_reason(&reason, &local_site)?;

let person_id = local_user_view.person.id;
Expand Down
4 changes: 2 additions & 2 deletions crates/api/src/site/purge/comment.rs
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@ use actix_web::web::{Data, Json};
use lemmy_api_common::{
context::LemmyContext,
site::{PurgeComment, PurgeItemResponse},
utils::{is_admin, local_user_view_from_jwt, sanitize_html_opt},
utils::{is_admin, local_user_view_from_jwt, sanitize_html_api_opt},
};
use lemmy_db_schema::{
source::{
Expand Down Expand Up @@ -35,7 +35,7 @@ pub async fn purge_comment(
Comment::delete(&mut context.pool(), comment_id).await?;

// Mod tables
let reason = sanitize_html_opt(&data.reason);
let reason = sanitize_html_api_opt(&data.reason);
let form = AdminPurgeCommentForm {
admin_person_id: local_user_view.person.id,
reason,
Expand Down
9 changes: 7 additions & 2 deletions crates/api/src/site/purge/community.rs
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,12 @@ use lemmy_api_common::{
context::LemmyContext,
request::purge_image_from_pictrs,
site::{PurgeCommunity, PurgeItemResponse},
utils::{is_admin, local_user_view_from_jwt, purge_image_posts_for_community, sanitize_html_opt},
utils::{
is_admin,
local_user_view_from_jwt,
purge_image_posts_for_community,
sanitize_html_api_opt,
},
};
use lemmy_db_schema::{
source::{
Expand Down Expand Up @@ -42,7 +47,7 @@ pub async fn purge_community(
Community::delete(&mut context.pool(), community_id).await?;

// Mod tables
let reason = sanitize_html_opt(&data.reason);
let reason = sanitize_html_api_opt(&data.reason);
let form = AdminPurgeCommunityForm {
admin_person_id: local_user_view.person.id,
reason,
Expand Down
9 changes: 7 additions & 2 deletions crates/api/src/site/purge/person.rs
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,12 @@ use lemmy_api_common::{
context::LemmyContext,
request::purge_image_from_pictrs,
site::{PurgeItemResponse, PurgePerson},
utils::{is_admin, local_user_view_from_jwt, purge_image_posts_for_person, sanitize_html_opt},
utils::{
is_admin,
local_user_view_from_jwt,
purge_image_posts_for_person,
sanitize_html_api_opt,
},
};
use lemmy_db_schema::{
source::{
Expand Down Expand Up @@ -41,7 +46,7 @@ pub async fn purge_person(
Person::delete(&mut context.pool(), person_id).await?;

// Mod tables
let reason = sanitize_html_opt(&data.reason);
let reason = sanitize_html_api_opt(&data.reason);
let form = AdminPurgePersonForm {
admin_person_id: local_user_view.person.id,
reason,
Expand Down
4 changes: 2 additions & 2 deletions crates/api/src/site/purge/post.rs
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@ use lemmy_api_common::{
context::LemmyContext,
request::purge_image_from_pictrs,
site::{PurgeItemResponse, PurgePost},
utils::{is_admin, local_user_view_from_jwt, sanitize_html_opt},
utils::{is_admin, local_user_view_from_jwt, sanitize_html_api_opt},
};
use lemmy_db_schema::{
source::{
Expand Down Expand Up @@ -43,7 +43,7 @@ pub async fn purge_post(
Post::delete(&mut context.pool(), post_id).await?;

// Mod tables
let reason = sanitize_html_opt(&data.reason);
let reason = sanitize_html_api_opt(&data.reason);
let form = AdminPurgePostForm {
admin_person_id: local_user_view.person.id,
reason,
Expand Down
2 changes: 0 additions & 2 deletions crates/api_common/Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,6 @@ full = [
"actix-web",
"futures",
"once_cell",
"ammonia",
]

[dependencies]
Expand Down Expand Up @@ -67,4 +66,3 @@ once_cell = { workspace = true, optional = true }
actix-web = { workspace = true, optional = true }
# necessary for wasmt compilation
getrandom = { version = "0.2.10", features = ["js"] }
ammonia = { version = "3.3.0", optional = true }
52 changes: 28 additions & 24 deletions crates/api_common/src/utils.rs
Original file line number Diff line number Diff line change
Expand Up @@ -798,29 +798,43 @@ pub fn generate_moderators_url(community_id: &DbUrl) -> Result<DbUrl, LemmyError
Ok(Url::parse(&format!("{community_id}/moderators"))?.into())
}

/// Sanitize HTML with default options. Additionally, dont allow bypassing markdown
/// links and images
pub fn sanitize_html(data: &str) -> String {
ammonia::Builder::default()
.rm_tags(&["a", "img"])
.clean(data)
.to_string()
// restore markdown quotes
.replace("&gt;", ">")
// restore white space
.replace("&nbsp;", " ")
/// Replace special HTML characters in API parameters to prevent XSS attacks.
///
/// Taken from https://github.com/OWASP/CheatSheetSeries/blob/master/cheatsheets/Cross_Site_Scripting_Prevention_Cheat_Sheet.md#output-encoding-for-html-contexts
///
/// `>` is left in place because it is interpreted as markdown quote.
pub fn sanitize_html_api(data: &str) -> String {
data
.replace('&', "&amp;")
.replace('<', "&lt;")
.replace('\"', "&quot;")
.replace('\'', "&#x27;")
}

pub fn sanitize_html_opt(data: &Option<String>) -> Option<String> {
data.as_ref().map(|d| sanitize_html(d))
pub fn sanitize_html_api_opt(data: &Option<String>) -> Option<String> {
data.as_ref().map(|d| sanitize_html_api(d))
}

/// Replace special HTML characters in federation parameters to prevent XSS attacks.
///
/// Unlike [sanitize_html_api()] it leaves `&` in place to avoid double escaping.
pub fn sanitize_html_federation(data: &str) -> String {
data
.replace('<', "&lt;")
.replace('\"', "&quot;")
.replace('\'', "&#x27;")
}

pub fn sanitize_html_federation_opt(data: &Option<String>) -> Option<String> {
data.as_ref().map(|d| sanitize_html_federation(d))
}
Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Main change here.


#[cfg(test)]
mod tests {
#![allow(clippy::unwrap_used)]
#![allow(clippy::indexing_slicing)]

use crate::utils::{honeypot_check, password_length_check, sanitize_html};
use crate::utils::{honeypot_check, password_length_check};

#[test]
#[rustfmt::skip]
Expand All @@ -838,14 +852,4 @@ mod tests {
assert!(honeypot_check(&Some("1".to_string())).is_err());
assert!(honeypot_check(&Some("message".to_string())).is_err());
}

#[test]
fn test_sanitize_html() {
let sanitized = sanitize_html("<script>alert(1);</script> hello");
assert_eq!(sanitized, " hello");
let sanitized = sanitize_html("<img src='http://example.com'> test");
assert_eq!(sanitized, " test");
let sanitized = sanitize_html("Hello&nbsp;World");
assert_eq!(sanitized, "Hello World");
}
}
4 changes: 2 additions & 2 deletions crates/api_crud/src/comment/create.rs
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ use lemmy_api_common::{
get_post,
local_site_to_slur_regex,
local_user_view_from_jwt,
sanitize_html,
sanitize_html_api,
EndpointType,
},
};
Expand Down Expand Up @@ -52,7 +52,7 @@ pub async fn create_comment(
&local_site_to_slur_regex(&local_site),
);
is_valid_body_field(&Some(content.clone()), false)?;
let content = sanitize_html(&content);
let content = sanitize_html_api(&content);

// Check for a community ban
let post_id = data.post_id;
Expand Down