/*
* Copyright (C) 2015 CossackLabs
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/

#import <Foundation/Foundation.h>
#import <themis/themis.h>

@interface SMessage : NSObject

{
  NSMutableData* _priv_key;
  NSMutableData* _pub_key;
}

- (id)initWithPrivateKey: (NSData*)private_key peerPublicKey:(NSData*)peer_pub_key;
- (NSData*)wrap: (NSData*)message error:(NSError**)errorPtr;
- (NSData*)unwrap: (NSData*)message error:(NSError**)errorPtr;

@end

/*
* Copyright (C) 2015 CossackLabs
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/

#import <Foundation/Foundation.h>
#import <themis/themis.h>

@interface SKeygen : NSObject

{
  NSData* _priv_key;
  NSData* _pub_key;
}

- (id)init;
- (NSData*)priv_key: error:(NSError**)errorPtr;
- (NSData*)pub_key: error:(NSError**)errorPtr;

@end

--- 
+++ 
@@ -17,15 +17,15 @@
 #import <Foundation/Foundation.h>
 #import <themis/themis.h>
 
-@interface SKeygen : NSObject
+@interface SMessage : NSObject
 
 {
-  NSData* _priv_key;
-  NSData* _pub_key;
+  NSMutableData* _priv_key;
+  NSMutableData* _pub_key;
 }
 
-- (id)init;
-- (NSData*)priv_key: error:(NSError**)errorPtr;
-- (NSData*)pub_key: error:(NSError**)errorPtr;
+- (id)initWithPrivateKey: (NSData*)private_key peerPublicKey:(NSData*)peer_pub_key;
+- (NSData*)wrap: (NSData*)message error:(NSError**)errorPtr;
+- (NSData*)unwrap: (NSData*)message error:(NSError**)errorPtr;
 
 @end
Add some correction to Obj-C wrappers
