Update book

Author: Avaneesh-axiom

book/src/SUMMARY.md

@@ -19,7 +19,7 @@
 
 - [Overview](./custom-extensions/overview.md)
 - [Keccak](./custom-extensions/keccak.md)
-- [SHA-256](./custom-extensions/sha256.md)
+- [SHA-2](./custom-extensions/sha2.md)
 - [Big Integer](./custom-extensions/bigint.md)
 - [Algebra (Modular Arithmetic)](./custom-extensions/algebra.md)
 - [Elliptic Curve Cryptography](./custom-extensions/ecc.md)

book/src/custom-extensions/overview.md

@@ -5,13 +5,13 @@ You can seamlessly integrate certain performance-optimized extensions maintained
 In this chapter, we will explain how to use the following existing extensions:
 
 - [`openvm-keccak-guest`](./keccak.md) - Keccak256 hash function.
-- [`openvm-sha256-guest`](./sha256.md) - SHA2-256 hash function.
+- [`openvm-sha2-guest`](./sha2.md) - SHA-2 hash functions.
 - [`openvm-bigint-guest`](./bigint.md) - Big integer arithmetic for 256-bit signed and unsigned integers.
 - [`openvm-algebra-guest`](./algebra.md) - Modular arithmetic and complex field extensions.
 - [`openvm-ecc-guest`](./ecc.md) - Elliptic curve cryptography.
 - [`openvm-pairing-guest`](./pairing.md) - Elliptic curve optimal Ate pairings.
 
-Some extensions such as `openvm-keccak-guest`, `openvm-sha256-guest`, and `openvm-bigint-guest` can be enabled without specifying any additional configuration.
+Some extensions such as `openvm-keccak-guest`, `openvm-sha2-guest`, and `openvm-bigint-guest` can be enabled without specifying any additional configuration.
 
 On the other hand certain arithmetic operations, particularly modular arithmetic, can be optimized significantly when the modulus is known at compile time. This approach requires a framework to inform the compiler about all the moduli and associated arithmetic structures we intend to use. To achieve this, three steps are involved:
 
@@ -35,7 +35,7 @@ The template `openvm.toml` file is as follows:
 [app_vm_config.rv32m]
 [app_vm_config.io]
 [app_vm_config.keccak]
-[app_vm_config.sha256]
+[app_vm_config.sha2]
 [app_vm_config.native]
 [app_vm_config.bigint]
 [app_vm_config.modular]

book/src/custom-extensions/sha2.md

@@ -1,36 +1,40 @@
-# SHA-256
+# SHA-2
 
-The OpenVM SHA-256 extension provides tools for using the SHA-256 hash function. Refer [here](https://nvlpubs.nist.gov/nistpubs/FIPS/NIST.FIPS.180-4.pdf) for more details on SHA-256.
-The functional part is provided by the `openvm-sha256-guest` crate, which is a guest library that can be used in any OpenVM program.
+The OpenVM SHA-2 extension provides tools for using the SHA-256, SHA-512, and SHA-384 hash functions. Refer [here](https://nvlpubs.nist.gov/nistpubs/FIPS/NIST.FIPS.180-4.pdf) for more details on the SHA-2 family of hash functions.
+The functional part is provided by the `openvm-sha2-guest` crate, which is a guest library that can be used in any OpenVM program.
 
 ## Functions for guest code
 
-The OpenVM SHA-256Guest extension provides two functions for using in your guest code:
+The OpenVM SHA-2 Guest extension provides three pairs of functions for using in your guest code:
 
 - `sha256(input: &[u8]) -> [u8; 32]`: Computes the SHA-256 hash of the input data and returns it as an array of 32 bytes.
-- `set_sha256(input: &[u8], output: &mut [u8; 32])`: Sets the output to the SHA-256 hash of the input data into the provided output buffer.
+- `set_sha256(input: &[u8], output: &mut [u8; 32])`: Sets the provided output buffer to the SHA-256 hash of the input data.
+- `sha512(input: &[u8]) -> [u8; 64]`: Computes the SHA-512 hash of the input data and returns it as an array of 64 bytes.
+- `set_sha512(input: &[u8], output: &mut [u8; 64])`: Sets the provided output buffer to the SHA-512 hash of the input data.
+- `sha384(input: &[u8]) -> [u8; 48]`: Computes the SHA-384 hash of the input data and returns it as an array of 48 bytes.
+- `set_sha384(input: &[u8], output: &mut [u8; 64])`: Sets the first 48 bytes of the provided output buffer to the SHA-384 hash of the input data and sets the rest of the buffer to zero.
 
-See the full example [here](https://github.com/openvm-org/openvm/blob/main/examples/sha256/src/main.rs).
+See the full example [here](https://github.com/openvm-org/openvm/blob/main/examples/sha2/src/main.rs).
 
 ### Example
 
 ```rust,no_run,noplayground
-{{ #include ../../../examples/sha256/src/main.rs:imports }}
-{{ #include ../../../examples/sha256/src/main.rs:main }}
+{{ #include ../../../examples/sha2/src/main.rs:imports }}
+{{ #include ../../../examples/sha2/src/main.rs:main }}
 ```
 
-To be able to import the `sha256` function, add the following to your `Cargo.toml` file:
+To be able to import the `shaXXX` functions, add the following to your `Cargo.toml` file:
 
 ```toml
-openvm-sha256-guest = { git = "https://github.com/openvm-org/openvm.git" }
+openvm-sha2-guest = { git = "https://github.com/openvm-org/openvm.git" }
 hex = { version = "0.4.3", default-features = false, features = ["alloc"] }
 ```
 
 ## External Linking
 
-The SHA-256 guest extension also provides another way to use the intrinsic SHA-256 implementation. It provides a function that is meant to be linked to other external libraries. The external libraries can use this function as a hook for the SHA-256 intrinsic. This is enabled only when the target is `zkvm`.
+The SHA-2 guest extension also provides another way to use the intrinsic SHA-2 implementations. It provides functions that are meant to be linked to other external libraries. The external libraries can use these functions as hooks for the SHA-2 intrinsics. This is enabled only when the target is `zkvm`.
 
-- `zkvm_sha256_impl(input: *const u8, len: usize, output: *mut u8)`: This function has `C` ABI. It takes in a pointer to the input, the length of the input, and a pointer to the output buffer.
+- `zkvm_shaXXX_impl(input: *const u8, len: usize, output: *mut u8)`: where `XXX` is `256`, `512`, or `384`. These functions have `C` ABI. They take in a pointer to the input, the length of the input, and a pointer to the output buffer.
 
 In the external library, you can do the following:
 
@@ -59,5 +63,5 @@ fn sha256(input: &[u8]) -> [u8; 32] {
 For the guest program to build successfully add the following to your `.toml` file:
 
 ```toml
-[app_vm_config.sha256]
+[app_vm_config.sha2]
 ```

book/src/introduction.md

@@ -12,7 +12,7 @@ OpenVM is an open-source zero-knowledge virtual machine (zkVM) framework focused
 
   - RISC-V support via RV32IM
   - A native field arithmetic extension for proof recursion and aggregation
-  - The Keccak-256 and SHA2-256 hash functions
+  - The Keccak-256, SHA-256, SHA-512, and SHA-384 hash functions
   - Int256 arithmetic
   - Modular arithmetic over arbitrary fields
   - Elliptic curve operations, including multi-scalar multiplication and ECDSA signature verification, including for the secp256k1 and secp256r1 curves.