forked from DaveWoodCom/XCGLogger
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add extensions to URL to read/write extended file attributes
Thanks martin-r https://stackoverflow.com/users/1187415/martin-r
- Loading branch information
1 parent
216fe75
commit bfa9fbf
Showing
2 changed files
with
92 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,82 @@ | ||
| // | ||
| // URL+ExtendedAttributes.swift | ||
| // XCGLogger: https://github.com/DaveWoodCom/XCGLogger | ||
| // | ||
| // Created by Dave Wood on 2017-04-04. | ||
| // Copyright © 2017 Dave Wood, Cerebral Gardens. | ||
| // Some rights reserved: https://github.com/DaveWoodCom/XCGLogger/blob/master/LICENSE.txt | ||
| // | ||
| // Based on code by Martin-R here: https://stackoverflow.com/a/38343753/144857 | ||
|
|
||
| import Foundation | ||
|
|
||
| extension URL { | ||
|
|
||
| /// Get extended attribute. | ||
| func extendedAttribute(forName name: String) throws -> Data? { | ||
| let data: Data? = try self.withUnsafeFileSystemRepresentation { (fileSystemPath: (UnsafePointer<Int8>?)) -> Data? in | ||
| // Determine attribute size | ||
| let length = getxattr(fileSystemPath, name, nil, 0, 0, 0) | ||
| guard length >= 0 else { return nil } | ||
|
|
||
| // Create buffer with required size | ||
| var data = Data(count: length) | ||
|
|
||
| // Retrieve attribute | ||
| let result = data.withUnsafeMutableBytes { | ||
| getxattr(fileSystemPath, name, $0, data.count, 0, 0) | ||
| } | ||
| guard result >= 0 else { throw URL.posixError(errno) } | ||
| return data | ||
| } | ||
|
|
||
| return data | ||
| } | ||
|
|
||
| /// Set extended attribute. | ||
| func setExtendedAttribute(data: Data, forName name: String) throws { | ||
| try self.withUnsafeFileSystemRepresentation { fileSystemPath in | ||
| let result = data.withUnsafeBytes { | ||
| setxattr(fileSystemPath, name, $0, data.count, 0, 0) | ||
| } | ||
| guard result >= 0 else { throw URL.posixError(errno) } | ||
| } | ||
| } | ||
|
|
||
| /// Remove extended attribute. | ||
| func removeExtendedAttribute(forName name: String) throws { | ||
| try self.withUnsafeFileSystemRepresentation { fileSystemPath in | ||
| let result = removexattr(fileSystemPath, name, 0) | ||
| guard result >= 0 else { throw URL.posixError(errno) } | ||
| } | ||
| } | ||
|
|
||
| /// Get list of all extended attributes. | ||
| func listExtendedAttributes() throws -> [String] { | ||
| let list = try self.withUnsafeFileSystemRepresentation { (fileSystemPath: (UnsafePointer<Int8>?)) -> [String] in | ||
| let length = listxattr(fileSystemPath, nil, 0, 0) | ||
| guard length >= 0 else { throw URL.posixError(errno) } | ||
|
|
||
| // Create buffer with required size | ||
| var data = Data(count: length) | ||
|
|
||
| // Retrieve attribute list | ||
| let result = data.withUnsafeMutableBytes { | ||
| listxattr(fileSystemPath, $0, data.count, 0) | ||
| } | ||
| guard result >= 0 else { throw URL.posixError(errno) } | ||
|
|
||
| // Extract attribute names | ||
| let list = data.split(separator: 0).flatMap { | ||
| String(data: Data($0), encoding: .utf8) | ||
| } | ||
| return list | ||
| } | ||
| return list | ||
| } | ||
|
|
||
| /// Helper function to create an NSError from a Unix errno. | ||
| private static func posixError(_ err: Int32) -> NSError { | ||
| return NSError(domain: NSPOSIXErrorDomain, code: Int(err), userInfo: [NSLocalizedDescriptionKey: String(cString: strerror(err))]) | ||
| } | ||
| } |