From 7ff96175e4e60e6f1f473173f902114bb9e87a26 Mon Sep 17 00:00:00 2001
From: Andrew Walker <awalker@ixsystems.com>
Date: Wed, 20 Nov 2024 15:51:26 -0600
Subject: [PATCH] Add test for maximum SMB valid pwrite range

We need to validate we can write above 16 TiB offset and that
we get properly stopped writing about 64 TiB offset.
---
 tests/api2/test_smb_write_offset.py | 60 +++++++++++++++++++++++++++++
 1 file changed, 60 insertions(+)
 create mode 100644 tests/api2/test_smb_write_offset.py

diff --git a/tests/api2/test_smb_write_offset.py b/tests/api2/test_smb_write_offset.py
new file mode 100644
index 000000000000..46c085bf0318
--- /dev/null
+++ b/tests/api2/test_smb_write_offset.py
@@ -0,0 +1,60 @@
+import pytest
+import random
+
+from middlewared.test.integration.assets.account import user, group
+from middlewared.test.integration.assets.pool import dataset
+
+from protocols import smb_connection
+
+from samba import ntstatus
+from samba import NTSTATUSError
+
+SHARE_NAME = 'offset_test'
+
+LARGE_OFFSET = 0x0000140000000000
+INVALID_OFFSET = 0x0000410000000000
+
+
+@pytest.fixture(scope='module')
+def setup_smb_tests():
+    with dataset('smbclient-testing', data={'share_type': 'SMB'}) as ds:
+        with user({
+            'username': 'smbuser',
+            'full_name': 'smbuser',
+            'group_create': True,
+            'password': 'Abcd1234'
+        }) as u:
+            with smb_share(os.path.join('/mnt', ds), SHARE_NAME) as s:
+                try:
+                    call('service.start', 'cifs')
+                    yield {'dataset': ds, 'share': s, 'user': u}
+                finally:
+                    call('service.stop', 'cifs')
+
+
+def test_valid_offset(setup_smb_tests):
+    ds, share, smb_user = setup_smb_tests
+    with smb_connection(
+        share=SHARE_NAME,
+        username=smb_user['username'],
+        password='Abcd1234',
+        smb1=False
+    ) as c:
+        fd = c.create_file('file_valid_offset', 'w')
+        buf = random.randbytes(1024)
+
+        c.write(fd, offset=LARGE_OFFSET, data=buf)
+        out = c.read(fd, offset=LARGE_OFFSET, cnt=1024)
+        assert buf == out
+
+
+def test_invalid_offset(setup_smb_tests):
+    ds, share, smb_user = setup_smb_tests
+    with smb_connection(
+        share=SHARE_NAME,
+        username=smb_user['username'],
+        password='Abcd1234',
+        smb1=False
+    ) as c:
+        fd = c.create_file('file_valid_offset', 'w')
+        c.write(fd, offset=INVALID_OFFSET, data=b'CANARY')