input length and range checking

3 files changed, 67 insertions, 3 deletions

diff --git a/src/api/auth.rs b/src/api/auth.rs
index d60483f..b469d4d 100644
--- a/src/api/auth.rs
+++ b/src/api/auth.rs
@@ -3,7 +3,7 @@ use serde::Deserialize;
 use time::{OffsetDateTime, Duration};
 use tower_cookies::{Cookies, Cookie};
 
-use crate::types::{user::User, response::ResponseCode, session::Session, extract::{Json, AuthorizedUser}};
+use crate::types::{user::User, response::ResponseCode, session::Session, extract::{Json, AuthorizedUser, Check, CheckResult}};
 
 #[derive(Deserialize)]
 struct RegistrationRequet {
@@ -17,6 +17,20 @@ struct RegistrationRequet {
     year: u32
 }
 
+impl Check for RegistrationRequet {
+    fn check(&self) -> CheckResult {
+        Self::assert_length(&self.firstname, 1, 20, "First name can only by 1-20 characters long")?;
+        Self::assert_length(&self.lastname, 1, 20, "Last name can only by 1-20 characters long")?;
+        Self::assert_length(&self.email, 1, 50, "Email can only by 1-50 characters long")?;
+        Self::assert_length(&self.password, 1, 50, "Password can only by 1-50 characters long")?;
+        Self::assert_length(&self.gender, 1, 100, "Gender can only by 1-100 characters long")?;
+        Self::assert_range(self.day as u64, 1, 255, "Birthday day can only be between 1-255")?;
+        Self::assert_range(self.month as u64, 1, 255, "Birthday month can only be between 1-255")?;
+        Self::assert_range(self.year as u64, 1, 2147483647, "Birthday year can only be between 1-2147483647")?;
+        Ok(())
+    }
+}
+
 
 async fn register(cookies: Cookies, Json(body): Json<RegistrationRequet>) -> Response {
 
@@ -50,6 +64,12 @@ struct LoginRequest {
     password: String,
 }
 
+impl Check for LoginRequest {
+    fn check(&self) -> CheckResult {
+        Ok(())
+    }
+}
+
 async fn login(cookies: Cookies, Json(body): Json<LoginRequest>) -> Response {
     
     let Ok(user) = User::from_email(&body.email) else {
diff --git a/src/api/posts.rs b/src/api/posts.rs
index 405dfa6..85ff2b2 100644
--- a/src/api/posts.rs
+++ b/src/api/posts.rs
@@ -1,7 +1,7 @@
 use axum::{response::Response, Router, routing::{post, patch}};
 use serde::Deserialize;
 
-use crate::types::{extract::{AuthorizedUser, Json}, post::Post, response::ResponseCode};
+use crate::types::{extract::{AuthorizedUser, Json, Check, CheckResult}, post::Post, response::ResponseCode};
 
 
 #[derive(Deserialize)]
@@ -9,6 +9,13 @@ struct PostCreateRequest {
     content: String
 }
 
+impl Check for PostCreateRequest {
+    fn check(&self) -> CheckResult {
+        Self::assert_length(&self.content, 1, 500, "Comments must be between 1-500 characters long")?;
+        Ok(())
+    }
+}
+
 async fn create(AuthorizedUser(user): AuthorizedUser, Json(body): Json<PostCreateRequest>) -> Response {
 
     let Ok(_post) = Post::new(user.user_id, body.content) else {
@@ -23,6 +30,12 @@ struct PostPageRequest {
     page: u64
 }
 
+impl Check for PostPageRequest {
+    fn check(&self) -> CheckResult {
+        Ok(())
+    }
+}
+
 async fn page(AuthorizedUser(_user): AuthorizedUser, Json(body): Json<PostPageRequest>) -> Response {
     
     let Ok(posts) = Post::from_post_page(body.page) else {
@@ -41,6 +54,12 @@ struct UsersPostsRequest {
     user_id: u64
 }
 
+impl Check for UsersPostsRequest {
+    fn check(&self) -> CheckResult {
+        Ok(())
+    }
+}
+
 async fn user(AuthorizedUser(_user): AuthorizedUser, Json(body): Json<UsersPostsRequest>) -> Response {
     
     let Ok(posts) = Post::from_user_id(body.user_id) else {
@@ -60,6 +79,13 @@ struct PostCommentRequest {
     post_id: u64
 }
 
+impl Check for PostCommentRequest {
+    fn check(&self) -> CheckResult {
+        Self::assert_length(&self.content, 1, 255, "Comments must be between 1-255 characters long")?;
+        Ok(())
+    }
+}
+
 async fn comment(AuthorizedUser(user): AuthorizedUser, Json(body): Json<PostCommentRequest>) -> Response {
     
     let Ok(mut post) = Post::from_post_id(body.post_id) else {
@@ -79,6 +105,12 @@ struct PostLikeRequest {
     post_id: u64
 }
 
+impl Check for PostLikeRequest {
+    fn check(&self) -> CheckResult {
+        Ok(())
+    }
+}
+
 async fn like(AuthorizedUser(user): AuthorizedUser, Json(body): Json<PostLikeRequest>) -> Response {
     
     let Ok(mut post) = Post::from_post_id(body.post_id) else {
diff --git a/src/api/users.rs b/src/api/users.rs
index 283ec96..45ed195 100644
--- a/src/api/users.rs
+++ b/src/api/users.rs
@@ -1,12 +1,18 @@
 use axum::{Router, response::Response, routing::post};
 use serde::Deserialize;
-use crate::types::{extract::{AuthorizedUser, Json}, response::ResponseCode, user::User};
+use crate::types::{extract::{AuthorizedUser, Json, Check, CheckResult}, response::ResponseCode, user::User};
 
 #[derive(Deserialize)]
 struct UserLoadRequest {
     ids: Vec<u64>
 }
 
+impl Check for UserLoadRequest {
+    fn check(&self) -> CheckResult {
+        Ok(())
+    }
+}
+
 async fn load_batch(AuthorizedUser(_user): AuthorizedUser, Json(body): Json<UserLoadRequest>) -> Response {
     
     let users = User::from_user_ids(body.ids);
@@ -22,6 +28,12 @@ struct UserPageReqiest {
     page: u64
 }
 
+impl Check for UserPageReqiest {
+    fn check(&self) -> CheckResult {
+        Ok(())
+    }
+}
+
 async fn load_page(AuthorizedUser(_user): AuthorizedUser, Json(body): Json<UserPageReqiest>) -> Response {
     
     let Ok(users) = User::from_user_page(body.page) else {