235 lines
6.6 KiB
C
235 lines
6.6 KiB
C
#include "libnazgul.h"
|
|
#include "ids.h"
|
|
|
|
/* TODO: remplacer le bool de msgPoolData par un identifiant
|
|
de semaphore. Le semaphore contient poolNb valeurs (et indique
|
|
donc le nombre de ressources disponnibles).
|
|
|
|
*/
|
|
|
|
/* TODO: prevoir le cas des négatifs dans la taille ... */
|
|
|
|
void * msgAllocate(msgSpace *space,
|
|
int pool,
|
|
int taille,
|
|
int option
|
|
){
|
|
void * resultAddr=NULL;
|
|
int i, mSPoolDataTabFd;
|
|
msgPoolId resultPoolId;
|
|
/* tableau des valeurs des semPoolCoef/pool pour identifier le pool
|
|
* qui sera libéré le plus rapidement */
|
|
float semPoolCoef[space->poolNb];
|
|
int idxPoolOptimum;
|
|
bool gotRessourceSem;
|
|
sem_t * poolDataTabSemFd;
|
|
msgPoolDataTabSemId ressourceSemId;
|
|
sem_t * ressourceSemFd;
|
|
int ressourceSemVal;
|
|
int nbLockedSem;
|
|
float minPoolCoef;
|
|
int selectedPoolIndex;
|
|
int bufferFreeIndex;
|
|
int mSPoolFd;
|
|
msgPoolData * mSPoolDataTabAddr;
|
|
|
|
|
|
selectedPoolIndex=-1;
|
|
|
|
|
|
|
|
/* verifier le premier arg du shm_open */
|
|
mSPoolDataTabFd=shm_open(space->poolDataTabId,
|
|
O_RDWR,
|
|
MSGSPACE_DEFAULT_MODE);
|
|
if (mSPoolDataTabFd == -1 ) {
|
|
fprintf( stderr, "Allocate %s failed: %s\n",
|
|
(char*)space->poolDataTabId,
|
|
strerror( errno ) );
|
|
return NULL;
|
|
}
|
|
|
|
|
|
mSPoolDataTabAddr = mmap( 0, (space->poolNb) * sizeof( msgPoolData ),
|
|
PROT_READ | PROT_WRITE,
|
|
MAP_SHARED, mSPoolDataTabFd, 0 );
|
|
|
|
if( mSPoolDataTabAddr == MAP_FAILED) {
|
|
fprintf( stderr, "mmap failed: %s\n",
|
|
strerror( errno ) );
|
|
return NULL;
|
|
}
|
|
|
|
gotRessourceSem=false;
|
|
|
|
/* initialisation des coefs */
|
|
for (i=0;i<(space->poolNb);i++){
|
|
semPoolCoef[i]=-1;
|
|
}
|
|
|
|
nbLockedSem=0;
|
|
if ( pool == ANYPOOL){
|
|
fprintf(stderr,"[ ALLOCATION ANYPOOL : %d ]\n",(int)getpid());
|
|
// choisir le pool au hasard (ou presque)
|
|
for(i=0; i<(space->poolNb); i++) {
|
|
printf("- boucle %d\n",i); fflush(stdout);
|
|
if(mSPoolDataTabAddr[i].bufferSize >= taille) {
|
|
printf("( buffSize > taille )\n"); fflush(stdout);
|
|
/* choisir le numero du semaphore
|
|
en fonction du nombre de lock poses / nombre de buffer */
|
|
msgPoolSemIdIntern(ressourceSemId,space->id,i);
|
|
ressourceSemFd = sem_open(ressourceSemId,O_CREAT,SEM_DEFAULT_MODE,0);
|
|
if (ressourceSemFd == SEM_FAILED){
|
|
NZG_ERROR("sem_open",ressourceSemId);
|
|
return NULL;
|
|
}
|
|
/* on remplit le tableau avec les valeurs des semaphores */
|
|
|
|
if (sem_getvalue(ressourceSemFd, &ressourceSemVal) < 0){
|
|
NZG_ERROR("sem_getvalue",ressourceSemId);
|
|
return NULL;
|
|
}
|
|
printf("RESSOURCESEMVAL %d\n",ressourceSemVal);
|
|
if (ressourceSemVal <= 0){
|
|
printf("resVal < 0 : %d\n",ressourceSemVal);
|
|
/* il y a ressourceSemVal processus qui attendent déja... */
|
|
semPoolCoef[nbLockedSem] =
|
|
(float) ((1-ressourceSemVal) / mSPoolDataTabAddr[i].bufferNb);
|
|
nbLockedSem++;
|
|
}
|
|
if(sem_trywait(ressourceSemFd)==0) {
|
|
printf("got try_wait\n");
|
|
/* choisir la 1ere pool de taille plus grande
|
|
* libre si possible */
|
|
gotRessourceSem=true;
|
|
selectedPoolIndex=i;
|
|
break;
|
|
}
|
|
if( sem_close(ressourceSemFd) <0){
|
|
NZG_ERROR("sem_getvalue",ressourceSemId);
|
|
return NULL;
|
|
}
|
|
} // if buffSize > taille
|
|
} // for
|
|
|
|
printf("ERRORDETECT outFor\n"); fflush(stdout);
|
|
|
|
if (!gotRessourceSem) {
|
|
printf("Calcul du meilleur en cas de liberation\n");
|
|
minPoolCoef= semPoolCoef[0];
|
|
idxPoolOptimum = 0;
|
|
/* on cherche le pool avec le moins de lock poses / nbre de buffer
|
|
* le numéro du pool est stocké dans idxPoolOptimum */
|
|
for(i=0; i<nbLockedSem; i++) {
|
|
printf("Coef %d : %d\n",i,(int)semPoolCoef[i]);
|
|
if ((semPoolCoef[i] != -1)
|
|
&& (semPoolCoef[i] < minPoolCoef)) {
|
|
minPoolCoef = semPoolCoef[i];
|
|
idxPoolOptimum = i;
|
|
}
|
|
}
|
|
|
|
if (minPoolCoef == -1){
|
|
/* il n'y a aucune pool dont la taille satisfait la demande */
|
|
return NULL;
|
|
} else {
|
|
selectedPoolIndex=idxPoolOptimum;
|
|
}
|
|
printf("Optimum : %d\n",selectedPoolIndex);
|
|
}
|
|
}else {
|
|
fprintf(stderr,"[ ALLOCATION : %d ]\n",(int)getpid());
|
|
selectedPoolIndex=pool;
|
|
}
|
|
|
|
if (!gotRessourceSem){
|
|
msgPoolSemIdIntern(ressourceSemId,space->id,selectedPoolIndex);
|
|
ressourceSemFd=sem_open(ressourceSemId,O_CREAT,SEM_DEFAULT_MODE,0);
|
|
if(ressourceSemFd==SEM_FAILED){
|
|
NZG_ERROR("sem_open",ressourceSemId);
|
|
return NULL;
|
|
}
|
|
|
|
//TODO:virer la ligne suivante:
|
|
sem_getvalue(ressourceSemFd, &ressourceSemVal);
|
|
|
|
printf("SemWait... %s : %d\n",ressourceSemId,ressourceSemVal);
|
|
if (sem_wait(ressourceSemFd) < 0){
|
|
NZG_ERROR("sem_wait",ressourceSemId);
|
|
sem_close(ressourceSemFd);
|
|
return NULL;
|
|
}
|
|
printf("Passed %s\n",ressourceSemId);
|
|
}
|
|
|
|
/* on a acqui un semaphore pour la ressouce */
|
|
/* on acquiert le droit de modifier les infos sur la ressource */
|
|
/* on protege le tableau des associations */
|
|
poolDataTabSemFd=sem_open(space->poolDataTabSemId,
|
|
O_CREAT,SEM_DEFAULT_MODE,1);
|
|
if (poolDataTabSemFd == SEM_FAILED){
|
|
NZG_ERROR("sem_open : ouverture de la ressource",
|
|
space->poolDataTabSemId);
|
|
return NULL;
|
|
}
|
|
sem_wait(poolDataTabSemFd);
|
|
|
|
/* on modifie maintenant les données */
|
|
/* - on récupere l'index du premier buffer libre */
|
|
bufferFreeIndex = msgBufferGetFreeIndex(mSPoolDataTabAddr,selectedPoolIndex);
|
|
if (bufferFreeIndex < 0){
|
|
sem_close(poolDataTabSemFd);
|
|
// aucun buffer libre ?
|
|
return NULL;
|
|
}
|
|
printf("Buffer selected : %d,%d\n",selectedPoolIndex,bufferFreeIndex);
|
|
/* mapper le buffer libre dans l'esp addr du proc */
|
|
strcpy(resultPoolId,mSPoolDataTabAddr[selectedPoolIndex].poolId);
|
|
|
|
|
|
mSPoolFd=shm_open(resultPoolId,O_RDWR,MSGSPACE_DEFAULT_MODE);
|
|
|
|
int bufferFreeSize;
|
|
bufferFreeSize=mSPoolDataTabAddr[selectedPoolIndex].bufferSize;
|
|
printf("BufferSize : %d", bufferFreeSize);
|
|
|
|
// on mappe la totalité du pool
|
|
resultAddr = mmap( NULL,
|
|
bufferFreeSize,
|
|
PROT_NONE, //prot PROT_READ | PROT_WRITE
|
|
MAP_SHARED, mSPoolFd,
|
|
(off_t)0);
|
|
int bufferNb=mSPoolDataTabAddr[selectedPoolIndex].bufferNb;
|
|
|
|
// mprotect(resultAddr,bufferFreeSize*bufferNb,PROT_NONE);
|
|
mprotect(resultAddr,bufferFreeSize*(bufferFreeIndex+1),PROT_READ|PROT_WRITE);
|
|
//mprotect(resultAddr,bufferFreeSize*bufferFreeIndex,PROT_NONE);
|
|
resultAddr=resultAddr +( bufferFreeSize*bufferFreeIndex);
|
|
|
|
if( resultAddr == MAP_FAILED) {
|
|
NZG_ERROR("resultAddr mmap","");
|
|
goto ERROR;
|
|
}
|
|
|
|
//TODO:
|
|
/* - on s'enregistre aupres de ce buffer */
|
|
msgBufferAttachProc(mSPoolDataTabAddr,
|
|
selectedPoolIndex,
|
|
bufferFreeIndex,
|
|
resultAddr);
|
|
close(mSPoolFd);
|
|
|
|
sem_post(poolDataTabSemFd);
|
|
sem_close(poolDataTabSemFd);
|
|
|
|
/* unmapper le msgPoolDataTab */
|
|
munmap(mSPoolDataTabAddr,(space->poolNb) * sizeof( msgPoolData ));
|
|
printf( "alloc de 0x%08x\n", (int)resultAddr);
|
|
return resultAddr;
|
|
ERROR:
|
|
NZG_ERROR("msgAllocate","error processing");
|
|
sem_post(poolDataTabSemFd);
|
|
sem_close(poolDataTabSemFd);
|
|
munmap(mSPoolDataTabAddr,(space->poolNb) * sizeof( msgPoolData ));
|
|
return NULL;
|
|
}
|