mirror of
https://github.com/Sneed-Group/pypush-plus-plus
synced 2024-10-30 08:27:52 +00:00
remove cloudkit stuff not ready yet
This commit is contained in:
JJTech0130
parent
3be5a17b0f
commit
cbe553d369
4 changed files with 0 additions and 1029 deletions
|
|
@ -1,60 +0,0 @@
|
|||
import sys
|
||||
sys.path.append(".")
|
||||
|
||||
import requests
|
||||
import uuid
|
||||
import plistlib
|
||||
from base64 import b64encode, b64decode
|
||||
import json
|
||||
import random
|
||||
import icloud.gsa as gsa
|
||||
import icloud.cloudkit as cloudkit
|
||||
import icloud
|
||||
|
||||
from rich.logging import RichHandler
|
||||
import logging
|
||||
logging.basicConfig(
|
||||
level=logging.INFO, format="%(message)s", datefmt="[%X]", handlers=[RichHandler()]
|
||||
)
|
||||
|
||||
def main():
|
||||
CONFIG_PATH = "config/cloudkit.json"
|
||||
# See if we have a search party token saved
|
||||
import os
|
||||
if os.path.exists(CONFIG_PATH):
|
||||
logging.info("Using saved config...")
|
||||
#print("Found search party token!")
|
||||
with open(CONFIG_PATH, "r") as f:
|
||||
j = json.load(f)
|
||||
cloudkit_token = j["cloudkit_token"]
|
||||
ds_prs_id = j["ds_prs_id"]
|
||||
mme_token = j["mme_token"]
|
||||
|
||||
else:
|
||||
# Prompt for username and password
|
||||
USERNAME = input("Username: ")
|
||||
PASSWORD = input("Password: ")
|
||||
|
||||
r = icloud.login(USERNAME, PASSWORD, delegates=["com.apple.mobileme"])
|
||||
|
||||
cloudkit_token = r['delegates']['com.apple.mobileme']['service-data']['tokens']['cloudKitToken']
|
||||
mme_token = r['delegates']['com.apple.mobileme']['service-data']['tokens']['mmeAuthToken']
|
||||
#ds_prs_id = r['delegates']['com.apple.mobileme']['service-data']['appleAccountInfo']['dsPrsID'] # This can also be obtained from the grandslam response
|
||||
ds_prs_id = r['dsid']
|
||||
|
||||
logging.info("Logged in!")
|
||||
|
||||
with open(CONFIG_PATH, "w") as f:
|
||||
json.dump({
|
||||
"cloudkit_token": cloudkit_token,
|
||||
"ds_prs_id": ds_prs_id,
|
||||
"mme_token": mme_token,
|
||||
}, f, indent=4)
|
||||
|
||||
logging.debug("CloudKit token: ", cloudkit_token)
|
||||
|
||||
ck = cloudkit.CloudKit(ds_prs_id, cloudkit_token, mme_token, sandbox=True)
|
||||
ck.container("iCloud.dev.jjtech.experiments.cktest").save_record(cloudkit.Record(uuid.uuid4(), "ToDoItem", {"title": "Test"}))
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
|
|
@ -1,594 +0,0 @@
|
|||
syntax = "proto2";
|
||||
|
||||
// Based on https://gist.github.com/horrorho/9c0ff20055b854f4298750cfad635fe2
|
||||
// and https://github.com/horrorho/InflatableDonkey/blob/master/src/main/resources/cloud_kit.proto
|
||||
message RequestOperation {
|
||||
message Header {
|
||||
enum ContainerEnvironment {
|
||||
PRODUCTION = 1;
|
||||
SANDBOX = 2;
|
||||
}
|
||||
|
||||
enum Database {
|
||||
PRIVATE_DB = 1;
|
||||
PUBLIC_DB = 2;
|
||||
SHARED_DB = 3;
|
||||
}
|
||||
|
||||
enum IsolationLevel {
|
||||
ZONE = 1;
|
||||
OPERATION = 2;
|
||||
}
|
||||
|
||||
optional string userToken = 1;
|
||||
optional string applicationContainer = 2;
|
||||
optional string applicationBundle = 3;
|
||||
optional string applicationVersion = 4;
|
||||
optional uint64 applicationConfigVersion = 5;
|
||||
optional uint64 globalConfigVersion = 6;
|
||||
optional Identifier deviceIdentifier = 7;
|
||||
optional string deviceSoftwareVersion = 8;
|
||||
optional string deviceHardwareVersion = 9;
|
||||
optional string deviceLibraryName = 10;
|
||||
optional string deviceLibraryVersion = 11;
|
||||
optional string deviceFlowControlKey = 12;
|
||||
optional uint64 deviceFlowControlBudget = 13;
|
||||
optional uint64 deviceFlowControlBudgetCap = 14;
|
||||
optional float deviceFlowControlRegeneration = 15;
|
||||
optional uint64 deviceProtocolVersion = 16;
|
||||
optional Locale locale = 17;
|
||||
optional string mmcsProtocolVersion = 18;
|
||||
optional ContainerEnvironment applicationContainerEnvironment = 19;
|
||||
optional bytes clientChangeToken = 20;
|
||||
optional string deviceAssignedName = 21;
|
||||
optional string deviceHardwareID = 22;
|
||||
optional Database targetDatabase = 23;
|
||||
optional string userIDContainerID = 24;
|
||||
optional IsolationLevel isolationLevel = 25;
|
||||
optional int32 unk1 = 29;
|
||||
optional string unk2 = 32; // UUID
|
||||
optional string deviceSerial = 33;
|
||||
optional int32 unk3 = 34;
|
||||
optional int32 unk4 = 35;
|
||||
|
||||
}
|
||||
|
||||
optional Header header = 1;
|
||||
optional Operation request = 2;
|
||||
optional ZoneRetrieveRequest zoneRetrieveRequest = 201;
|
||||
optional RecordSaveRequest recordSaveRequest = 210;
|
||||
optional RecordRetrieveRequest recordRetrieveRequest = 211;
|
||||
optional QueryRetrieveRequest queryRetrieveRequest = 220;
|
||||
}
|
||||
|
||||
message RecordSaveRequest {
|
||||
optional Record record = 1;
|
||||
optional int32 unk1 = 2; // 1
|
||||
optional int32 unk2 = 6; // 2
|
||||
}
|
||||
|
||||
message ResponseOperation {
|
||||
message Result {
|
||||
message Error {
|
||||
message Client {
|
||||
enum Code {
|
||||
UNKNOWN = 1;
|
||||
EXPIRED_APP_CONFIG = 2;
|
||||
EXPIRED_GLOBAL_CONFIG = 3;
|
||||
BAD_SYNTAX = 4;
|
||||
FORBIDDEN = 5;
|
||||
THROTTLED = 6;
|
||||
REFUSED = 7;
|
||||
NOT_SUPPORTED = 8;
|
||||
EXISTS = 9;
|
||||
REQUEST_ALREADY_PROCESSED = 10;
|
||||
BAD_AUTH_TOKEN = 11;
|
||||
NEEDS_AUTHENTICATION = 12;
|
||||
MESCAL_SIGNATURE_MISSING = 13;
|
||||
INVALID_MESCAL_SIGNATURE = 14;
|
||||
OP_LOCK_FAILURE = 15;
|
||||
ATOMIC_FAILURE = 16;
|
||||
RESET_NEEDED = 17;
|
||||
FIELDS_PER_TYPE_LIMIT_EXCEEDED = 18;
|
||||
TYPE_BUSY = 19;
|
||||
NUM_RECORD_TYPES_LIMIT_EXCEEDED = 20;
|
||||
INVALID_CONTAINER = 21;
|
||||
INVALID_RECORD_TYPE_NAME = 22;
|
||||
INVALID_FIELD_NAME = 23;
|
||||
INVALID_FIELD_VALUE = 24;
|
||||
INVALID_IDENTIFIER = 25;
|
||||
FIELD_NOT_QUERYABLE = 26;
|
||||
FIELD_NOT_SORTABLE = 27;
|
||||
QUERY_FILTER_LIMIT_EXCEEDED = 28;
|
||||
QUERY_FILTER_VALUES_LIMIT_EXCEEDED = 29;
|
||||
MEMBERSHIP_QUERY_LIMIT_EXCEEDED = 30;
|
||||
NOTIFICATION_ADDITIONAL_FIELD_LIMIT_EXCEEDED = 31;
|
||||
NOTIFICATION_ADDITIONAL_FIELD_INVALID_TYPE = 32;
|
||||
SUBSCRIPTION_LIMIT_EXCEEDED = 33;
|
||||
UNIQUE_TRIGGER_LIMIT_EXCEEDED = 34;
|
||||
TRIGGER_SUBSCRIPTION_BINDING_TYPE_MISMATCH = 35;
|
||||
ZONE_SIZE_LIMIT_EXCEEDED = 36;
|
||||
ZONE_COUNT_LIMIT_EXCEEDED = 37;
|
||||
UNIQUE_FIELD_FAILURE = 38;
|
||||
VALIDATING_REFERENCE_ERROR = 39;
|
||||
FULL_RESET_NEEDED = 40;
|
||||
ALREADY_SHARED = 41;
|
||||
EMAIL_OUT_OF_NETWORK = 42;
|
||||
DUPLICATE_SUBSCRIPTION = 43;
|
||||
EXPIRED_PUT_RECEIPT = 46;
|
||||
QUOTA_EXCEEDED = 47;
|
||||
ZONE_NOT_FOUND = 48;
|
||||
INVALID_BUNDLE_ID = 49;
|
||||
UNSUPPORTED_DEVICE = 50;
|
||||
BLACK_LISTED = 51;
|
||||
RECORD_PROTECTION_INFO_TAG_MISMATCH = 52;
|
||||
ZONE_PROTECTION_INFO_TAG_MISMATCH = 53;
|
||||
ASSET_SIZE_LIMIT_EXCEEDED = 54;
|
||||
BATCH_OPERATION_LIMIT_EXCEEDED = 55;
|
||||
REQUEST_SIZE_LIMIT_EXCEEDED = 56;
|
||||
RECORD_SIZE_LIMIT_EXCEEDED = 57;
|
||||
DATABASE_COMMIT_SIZE_EXCEEDED = 58;
|
||||
USER_DELETED_DATA_FOR_ZONE = 59;
|
||||
STALE_RECORD_UPDATE = 60;
|
||||
SHARE_PARTICIPANT_LIMIT_EXCEEDED = 61;
|
||||
SHARE_PARTICIPANT_ERROR = 62;
|
||||
}
|
||||
|
||||
optional Code type = 1;
|
||||
}
|
||||
|
||||
message Server {
|
||||
enum Code {
|
||||
UNKNOWN = 1;
|
||||
OVERLOADED = 2;
|
||||
NOT_FOUND = 3;
|
||||
CONTAINER_UNAVAILABLE = 4;
|
||||
MESCAL_SIGNATURE_PARSE_ERROR = 6;
|
||||
ZONE_BUSY = 7;
|
||||
ZONE_UNAVAILABLE = 8;
|
||||
}
|
||||
|
||||
optional Code type = 1;
|
||||
}
|
||||
|
||||
message Extension {
|
||||
optional string extensionName = 1;
|
||||
optional uint32 typeCode = 2;
|
||||
optional bytes extensionPayload = 3;
|
||||
}
|
||||
|
||||
optional Client clientError = 1;
|
||||
optional Server serverError = 2;
|
||||
optional Extension extensionError = 7;
|
||||
optional int32 retryAfterSeconds = 3;
|
||||
optional string errorDescription = 4;
|
||||
optional string errorKey = 5;
|
||||
optional string errorInternal = 6;
|
||||
}
|
||||
|
||||
enum Code {
|
||||
SUCCESS = 1;
|
||||
PARTIAL = 2;
|
||||
FAILURE = 3;
|
||||
INDETERMINATE = 4;
|
||||
}
|
||||
|
||||
optional Code code = 1;
|
||||
optional Error error = 2;
|
||||
}
|
||||
|
||||
optional uint32 operationCost = 1;
|
||||
optional Operation response = 2;
|
||||
optional Result result = 3;
|
||||
optional ZoneRetrieveResponse zoneRetrieveResponse = 201;
|
||||
optional RecordRetrieveResponse recordRetrieveResponse = 211;
|
||||
optional QueryRetrieveResponse queryRetrieveResponse = 220;
|
||||
}
|
||||
|
||||
message Operation {
|
||||
enum Type {
|
||||
NONE_TYPE = 0;
|
||||
ZONE_SAVE_TYPE = 200;
|
||||
ZONE_RETRIEVE_TYPE = 201;
|
||||
ZONE_DELETE_TYPE = 202;
|
||||
ZONE_RETRIEVE_CHANGES_TYPE = 203;
|
||||
RECORD_SAVE_TYPE = 210;
|
||||
RECORD_RETRIEVE_TYPE = 211;
|
||||
RECORD_RETRIEVE_VERSIONS_TYPE = 212;
|
||||
RECORD_RETRIEVE_CHANGES_TYPE = 213;
|
||||
RECORD_DELETE_TYPE = 214;
|
||||
QUERY_RETRIEVE_TYPE = 220;
|
||||
ASSET_UPLOAD_TOKEN_RETRIEVE_TYPE = 230;
|
||||
CONTAINER_DELETE_TYPE = 240;
|
||||
CONTAINER_RESET_TYPE = 241;
|
||||
CONTAINER_SCHEMA_PROMOTION_TYPE = 242;
|
||||
USER_AVAILABLE_QUOTA_TYPE = 243;
|
||||
WEB_AUTH_TOKEN_RETRIEVE_TYPE = 250;
|
||||
SUBSCRIPTION_CREATE_TYPE = 300;
|
||||
SUBSCRIPTION_RETRIEVE_TYPE = 301;
|
||||
SUBSCRIPTION_DELETE_TYPE = 302;
|
||||
USER_RETRIEVE_TYPE = 400;
|
||||
USER_QUERYTYPE = 401;
|
||||
USER_PRIVACY_SETTINGS_RETRIEVE_TYPE = 402;
|
||||
USER_PRIVACY_SETTINGS_UPDATE_TYPE = 403;
|
||||
USER_PRIVACY_SETTINGS_RESET_TYPE = 404;
|
||||
USER_PRIVACY_SETTINGS_BATCH_LOOKUP = 405;
|
||||
SHARE_SAVE_TYPE = 500;
|
||||
SHARE_RETRIEVE_TYPE = 501;
|
||||
SHARE_DELETE_TYPE = 502;
|
||||
SHARE_ACCEPT_TYPE = 503;
|
||||
SHARE_TOKEN_SAVE_TYPE = 510;
|
||||
SHARE_TOKEN_RETRIEVE_TYPE = 511;
|
||||
SHARE_TOKEN_DELETE_TYPE = 512;
|
||||
POST_COMMENT_TYPE = 600;
|
||||
GET_COMMENTS_TYPE = 601;
|
||||
GET_COMMENT_TYPE = 613;
|
||||
DELETE_COMMENT_TYPE = 602;
|
||||
LIKE_TYPE = 610;
|
||||
UNLIKE_TYPE = 611;
|
||||
GET_LIKES_TYPE = 612;
|
||||
PULSE_TYPE = 700;
|
||||
PUSH_REGISTER_TYPE = 800;
|
||||
PUSH_UNREGISTER_TYPE = 801;
|
||||
PUSH_BADGE_TYPE = 802;
|
||||
PUSH_SYNC_TYPE = 803;
|
||||
PUSH_READ_TYPE = 804;
|
||||
MESCAL_SIGNATURE_TYPE = 1000;
|
||||
MESCAL_SESSION_INFO_TYPE = 1001;
|
||||
MESCAL_CERTIFICATE_TYPE = 1002;
|
||||
}
|
||||
|
||||
optional string operationUUID = 1;
|
||||
optional Type type = 2;
|
||||
optional bool synchronousMode = 3;
|
||||
optional bool last = 4;
|
||||
}
|
||||
|
||||
// /api/client/zone/retrieve
|
||||
message ZoneRetrieveRequest {
|
||||
optional RecordZoneIdentifier zoneIdentifier = 1;
|
||||
}
|
||||
|
||||
message ZoneRetrieveResponse {
|
||||
message ZoneSummary {
|
||||
optional Zone targetZone = 1;
|
||||
optional bytes currentServerContinuationToken = 2;
|
||||
optional bytes clientChangeToken = 3;
|
||||
optional int32 deviceCount = 4;
|
||||
optional int64 assetQuotaUsage = 5;
|
||||
optional int64 metadataQuotaUsage = 6;
|
||||
}
|
||||
|
||||
repeated ZoneSummary zoneSummary = 1;
|
||||
}
|
||||
|
||||
// /api/client/record/retrieve
|
||||
message RecordRetrieveRequest {
|
||||
message RetrieveAssetURL {
|
||||
enum Type {
|
||||
PUBLISHED_URL = 1;
|
||||
STREAMING_URL = 2;
|
||||
}
|
||||
|
||||
optional RequestedFields assetFields = 1;
|
||||
optional int64 requestedTTL = 2;
|
||||
optional Type type = 3;
|
||||
}
|
||||
|
||||
optional RecordIdentifier recordIdentifier = 1;
|
||||
optional RequestedFields requestedFields = 2;
|
||||
optional string versionETag = 3;
|
||||
optional string clientVersionETag = 4;
|
||||
optional RetrieveAssetURL getAssetURL = 5;
|
||||
optional AssetsToDownload assetsToDownload = 6;
|
||||
}
|
||||
|
||||
message RecordRetrieveResponse {
|
||||
optional Record record = 1;
|
||||
optional bool clientVersionETagMatch = 2;
|
||||
}
|
||||
|
||||
// /api/client/query/retrieve
|
||||
message QueryRetrieveRequest {
|
||||
optional Query query = 1;
|
||||
optional bytes continuationMarker = 2;
|
||||
optional uint32 limit = 3;
|
||||
optional RecordZoneIdentifier zoneIdentifier = 4;
|
||||
optional RequestedFields requestedFields = 5;
|
||||
optional AssetsToDownload assetsToDownload = 6;
|
||||
}
|
||||
|
||||
message QueryRetrieveResponse {
|
||||
message QueryResult {
|
||||
enum Type {
|
||||
ID_AND_ETAG = 1;
|
||||
FULL_RECORD = 2;
|
||||
}
|
||||
|
||||
optional RecordIdentifier identifier = 1;
|
||||
optional string etag = 2;
|
||||
optional Type type = 3;
|
||||
optional Record record = 4;
|
||||
}
|
||||
|
||||
repeated QueryResult queryResults = 1;
|
||||
optional bytes continuationMarker = 2;
|
||||
}
|
||||
|
||||
message Query {
|
||||
message Filter {
|
||||
enum Type {
|
||||
EQUALS = 1;
|
||||
NOT_EQUALS = 2;
|
||||
IN = 3;
|
||||
NOT_IN = 4;
|
||||
LESS_THAN = 5;
|
||||
LESS_THAN_OR_EQUALS = 6;
|
||||
GREATER_THAN = 7;
|
||||
GREATER_THAN_OR_EQUALS = 8;
|
||||
NEAR = 9;
|
||||
CONTAINS_ALL_TOKENS = 10;
|
||||
CONTAINS_ANY_TOKENS = 11;
|
||||
LIST_CONTAINS = 12;
|
||||
LIST_NOT_CONTAINS = 13;
|
||||
LIST_CONTAINS_ANY = 14;
|
||||
LIST_NOT_CONTAINS_ANY = 15;
|
||||
BEGINS_WITH = 16;
|
||||
NOT_BEGINS_WITH = 17;
|
||||
LIST_MEMBER_BEGINS_WITH = 18;
|
||||
NOT_LIST_MEMBER_BEGINS_WITH = 19;
|
||||
LIST_CONTAINS_ALL = 20;
|
||||
LIST_NOT_CONTAINS_ALL = 21;
|
||||
UNKNOWN = 22;
|
||||
}
|
||||
|
||||
optional Record.Field.Identifier fieldName = 1;
|
||||
optional Record.Field.Value fieldValue = 2;
|
||||
optional Location.Bound bounds = 3;
|
||||
optional Type type = 4;
|
||||
}
|
||||
|
||||
message Sort {
|
||||
enum Order {
|
||||
ASCENDING = 1;
|
||||
DESCENDING = 2;
|
||||
UNKNOWN = 3;
|
||||
}
|
||||
|
||||
optional Record.Field.Identifier fieldName= 1;
|
||||
optional Order order = 2;
|
||||
optional Location.Coordinate coordinate = 3;
|
||||
}
|
||||
|
||||
enum QueryOperator {
|
||||
AND = 1;
|
||||
OR = 2;
|
||||
}
|
||||
|
||||
repeated Record.Type types = 1;
|
||||
repeated Filter filters = 2;
|
||||
repeated Sort sorts = 3;
|
||||
optional bool distinct = 4;
|
||||
optional QueryOperator queryOperator = 5;
|
||||
}
|
||||
|
||||
message Date {
|
||||
optional double time = 1;
|
||||
}
|
||||
|
||||
message Asset {
|
||||
optional string owner = 1;
|
||||
optional bytes signature = 2;
|
||||
optional bytes header = 3;
|
||||
optional uint64 size = 4;
|
||||
optional string downloadToken = 5;
|
||||
optional bytes downloadRequest = 6;
|
||||
optional string derivedContentType = 7;
|
||||
optional string contentBaseURL = 8;
|
||||
optional string requestor = 9;
|
||||
optional RecordIdentifier recordID = 10;
|
||||
optional string uploadReceipt = 11;
|
||||
optional bytes data = 12;
|
||||
optional string downloadBaseURL = 13;
|
||||
optional uint64 downloadURLExpiration = 14;
|
||||
optional ProtectionInfo protectionInfo = 15;
|
||||
optional bytes referenceSignature = 17;
|
||||
optional uint64 downloadTokenExpiration = 18;
|
||||
}
|
||||
|
||||
message DateStatistics {
|
||||
optional Date creation = 1;
|
||||
optional Date modification = 2;
|
||||
}
|
||||
|
||||
message Location {
|
||||
message Bound {
|
||||
optional double radius = 1;
|
||||
}
|
||||
|
||||
message Coordinate {
|
||||
optional double lattitude = 1;
|
||||
optional double longitude = 2;
|
||||
optional double horizontalAccuracy = 3;
|
||||
optional double altitude = 4;
|
||||
optional double verticalAccuracy = 5;
|
||||
optional double course = 6;
|
||||
optional double speed = 7;
|
||||
optional Date timestamp = 8;
|
||||
}
|
||||
|
||||
optional Coordinate coordinate = 1;
|
||||
optional Bound bounds = 2;
|
||||
}
|
||||
|
||||
message Package {
|
||||
optional Asset manifest = 1;
|
||||
repeated Asset sections = 2;
|
||||
}
|
||||
|
||||
message Record {
|
||||
message Field {
|
||||
message Identifier {
|
||||
optional string name = 1;
|
||||
}
|
||||
|
||||
message Value {
|
||||
enum Type {
|
||||
BYTES_TYPE = 1;
|
||||
DATE_TYPE = 2;
|
||||
STRING_TYPE = 3;
|
||||
LOCATION_TYPE = 4;
|
||||
REFERENCE_TYPE =5;
|
||||
ASSET_TYPE = 6;
|
||||
INT64_TYPE = 7;
|
||||
DOUBLE_TYPE = 8;
|
||||
EMPTY_LIST = 9;
|
||||
DATE_LIST_TYPE = 10;
|
||||
BYTES_LIST_TYPE = 11;
|
||||
LOCATION_LIST_TYPE = 12;
|
||||
REFERENCE_LIST_TYPE = 13;
|
||||
ASSET_LIST_TYPE = 14;
|
||||
STRING_LIST_TYPE = 15;
|
||||
LIST_TYPE = 16;
|
||||
INT64_LIST_TYPE = 17;
|
||||
DOUBLE_LIST_TYPE = 18;
|
||||
PACKAGE_TYPE = 19;
|
||||
ENCRYPTED_BYTES_TYPE = 20;
|
||||
ENCRYPTED_BYTES_LIST_TYPE = 21;
|
||||
UNKNOWN = 22;
|
||||
}
|
||||
|
||||
optional Type type = 1;
|
||||
optional bytes bytesValue = 2;
|
||||
optional int64 signedValue = 4;
|
||||
optional double doubleValue = 5;
|
||||
optional Date dateValue = 6;
|
||||
optional string stringValue = 7;
|
||||
optional Location.Coordinate locationValue = 8;
|
||||
optional Reference referenceValue = 9;
|
||||
optional Asset assetValue = 10;
|
||||
repeated Value listValues = 11;
|
||||
optional Package packageValue = 12;
|
||||
}
|
||||
|
||||
optional Identifier identifier = 1;
|
||||
optional Value value = 2;
|
||||
}
|
||||
|
||||
message Reference {
|
||||
enum Type {
|
||||
OWNING = 1;
|
||||
WEAK = 2;
|
||||
VALIDATING = 3;
|
||||
}
|
||||
|
||||
optional Type type = 1;
|
||||
optional RecordIdentifier recordIdentifier = 2;
|
||||
}
|
||||
|
||||
message Type {
|
||||
optional string name = 1;
|
||||
}
|
||||
|
||||
optional string etag= 1;
|
||||
optional RecordIdentifier recordIdentifier = 2;
|
||||
optional Record.Type type = 3;
|
||||
optional Identifier createdBy = 4;
|
||||
optional DateStatistics timeStatistics = 5;
|
||||
repeated Record.Field recordField = 7;
|
||||
optional ShareIdentifier shareId = 8;;
|
||||
optional Identifier modifiedBy = 9;
|
||||
repeated string conflictLoserEtag = 10;
|
||||
optional string modifiedByDevice = 11;
|
||||
repeated Record.Field pluginFields = 12;
|
||||
optional ProtectionInfo protectionInfo = 13;
|
||||
optional uint32 permission = 15;
|
||||
}
|
||||
|
||||
message RequestedFields {
|
||||
repeated Record.Field.Identifier fields = 1;
|
||||
}
|
||||
|
||||
message AssetsToDownload {
|
||||
optional bool allAssets = 1;
|
||||
optional RequestedFields assetFields = 2;
|
||||
}
|
||||
|
||||
message Locale {
|
||||
optional string languageCode = 1;
|
||||
optional string regionCode = 2;
|
||||
repeated string enabledKeyboards = 3;
|
||||
optional string activeKeyboard = 4;
|
||||
}
|
||||
|
||||
message ShareIdentifier {
|
||||
optional Identifier value = 1;
|
||||
optional RecordZoneIdentifier zoneIdentifier = 2;
|
||||
}
|
||||
|
||||
message RecordIdentifier {
|
||||
optional Identifier value = 1;
|
||||
optional RecordZoneIdentifier zoneIdentifier = 2;
|
||||
}
|
||||
|
||||
message RecordZoneIdentifier {
|
||||
optional Identifier value = 1;
|
||||
optional Identifier ownerIdentifier = 2;
|
||||
}
|
||||
|
||||
message Identifier {
|
||||
enum Type {
|
||||
RECORD = 1;
|
||||
DEVICE = 2;
|
||||
SUBSCRIPTION = 3;
|
||||
SHARE = 4;
|
||||
COMMENT = 5;
|
||||
RECORD_ZONE = 6;
|
||||
USER = 7;
|
||||
}
|
||||
optional string name = 1;
|
||||
optional Type type = 2;
|
||||
}
|
||||
|
||||
message ProtectionInfo {
|
||||
optional bytes protectionInfo = 1;
|
||||
optional string protectionInfoTag = 2;
|
||||
}
|
||||
|
||||
message Zone {
|
||||
optional RecordZoneIdentifier zoneIdentifier = 1;
|
||||
optional string etag= 2;
|
||||
optional ProtectionInfo protectionInfo = 3;
|
||||
optional ProtectionInfo recordProtectionInfo = 6;
|
||||
}
|
||||
|
||||
message FileTokens {
|
||||
repeated FileToken fileTokens = 1;
|
||||
}
|
||||
|
||||
message FileToken {
|
||||
optional bytes signature = 1;
|
||||
optional string downloadToken = 2;
|
||||
optional uint64 size = 3;
|
||||
optional bytes referenceSignature = 4;
|
||||
}
|
||||
|
||||
message EncryptedAttributes {
|
||||
optional string relativePath = 1;
|
||||
optional string domain = 2;
|
||||
optional uint64 birth = 3;
|
||||
optional uint64 modified = 4;
|
||||
optional uint64 statusChanged = 5;
|
||||
optional uint64 size = 6;
|
||||
optional uint32 groupID = 7;
|
||||
optional uint32 userID = 8;
|
||||
optional uint32 mode = 9;
|
||||
optional uint64 sizeBeforeCopy = 10;
|
||||
optional bytes linkTarget = 11;
|
||||
optional bytes encryptionKey = 12;
|
||||
optional bytes sha256Signature = 13;
|
||||
optional int32 domainOrdinal = 14;
|
||||
optional int32 flags = 15;
|
||||
optional int32 contentEncodingMethod = 16;
|
||||
optional int32 contentCompressionMethod = 17;
|
||||
}
|
||||
|
|
@ -1,227 +0,0 @@
|
|||
import dataclasses
|
||||
import logging
|
||||
import random
|
||||
import typing
|
||||
import uuid
|
||||
from typing import Literal
|
||||
from io import BytesIO
|
||||
|
||||
|
||||
import requests
|
||||
|
||||
from . import _utils, cloudkit_pb2, gsa
|
||||
|
||||
logger = logging.getLogger("cloudkit")
|
||||
|
||||
|
||||
@dataclasses.dataclass
|
||||
class Record:
|
||||
name: uuid.UUID
|
||||
type: str
|
||||
fields: dict[str, typing.Any]
|
||||
|
||||
|
||||
class CloudKit:
|
||||
def __init__(
|
||||
self, dsid: str, cloudkit_token: str, mme_token: str, sandbox: bool = False
|
||||
):
|
||||
"""
|
||||
Represents a CloudKit user.
|
||||
`dsid`: The user's DSID.
|
||||
`cloudkit_token`: `cloudKitToken` from the `com.apple.mobileme` delegate.
|
||||
`mme_token`: `mmeAuthToken` from the `com.apple.mobileme` delegate.
|
||||
`sandbox`: Whether to use the CloudKit sandbox environment.
|
||||
"""
|
||||
self.dsid = dsid
|
||||
self.cloudkit_token = cloudkit_token
|
||||
self.mme_token = mme_token
|
||||
self.sandbox = sandbox
|
||||
|
||||
def container(
|
||||
self,
|
||||
container: str,
|
||||
scope: Literal["PUBLIC"] | Literal["PRIVATE"] | Literal["SHARED"] = "PUBLIC",
|
||||
) -> "CloudKitContainer":
|
||||
"""
|
||||
Convenience method for creating a CloudKitContainer object.
|
||||
"""
|
||||
return CloudKitContainer(container, self, scope)
|
||||
|
||||
|
||||
class CloudKitContainer:
|
||||
def __init__(
|
||||
self,
|
||||
container: str,
|
||||
user: CloudKit,
|
||||
scope: Literal["PUBLIC"] | Literal["PRIVATE"] | Literal["SHARED"] = "PUBLIC",
|
||||
):
|
||||
"""
|
||||
Represents a CloudKit container.
|
||||
container: The CloudKit container ID. (e.g. "iCloud.dev.jjtech.experiments.cktest")
|
||||
user: The CloudKit user to use for authentication.
|
||||
scope: The CloudKit database scope to use.
|
||||
"""
|
||||
self.container = container
|
||||
self.user = user
|
||||
self.scope = scope
|
||||
self.user_id = self._fetch_user_id()
|
||||
|
||||
def _fetch_user_id(self):
|
||||
headers = {
|
||||
"x-cloudkit-containerid": self.container,
|
||||
"x-cloudkit-bundleid": ".".join(
|
||||
self.container.split(".")[1:]
|
||||
), # Remove the "iCloud." prefix
|
||||
"x-cloudkit-databasescope": self.scope,
|
||||
"x-cloudkit-environment": "Sandbox" if self.user.sandbox else "Production",
|
||||
"accept": "application/x-protobuf",
|
||||
"x-apple-operation-id": random.randbytes(8).hex(),
|
||||
"x-apple-request-uuid": str(uuid.uuid4()).upper(),
|
||||
}
|
||||
|
||||
headers.update(gsa.generate_anisette_headers())
|
||||
|
||||
r = requests.post(
|
||||
"https://gateway.icloud.com/setup/setup/ck/v1/ckAppInit",
|
||||
params={"container": self.container},
|
||||
headers=headers,
|
||||
auth=(self.user.dsid, self.user.mme_token),
|
||||
verify=False,
|
||||
)
|
||||
|
||||
logger.debug("Got app init response: ", r.content)
|
||||
return r.json()["cloudKitUserId"]
|
||||
|
||||
def save_record(
|
||||
self, record: Record, zone: str = "_defaultZone", owner: str = "_defaultOwner"
|
||||
) -> None:
|
||||
"""
|
||||
Saves a record to the container.
|
||||
"""
|
||||
logger.info(f"Saving record {record.name} to {self.container}")
|
||||
|
||||
headers = {
|
||||
"x-cloudkit-authtoken": self.user.cloudkit_token,
|
||||
"x-cloudkit-userid": self.user_id,
|
||||
"x-cloudkit-containerid": self.container,
|
||||
"x-cloudkit-bundleid": ".".join(
|
||||
self.container.split(".")[1:]
|
||||
), # Remove the "iCloud." prefix
|
||||
"x-cloudkit-databasescope": self.scope,
|
||||
"x-cloudkit-environment": "Sandbox" if self.user.sandbox else "Production",
|
||||
"accept": "application/x-protobuf",
|
||||
"content-type": 'application/x-protobuf; desc="https://gateway.icloud.com:443/static/protobuf/CloudDB/CloudDBClient.desc"; messageType=RequestOperation; delimited=true',
|
||||
"x-apple-operation-id": random.randbytes(8).hex(),
|
||||
"x-apple-request-uuid": str(uuid.uuid4()).upper(),
|
||||
"user-agent": "CloudKit/2060.11 (22F82)",
|
||||
}
|
||||
|
||||
headers.update(gsa.generate_anisette_headers())
|
||||
|
||||
body = _build_record_save_request(
|
||||
record, self.container, self.user.sandbox, self.scope, zone, owner
|
||||
)
|
||||
r = requests.post(
|
||||
"https://gateway.icloud.com/ckdatabase/api/client/record/save",
|
||||
headers=headers,
|
||||
data=body,
|
||||
verify=False,
|
||||
)
|
||||
|
||||
_parse_response(r.content) # Will raise an exception if the response is an error
|
||||
|
||||
def _parse_response(response: bytes):
|
||||
from io import BytesIO
|
||||
length, read = _utils.ULEB128.decode_reader(BytesIO(response))
|
||||
if length + read < len(response):
|
||||
logger.warning(f"Response is longer than expected: {length + read} < {len(response)} (multiple messages?)")
|
||||
response = response[read:length+read]
|
||||
|
||||
try:
|
||||
r = cloudkit_pb2.ResponseOperation.FromString(response)
|
||||
except Exception as e:
|
||||
logger.warning(f"Failed to parse response: {e} {response.hex()}")
|
||||
raise
|
||||
|
||||
if r.result.code != cloudkit_pb2.ResponseOperation.Result.Code.SUCCESS:
|
||||
if r.result.code == cloudkit_pb2.ResponseOperation.Result.Code.FAILURE:
|
||||
raise Exception(f"CloudKit request failed: {r.result.error.errorDescription}")
|
||||
else:
|
||||
raise Exception("Unknown CloudKit error")
|
||||
|
||||
|
||||
def _build_record_save_request(
|
||||
record: Record,
|
||||
container: str,
|
||||
sandbox: bool = False,
|
||||
database: Literal["PUBLIC"] | Literal["PRIVATE"] | Literal["SHARED"] = "PUBLIC",
|
||||
zone: str = "_defaultZone",
|
||||
owner: str = "_defaultOwner",
|
||||
):
|
||||
hardware_id = uuid.uuid4() # Generate a new hardware ID for each request?
|
||||
operation_uuid = uuid.uuid4() # Generate a new operation UUID for each request?
|
||||
record_id = uuid.uuid4() # Generate a new record ID for each request?
|
||||
|
||||
request = cloudkit_pb2.RequestOperation()
|
||||
request.header.applicationContainer = container
|
||||
request.header.applicationContainerEnvironment = (
|
||||
cloudkit_pb2.RequestOperation.Header.ContainerEnvironment.SANDBOX
|
||||
if sandbox
|
||||
else cloudkit_pb2.RequestOperation.Header.ContainerEnvironment.PRODUCTION
|
||||
)
|
||||
|
||||
request.header.deviceHardwareID = str(hardware_id).upper()
|
||||
|
||||
if database == "PUBLIC":
|
||||
request.header.targetDatabase = (
|
||||
cloudkit_pb2.RequestOperation.Header.Database.PUBLIC_DB
|
||||
)
|
||||
elif database == "PRIVATE":
|
||||
request.header.targetDatabase = (
|
||||
cloudkit_pb2.RequestOperation.Header.Database.PRIVATE_DB
|
||||
)
|
||||
elif database == "SHARED":
|
||||
request.header.targetDatabase = (
|
||||
cloudkit_pb2.RequestOperation.Header.Database.SHARED_DB
|
||||
)
|
||||
|
||||
request.header.isolationLevel = (
|
||||
cloudkit_pb2.RequestOperation.Header.IsolationLevel.ZONE
|
||||
)
|
||||
|
||||
request.request.operationUUID = str(operation_uuid).upper()
|
||||
request.request.type = cloudkit_pb2.Operation.Type.RECORD_SAVE_TYPE
|
||||
request.request.last = True
|
||||
|
||||
request.recordSaveRequest.record.recordIdentifier.value.name = str(
|
||||
record_id
|
||||
).upper()
|
||||
request.recordSaveRequest.record.recordIdentifier.value.type = (
|
||||
cloudkit_pb2.Identifier.Type.RECORD
|
||||
)
|
||||
|
||||
request.recordSaveRequest.record.recordIdentifier.zoneIdentifier.value.name = zone
|
||||
request.recordSaveRequest.record.recordIdentifier.zoneIdentifier.value.type = (
|
||||
cloudkit_pb2.Identifier.Type.RECORD_ZONE
|
||||
)
|
||||
|
||||
request.recordSaveRequest.record.recordIdentifier.zoneIdentifier.ownerIdentifier.name = (
|
||||
owner
|
||||
)
|
||||
request.recordSaveRequest.record.recordIdentifier.zoneIdentifier.ownerIdentifier.type = (
|
||||
cloudkit_pb2.Identifier.Type.USER
|
||||
)
|
||||
|
||||
request.recordSaveRequest.record.type.name = record.type
|
||||
|
||||
for key, value in record.fields.items():
|
||||
request.recordSaveRequest.record.recordField.append(cloudkit_pb2.Record.Field())
|
||||
request.recordSaveRequest.record.recordField[-1].identifier.name = key
|
||||
request.recordSaveRequest.record.recordField[
|
||||
-1
|
||||
].value.type = cloudkit_pb2.Record.Field.Value.Type.STRING_TYPE
|
||||
request.recordSaveRequest.record.recordField[-1].value.stringValue = value
|
||||
|
||||
len_bytes = _utils.ULEB128.encode(len(request.SerializeToString()))
|
||||
|
||||
return len_bytes + request.SerializeToString()
|
||||
File diff suppressed because one or more lines are too long
Loading…
Reference in a new issue